正题
题目链接:https://www.luogu.org/problem/CF451E
题目大意
长度为nnn的序列aia_iai,有如下要求
- ai∈[0...xi]a_i\in[0...x_i]ai∈[0...xi]
- ∑i=1nai=s\sum_{i=1}^na_i=s∑i=1nai=s
求序列个数。
解题思路
考虑容斥,这样我们就可以将条件转换为ai>xia_i>x_iai>xi,然后我们可以让n−xi−1n-x_i-1n−xi−1这样就将条件变为了ai≥0a_i\geq 0ai≥0,然后可以用组合数插板法计算,也就是答案为
∑SCs−∑xi∈S(xi+1)−1n−1∗(−1)∣S∣\sum_{S}C_{s-\sum_{x_i\in S}(x_i+1)-1}^{n-1}*(-1)^{|S|}S∑Cs−∑xi∈S(xi+1)−1n−1∗(−1)∣S∣
codecodecode
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const ll XJQ=1e9+7;
ll n,s,inv[50],x[50],ans;
bool v[50];
ll power(ll x,ll b)
{ll ans=1;while(b){if(b&1) ans=ans*x%XJQ;x=x*x%XJQ;b>>=1;}return ans;
}
ll C(ll n,ll m)
{if(n<0||m<0||n<m)return 0;if(n%XJQ==0||m==0)return 1;ll x=1;for(ll i=n-m+1;i<=n;i++)x=x*(i%XJQ)%XJQ;return x*inv[m]%XJQ;
}
void dfs(ll dep,ll _,bool flag)
{ll _n=s+n;for(ll i=1;i<=n;i++)if(v[i]) _n-=x[i]+1;if(flag)ans=((ans+_*C(_n-1,n-1))%XJQ+XJQ)%XJQ;if(dep>n) return;v[dep]=1;dfs(dep+1,-_,1);v[dep]=0;dfs(dep+1,_,0);
}
int main()
{scanf("%lld%lld",&n,&s);inv[1]=1;for(ll i=2;i<=20;i++)inv[i]=inv[i-1]*power(i,XJQ-2)%XJQ;for(ll i=1;i<=n;i++)scanf("%lld",&x[i]);ans=C(s+n-1,n-1);dfs(1,1,0);printf("%lld",ans);
}