COJ 0967 WZJ的数据结构（负三十三）

WZJ的数据结构（负三十三）

难度级别：E； 运行时间限制：7000ms； 运行空间限制：262144KB； 代码长度限制：2000000B

试题描述

请你设计一个数据结构，完成以下功能：

给定一个大小为N的整数组A，要求你回答执行N次操作。操作分两种：

操作1：每次操作给你l,r,v三个参数，求Al至Ar中值<=v的个数。

操作2：每次操作给你l,r,v三个参数，将Al至Ar所有数的值设为v。

输入

第一行为一个正整数N。
第二行为N个整数Ai。
接下来N行每行4个正整数t,l,r,v。若t=2表示操作1，t=1表示操作2。

输出

对每个操作1输出结果。

输入示例

6
1 2 2 1 2 3
2 1 4 2
2 1 4 1
1 2 4 1
2 1 4 1
1 3 5 2
2 2 5 2

输出示例

4
2
4
4

其他说明

1<=N<=200000
1<=Ai,v<=10^9

题解：标程肯定是什么线段树分治什么的奇怪的东西，窝不会写呀，，，就写分块吧，于是就虐标程了哦，excited！

 #include<iostream>
 #include<cstdio>
 #include<cmath>
 #include<algorithm>
 #include<queue>
 #include<cstring>
 #define PAU putchar(' ')
 #define ENT putchar('\n')
 using namespace std;
 const int maxn=+,maxb=,inf=-1u>>;
 int z[maxn],st[maxb],en[maxb],size,n,A[maxn],B[maxn],set[maxb],siz[maxb];
 inline int read(){
     int x=,sig=;char ch=getchar();
     for(;!isdigit(ch);ch=getchar())if(ch=='-')sig=;
     for(;isdigit(ch);ch=getchar())x=*x+ch-'';
     return sig?x:-x;
 }
 inline void write(int x){
     if(x==){putchar('');return;}if(x<)putchar('-'),x=-x;
     int len=,buf[];while(x)buf[len++]=x%,x/=;
     for(int i=len-;i>=;i--)putchar(buf[i]+'');return;
 }
 void build(int b){
     for(int i=st[b];i<=en[b];i++)B[i]=A[i];sort(B+st[b],B+en[b]+);return;
 }
 void down(int b){
     if(set[b]!=inf){for(int i=st[b];i<=en[b];i++)A[i]=B[i]=set[b];set[b]=inf;}return;
 }
 void sett(int L,int R,int cv){
     down(z[L]);for(int i=L;i<=R;i++)A[i]=cv;build(z[L]);return;
 }
 void setb(int L,int R,int cv){
     for(int b=L;b<=R;b++)set[b]=cv;return;
 }
 void seto(int ql,int qr,int cv){
     if(z[ql]==z[qr])sett(ql,qr,cv);
     else sett(ql,en[z[ql]],cv),setb(z[ql]+,z[qr]-,cv),sett(st[z[qr]],qr,cv);return;
 }
 int queryb(int L,int R,int cv){
     int ans=;
     for(int b=L;b<=R;b++){
         if(set[b]!=inf){
             if(set[b]<=cv)ans+=siz[b];
         }else ans+=upper_bound(B+st[b],B+en[b]+,cv)-B-st[b];
     }return ans;
 }
 int queryt(int L,int R,int cv){
     int ans=;
     if(set[z[L]]!=inf){
         if(set[z[L]]<=cv)ans+=(R-L+);
     }else for(int i=L;i<=R;i++)if(A[i]<=cv)ans++;
     return ans;
 }
 int queryo(int ql,int qr,int cv){
     if(z[ql]==z[qr])return queryt(ql,qr,cv);
     else return queryt(ql,en[z[ql]],cv)+queryb(z[ql]+,z[qr]-,cv)+queryt(st[z[qr]],qr,cv);
 }
 void init(){
     n=read();size=sqrt((double)n*0.9);int tp,ql,qr,cv;
     for(int i=;i<=n;i++){
         A[i]=B[i]=read();
         z[i]=(i-)/size+;
         if(!st[z[i]])st[z[i]]=i;
         en[z[i]]=i;
     }
     for(int b=;b<=z[n];b++)build(b),set[b]=inf,siz[b]=en[b]-st[b]+;
     for(int i=;i<=n;i++){
         tp=read();ql=read();qr=read();cv=read();
         if(tp==)write(queryo(ql,qr,cv)),ENT;
         else seto(ql,qr,cv);
     }
     return;
 }
 void work(){
     return;
 }
 void print(){
     return;
 }
 int main(){init();work();print();return ;}

总结一下这次WA的tip：queryt的时候窝萌是要暴力的呀。。。。。。。。。。。。

还有就是一开始没写B数组哦。。。。。。