Codeforces Round #433 (Div. 1, based on Olympiad of Metropolises) C. Boredom（主席树）

题目链接：http://codeforces.com/contest/853/problem/C

查二维平面里有多少点，按行维护主席树，然后每次查一个区间，做差就可以求出二维平面里的点了，然后剩下的容斥一哈就行了，具体是把周围一圈剪掉，然后把四个角加上就好了，比较简单

代码：

#include<bits/stdc++.h>using namespace std;typedef long long ll;const int MAXN=2e5+5;const int M=MAXN*40;int n,q,tot;int a[MAXN];int T[MAXN],lson[M],rson[M],c[M];int build(int l,int r){int root=tot++;c[root]=0;if(l!=r){int mid=(l+r)>>1;lson[root]=build(l,mid);rson[root]=build(mid+1,r);}return root;}int update(int root,int pos,int val){int newroot=tot++,tmp=newroot;c[newroot]=c[root]+val;int l=1,r=n;while(l<r){int mid=(l+r)>>1;if(pos<=mid){lson[newroot]=tot++;rson[newroot]=rson[root];newroot=lson[newroot];root=lson[root];r=mid;}else{rson[newroot]=tot++;lson[newroot]=lson[root];newroot=rson[newroot];root=rson[root];l=mid+1;}c[newroot]=c[root]+val;}return tmp;}int query(int root,int L,int R,int l,int r){if(L>R)return 0;if(L<=l&&r<=R)return c[root];int mid=(l+r)>>1;int ret=0;if(L<=mid)ret+=query(lson[root],L,R,l,mid);if(mid<R)ret+=query(rson[root],L,R,mid+1,r);return ret;}int Q(int l,int r,int L,int R){if(l>r)return 0;return query(T[r],L,R,1,n)-query(T[l-1],L,R,1,n);}int main(){//freopen("in.txt","r",stdin);//freopen("out.txt","w",stdout);tot=0;scanf("%d%d",&n,&q);T[0]=build(1,n);for(int i=1;i<=n;i++){scanf("%d",&a[i]);}for(int i=1;i<=n;i++){T[i]=update(T[i-1],a[i],1);}while(q--){int l,r,d,u;ll cnt=0,ans=0;scanf("%d%d%d%d",&l,&r,&d,&u);cnt=Q(1,l-1,1,n);//printf("%d\n",cnt);ans-=cnt*(cnt-1)/2;cnt=Q(d+1,n,1,n);//printf("%d\n",cnt);ans-=cnt*(cnt-1)/2;cnt=Q(1,n,u+1,n);//printf("%d\n",cnt);ans-=cnt*(cnt-1)/2;cnt=Q(1,n,1,r-1);//printf("%d\n",cnt);ans-=cnt*(cnt-1)/2;cnt=Q(1,l-1,1,r-1);//printf("%d\n",cnt);ans+=cnt*(cnt-1)/2;cnt=Q(1,l-1,u+1,n);//printf("%d\n",cnt);ans+=cnt*(cnt-1)/2;cnt=Q(d+1,n,1,r-1);//printf("%d\n",cnt);ans+=cnt*(cnt-1)/2;cnt=Q(d+1,n,u+1,n);//printf("%d\n",cnt);ans+=cnt*(cnt-1)/2;//printf("%lld\n",ans);ans+=1LL*n*(n-1)/2;printf("%lld\n",ans);}return 0;}