[CSP-S模拟测试]:多维网格(组合数学+容斥)

题目传送门(内部题138)


输入格式

  输入数据第一行为两个整数$d,n$。
  第二行$d$个非负整数$a_1,a_2,...,a_d$。    
  接下来$n$行,每行$d$个整数,表示一个坏点的坐标。数据保证坏点在网络范围内,且不会是点$A$或点$B$。


输出格式

  一个整数,为从点$A$移动到点$B$的不同的路径数对$10^9+7$取模后的值。


样例

样例输入:

2 1
2 1
1 0

样例输出:

1


数据范围与提示


题解

先来考虑$n=0$的情况,利用组合数学,答案就是:

$$ans=(\sum\limits_{i=1}^d a_i)!\times \prod\limits_{i=1}^d(a_i!)^{-1}$$。

再来考虑$n\neq 0$的情况。

考虑容斥。

不妨设$dp[i]$表示从$A$点出发到达$i$点的合法路径条数,$g[i][j]$表示从$i$到$j$的任意路径条数,则有:

$$dp[i]=g[A][i]-\sum\limits_{j=1}^{i-1}dp[j]\times g[j][i]$$

时间复杂度:$\Theta(n^2d)$。

期望得分:$100$分。

实际得分:$100$分。


代码时刻

#include<bits/stdc++.h>
using namespace std;
const int mod=1000000007;
struct rec{int d[101];}a[502];
int d,n;
int fac[10000001],inv[10000001];
long long dp[502];
long long qpow(long long x,long long y)
{
	long long res=1;
	while(y)
	{
		if(y&1)res=res*x%mod;
		x=x*x%mod;y>>=1;
	}
	return res;
}
void pre_work()
{
	fac[0]=1;
	for(int i=1;i<=10000000;i++)fac[i]=1LL*fac[i-1]*i%mod;
	inv[10000000]=qpow(fac[10000000],mod-2);
	for(int i=10000000;i;i--)inv[i-1]=1LL*inv[i]*i%mod;
}
bool cmp(rec a,rec b){for(int i=1;i<=d;i++)if(a.d[i]!=b.d[i])return a.d[i]<b.d[i];}
long long ask(rec a,rec b)
{
	int now=0;
	long long res=1;
	for(int i=1;i<=d;i++)
	{
		if(b.d[i]<a.d[i])return 0;
		now+=b.d[i]-a.d[i];
		res=res*inv[b.d[i]-a.d[i]]%mod;
	}
	return res*fac[now]%mod;
}
int main()
{
	pre_work();scanf("%d%d",&d,&n);
	for(int i=1;i<=n+1;i++)
		for(int j=1;j<=d;j++)
			scanf("%d",&a[i].d[j]);
	sort(a+1,a+n+2,cmp);
	for(int i=1;i<=n+1;i++)
	{
		dp[i]=ask(a[0],a[i]);
		for(int j=1;j<i;j++)dp[i]=(dp[i]-ask(a[j],a[i])*dp[j]%mod+mod)%mod;
	}
	printf("%lld",dp[n+1]);
	return 0;
}

rp++

posted @ 2019-11-10 15:17  HEOI-动动  阅读(204)  评论(0编辑  收藏  举报