2020 ICPC, COMPFEST 12, Indonesia Multi-Provincial Contest D. Danger of Mad Snakes(容斥 二维前缀和)

题目

1000*1000的cell图，某些cell里有蛇，一共n(n<=2e3)条蛇，

第i条的坐标(xi,yi)，其危险值为bi，

你需要选中其中m(m<=n)条蛇，

距离选中的蛇的距离在的蛇会一并被选中，

最终这个选法对答案的贡献是，所有被选中的蛇的bi值之和的平方，

求所有不同选法的贡献值之和，答案模1e9+7

思路来源

jiangly代码

题解

算贡献的一道题，是平方项加乘积二倍，

考虑将合式打开，考虑的出现次数，选法同时包含了ij，

这个并集很难取，难以确定ij是被哪个覆盖的，被一个同时覆盖还是被两个覆盖还是都有，

但是，同时包含ij=随便取-不同时包含ij，不同时包含ij=不包含i+不包含j-既不包含i也不包含j

而既不包含i也不包含j=随便取-i的矩形-j的矩形 + ij的矩形交，二维前缀和统计即可

而事实上时，另一个随便取，就有会出现在这个和式里，所以枚举上等于的即统一了情况

矩阵求交，就是两个一维的线段求交

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e3+10,mod=1e9+7;
int n,m,r,x[N],y[N],b[N],sum[N][N],c[N][N];
ll ans;
int cal(int l1,int r1,int l2,int r2){
	if(l1>l2 || r1>r2)return 0;
	l1=max(l1,1);r1=max(r1,1);//不能在(1,1)左上 
	l2=min(l2,1000);r2=min(r2,1000); 
	return sum[l2][r2]-sum[l2][r1-1]-sum[l1-1][r2]+sum[l1-1][r1-1];
}
int main(){
	scanf("%d%d%d",&n,&m,&r);
	for(int i=1;i<=n;++i){
		scanf("%d%d%d",&x[i],&y[i],&b[i]);
		sum[x[i]][y[i]]++;
	}
	for(int i=1;i<=1000;++i){
		for(int j=1;j<=1000;++j){
			sum[i][j]=sum[i][j]+sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1];
		}
	}
	for(int i=0;i<=n;++i){
		c[i][0]=1;
		for(int j=1;j<=i;++j){
			c[i][j]=(c[i-1][j-1]+c[i-1][j])%mod;
		}
	}
	for(int i=1;i<=n;++i){//i==j时 说明另一个是什么都无所谓 都有贡献bi^2 
		for(int j=1;j<=n;++j){
			ll v=1ll*b[i]*b[j]%mod;//同时包含ij的方案 总方案减去不全包含ij的方案 
			//不全包含=不包含i+不包含j-同时不包含ij 
			//printf("num:%lld\n",n-cal(x[i]-r,y[i]-r,x[i]+r,y[i]+r));
			//p/rintf("num:%lld\n",n-cal(x[j]-r,y[j]-r,x[j]+r,y[j]+r));
			//printf("num:%lld\n",n-cal(x[i]-r,y[i]-r,x[i]+r,y[i]+r)-cal(x[j]-r,y[j]-r,x[j]+r,y[j]+r)+cal(max(x[i]-r,x[j]-r),max(y[i]-r,y[j]-r),min(x[i]+r,x[j]+r),min(y[i]+r,y[j]+r)));
			ll noi=c[n-cal(x[i]-r,y[i]-r,x[i]+r,y[i]+r)][m];
			ll noj=c[n-cal(x[j]-r,y[j]-r,x[j]+r,y[j]+r)][m];
			ll noij=c[n-cal(x[i]-r,y[i]-r,x[i]+r,y[i]+r)-cal(x[j]-r,y[j]-r,x[j]+r,y[j]+r)+cal(max(x[i]-r,x[j]-r),max(y[i]-r,y[j]-r),min(x[i]+r,x[j]+r),min(y[i]+r,y[j]+r))][m];//独立观察两维 矩形求交即线段求交 
			ll no=noi+noj-noij;
			ll ways=c[n][m]-no;
			ways%=mod;
			if(ways<0)ways+=mod;
			ans=(ans+ways*v%mod)%mod; 
			//p//rintf("i:%d j:%d fangan:%d\n",i,j,ways*v%mod);
		}
	}
	printf("%lld\n",ans);
	return 0;	
}