BZOJ.4767.两双手(组合 容斥 DP)

题目链接

(Description)

棋盘上((0,0))处有一个棋子。棋子只有两种走法,分别对应向量((A_x,A_y),(B_x,B_y))。同时棋盘上有(n)个障碍点((x_i,y_i)),棋子在任何时刻都不能跳到障碍点。
求棋子从((0,0))跳到((E_x,E_y))的方案数。答案对(10^9+7)取模。

(Solution)

注意到(A_x*B_y-A_y*B_x eq0),即两向量不共线,从某个点走到另一个点,两种方式分别所用次数(x,y)是确定的。即求该方程组的非负整数解:$$left{egin{array}{lr}A_xx+B_xy=X_iA_yx+B_yy=Y_iend{array} ight.$$

同网格图方案数,从某个点以两种方式分别走(x,y)步到达另一个点,这样的方案数为(inom{x+y}{x})
将每个点表示成这样的(x,y)(从((0,0))出发到达该点两种方式分别所需步数)后,任意两点所需的步数就是(x_i-x_j,y_i-y_j)了。方案数同样可以用组合数求。
然后就可以排序后容斥了。
(f(i))表示在(i)之前不经过任何障碍点,到达障碍点(i)的方案数。记(cnt(i,j))表示从障碍点(i)(j)的方案数。将起点视为障碍点(0),那么$$f(i)=cnt(0,i)-sum_{j=1}^{i-1}f(j)*cnt(j,i)$$

将终点视为第(n+1)个障碍点,答案就是(f(n+1))了。
复杂度(O(n^2))

因为(A_x,A_y,B_x,B_y)可能有负数,所以要走的步数是(n^2)级别的(比如((1,0),(-500,1)))。组合数要(n+m)所以上界要到(2n^2)

//4736kb	448ms 为啥这么慢呢
#include <cstdio>
#include <cctype>
#include <algorithm>
#define gc() getchar()
#define mod 1000000007
typedef long long LL;
const int N=505,M=500000;

int Ax,Ay,Bx,By,f[N],fac[M+2],ifac[M+2];
struct Point
{
	int x,y;
	bool operator <(const Point &a)const{
		return x==a.x?y<a.y:x<a.x;
	}
}p[N];

inline int read()
{
	int now=0,f=1;register char c=gc();
	for(;!isdigit(c);c=='-'&&(f=-1),c=gc());
	for(;isdigit(c);now=now*10+c-'0',c=gc());
	return now*f;
}
inline int FP(int x,int k)
{
	int t=1;
	for(; k; k>>=1,x=1ll*x*x%mod)
		if(k&1) t=1ll*t*x%mod;
	return t;
}
void Calc(int yi,int xi,int &x,int &y)
{
	int a=xi*By-yi*Bx,b=Ax*By-Ay*Bx;
	if(!b||a%b) {x=-1; return;}//判0。。
	int c=xi*Ay-yi*Ax,d=Bx*Ay-By*Ax;
	if(!d||c%d) {x=-1; return;}
	x=a/b, y=c/d;
}
inline int C(int n,int m)
{
	if(n<0||m<0) return 0;//return C(n+m,n)
	return 1ll*fac[n+m]*ifac[n]%mod*ifac[m]%mod;
}

int main()
{
	fac[0]=fac[1]=1;
	for(int i=2; i<=M; ++i) fac[i]=1ll*fac[i-1]*i%mod;
	ifac[M]=FP(fac[M],mod-2);
	for(int i=M-1; ~i; --i) ifac[i]=1ll*ifac[i+1]*(i+1)%mod;

	int Ex=read(),Ey=read(),n=read(),cnt=0;
	Ax=read(), Ay=read(), Bx=read(), By=read();

	Calc(Ey,Ex,Ex,Ey);
	if(Ex<0||Ey<0) return puts("0"),0;
	p[++cnt]=(Point){Ex,Ey};

	for(int x,y; n--; )
	{
		Calc(read(),read(),x,y);
		if(!(x<0||y<0||x>Ex||y>Ey)) p[++cnt]=(Point){x,y};//需要走更多步的不会相交(否则排序还出问题?)
	}
	n=cnt, std::sort(p+1,p+1+n);

	for(int i=1; i<=n; ++i)
	{
		Point now=p[i]; LL tmp=C(now.x,now.y);
		if(!tmp) continue;
		for(int j=1; j<i; ++j)
			tmp+=mod-1ll*f[j]*C(now.x-p[j].x,now.y-p[j].y)%mod;
		f[i]=(int)(tmp%mod);
	}
	printf("%d
",f[n]);

	return 0;
}
原文地址:https://www.cnblogs.com/SovietPower/p/9753152.html