CodeForces - 853C 主席树维护矩形内的点数

n*n的矩阵 每一行给一个点 （保证列不同） 共n个点  这些点可以构成 n*(n-1)/2 个美丽矩阵  美丽矩阵的定义是 以两个点为对角的矩阵   现在有q个询问  每个询问给一个矩阵 问这个矩阵和多少个美丽矩阵相交  

我们得这样考虑问题   把这个n*n的矩阵分成9块

7 8 9

4 5 6

1 2 3  

其中5对应的是我们询问的矩阵 

那么我们可以一次求出每一块矩阵有多少个点  再来统计答案 （保证和5相交）我们用cnt[i] 来表示第i块矩阵有多少个点

以第一块矩阵为例  满足条件的答案是  cnt[1]*(cnt[5]+cnt[6]+cnt[7]+cnt[8]) 

至于cnt如何计算  这其实就是个简单的二维前缀和  但是n比较大我们就不能 用数组去表示（空间太大）

那就只能用主席树了 （树状数组也行 不过感觉有点麻烦）   另外这样算的话 每一个美丽矩阵我们算了两次 答案应该除2

不过主席树的可行性也是因为点最多只有n个  假如点更多的话 空间的消耗会非常大

#include<bits/stdc++.h>
using namespace std;
const int N = 2e5+100;
int tot,sum[N*25],L[N*25],R[N*25],rt[N];
typedef long long ll;
void update(int &o,int last,int l,int r,int pos){
	o=++tot;
	sum[o]=sum[last]+1;
	if(l==r) return;
	int mid = l+r>>1;
	if(pos<=mid) R[o]=R[last],update(L[o],L[last],l,mid,pos);
	else L[o]=L[last],update(R[o],R[last],mid+1,r,pos); 
}
int query(int rtl,int rtr,int ql,int qr,int l,int r){
	if(ql<=l&&qr>=r) return sum[rtr]-sum[rtl];
	int ret = 0,mid = l+r>>1;
	if(ql<=mid) ret+=query(L[rtl],L[rtr],ql,qr,l,mid);
	if(qr>mid) ret+=query(R[rtl],R[rtr],ql,qr,mid+1,r);
	return ret;
}
int main(){
	int n,m;
	scanf("%d%d",&n,&m);
	for(int i = 1; i <= n; i++){
		int x;
		scanf("%d",&x);
		update(rt[i],rt[i-1],0,n,x);
	}
	for(int i = 1; i <= m; i++){
		int l,d,r,u;
		ll sum[10],ans=0;
		scanf("%d%d%d%d",&l,&d,&r,&u);
		sum[1]=1ll*query(rt[0],rt[l-1],1,d-1,0,n);
		sum[2]=1ll*query(rt[l-1],rt[r],1,d-1,0,n);
		sum[3]=1ll*query(rt[r],rt[n],1,d-1,0,n);
		sum[4]=1ll*query(rt[0],rt[l-1],d,u,0,n);
		sum[5]=1ll*query(rt[l-1],rt[r],d,u,0,n);
		sum[6]=1ll*query(rt[r],rt[n],d,u,0,n);
		sum[7]=1ll*query(rt[0],rt[l-1],u+1,n,0,n);
		sum[8]=1ll*query(rt[l-1],rt[r],u+1,n,0,n);
		sum[9]=1ll*query(rt[r],rt[n],u+1,n,0,n);
		ans+=sum[1]*(sum[5]+sum[6]+sum[8]+sum[9]);
		ans+=sum[2]*(sum[4]+sum[5]+sum[6]+sum[7]+sum[8]+sum[9]);
		ans+=sum[3]*(sum[4]+sum[5]+sum[7]+sum[8]);
		ans+=sum[4]*(sum[2]+sum[3]+sum[5]+sum[6]+sum[8]+sum[9]);
		ans+=sum[5]*1ll*(n-1);
		ans+=sum[6]*(sum[1]+sum[2]+sum[4]+sum[5]+sum[7]+sum[8]);
		ans+=sum[7]*(sum[2]+sum[3]+sum[5]+sum[6]);
		ans+=sum[8]*(sum[1]+sum[2]+sum[3]+sum[4]+sum[5]+sum[6]);
		ans+=sum[9]*(sum[1]+sum[2]+sum[4]+sum[5]);
		printf("%lld\n",ans/2ll);
	}
	return 0;
}