Codechef：Counting D-sets/CNTDSETS（容斥）

最新推荐文章于 2021-01-19 08:17:14 发布

原创最新推荐文章于 2021-01-19 08:17:14 发布 · 419 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

容斥原理专栏收录该内容

23 篇文章

订阅专栏

本文探讨了在计算几何中，如何使用组合数进行容斥原理计算特定条件下点集方案的数量。通过分析最大值限制和坐标限制，提出了一种时间复杂度为O(Tnlogn)的算法，用于求解最大值小于等于D的方案数减去小于等于D-1的方案数。

传送门

题解：
平移的限制相当于强制每一维存在坐标0。
距离为D的限制相当于存在一维的最大值为D。

考虑计算最大值小等D的方案数减去小等于D-1的方案数，限制只有每一维都存在0，这个用组合数来容斥一下就行了，时间复杂度 $\log n)$ 。

#include <bits/stdc++.h>
using namespace std;

const int RLEN=1<<18|1;
inline char nc() {
	static char ibuf[RLEN],*ib,*ob;
	(ib==ob) && (ob=(ib=ibuf)+fread(ibuf,1,RLEN,stdin));
	return (ib==ob) ? -1 : *ib++;
}
inline int rd() {
	char ch=nc(); int i=0,f=1;
	while(!isdigit(ch)) {if(ch=='-')f=-1; ch=nc();}
	while(isdigit(ch)) {i=(i<<1)+(i<<3)+ch-'0'; ch=nc();}
	return i*f;
}

const int N=1e3+50, mod=1e9+7;
inline int add(int x,int y,int md=mod) {return (x+y>=md) ? (x+y-md) : (x+y);}
inline int dec(int x,int y,int md=mod) {return (x-y<0) ? (x-y+md) : (x-y);}
inline int mul(int x,int y,int md=mod) {return (long long)x*y%md;}
inline int power(int a,int b,int md=mod,int rs=1) {for(;b;b>>=1,a=mul(a,a,md)) if(b&1) rs=mul(rs,a,md); return rs;}
inline void up(int &x,int y) {x=add(x,y);}
inline void dn(int &x,int y) {x=dec(x,y);}
int n;
struct Combin {
	int fac[N],ifac[N];
	Combin() {
		fac[0]=1;
		for(int i=1;i<N;i++) fac[i]=mul(fac[i-1],i);
		ifac[N-1]=power(fac[N-1],mod-2);
		for(int i=N-2;~i;i--) ifac[i]=mul(ifac[i+1],i+1);
	}
	inline int C(int a,int b) {return mul(fac[a],mul(ifac[b],ifac[a-b]));}
} C;
inline int solve(int x) {
	if(x<0) return 0;
	int ans=0;
	for(int i=0;i<=n;i++) {
		int tot=mul(power(x+1,n-i,mod-1),power(x,i,mod-1),mod-1);
		int res=power(2,tot);
		if(i&1) dn(ans,mul(res,C.C(n,i)));
		else up(ans,mul(res,C.C(n,i)));
	} return ans;
}
inline void solve() {
	n=rd(); int D=rd();
	printf("%d\n",dec(solve(D),solve(D-1)));
}
int main() {
	for(int i=rd();i;i--) solve();
}