COJ 0986 WZJ的数据结构（负十四） 区间动态k大

 

题解：哈哈哈我过了！！！主席树+树状数组写起来还真是hentai啊。。。

在这里必须分享我的一个沙茶错！！！看这段代码：

void get(int x,int d){
    if(!d)for(lt[ltot=1]=root[x];x;x-=x&-x)if(c[x])lt[++ltot]=c[x];
    else  for(rt[rtot=1]=root[x];x;x-=x&-x)if(c[x])rt[++rtot]=c[x];return;
}

似乎挺正常是吧？而且缩进也很优美对不对！！！

可是！！！！！！！！窝萌发现它是错的！！！！！！！！！！

因为下面那个else跟上面的if配对了！！！！！！所以这段程序其实是这样的：

void get(int x,int d){
    if(!d)for(lt[ltot=1]=root[x];x;x-=x&-x){
        if(c[x])lt[++ltot]=c[x];
        else{
            for(rt[rtot=1]=root[x];x;x-=x&-x)
            if(c[x])rt[++rtot]=c[x];
        }
    }return;
}

我要疯了！！！！！！！！！！！！！！！！！！！

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<cmath>
 4 #include<algorithm>
 5 #include<queue>
 6 #include<cstring>
 7 #define PAU putchar(' ')
 8 #define ENT putchar('\n')
 9 #define CH for(int d=0;d<2;d++)if(ch[d])
10 #define lson ls(x),y->ch[0],L,M
11 #define rson rs(x),y->ch[1],M+1,R
12 using namespace std;
13 const int maxn=100000+10,maxnode=20000000+10,inf=-1u>>1,vl=1,vr=100000;
14 struct node{
15     node*ch[2];int siz;node(){siz=0;}
16 }pol[maxnode],*nodecnt=pol,*root[maxn],*c[maxn],*lt[maxn],*rt[maxn];int ltot,rtot;
17 node*ls(node*x){return x?x->ch[0]:x;}
18 node*rs(node*x){return x?x->ch[1]:x;}
19 int sz(node*x){return x?x->siz:0;}
20 int n,m,A[maxn],cv;
21 void build(int v,int pos,node*x,node*&y,int L=vl,int R=vr){
22     y=nodecnt++;y->siz=sz(x)+v;if(L==R)return;int M=L+R>>1;
23     if(pos<=M)y->ch[1]=rs(x),build(v,pos,lson);else y->ch[0]=ls(x),build(v,pos,rson);return;
24 }
25 void update(int x,int v){
26     for(int w=x;w<=n;w+=w&-w)build(-1,A[x],c[w],c[w]);
27     for(int w=x;w<=n;w+=w&-w)build(1,A[x]=v,c[w],c[w]);return;
28 }
29 void get(int x,int d){
30     if(!d){for(lt[ltot=1]=root[x];x;x-=x&-x)if(c[x])lt[++ltot]=c[x];}
31     else  {for(rt[rtot=1]=root[x];x;x-=x&-x)if(c[x])rt[++rtot]=c[x];}return;
32 }
33 inline int read(){
34     int x=0,sig=1;char ch=getchar();
35     for(;!isdigit(ch);ch=getchar())if(ch=='-')sig=0;
36     for(;isdigit(ch);ch=getchar())x=10*x+ch-'0';
37     return sig?x:-x;
38 }
39 inline void write(int x){
40     if(x==0){putchar('0');return;}if(x<0)putchar('-'),x=-x;
41     int len=0,buf[15];while(x)buf[len++]=x%10,x/=10;
42     for(int i=len-1;i>=0;i--)putchar(buf[i]+'0');return;
43 }
44 void init(){
45     n=read();m=read();
46     for(int i=1;i<=n;i++)build(1,A[i]=read(),root[i-1],root[i]);
47     return;
48 }
49 void work(){
50     int x,y,v;
51     while(m--){
52         if(!read())x=read(),v=read(),update(x,v);
53         else{x=read();y=read();v=read();
54             get(x-1,0);get(y,1);int L=vl,R=vr,M,kth;
55             while(L<R){M=L+R>>1;kth=0;//printf("%d %d %d %d\n",L,R,ltot,rtot);
56                 for(int i=1;i<=ltot;i++)kth-=sz(ls(lt[i]));
57                 for(int i=1;i<=rtot;i++)kth+=sz(ls(rt[i]));
58                 if(kth>=v){R=M;
59                     for(int i=1;i<=ltot;i++)lt[i]=ls(lt[i]);
60                     for(int i=1;i<=rtot;i++)rt[i]=ls(rt[i]);
61                 }else{L=M+1;v-=kth;
62                     for(int i=1;i<=ltot;i++)lt[i]=rs(lt[i]);
63                     for(int i=1;i<=rtot;i++)rt[i]=rs(rt[i]);
64                 }
65             }write(L);ENT;
66         }
67     }
68     return;
69 }
70 void print(){
71     return;
72 }
73 int main(){init();work();print();return 0;}