题目大意
有n个小朋友手里各有1个球,其中每个人只能把手里的球和别人交换1或2次。输入每个人可交换的次数,求可能的交换方案数。
题解
每个人都把球给一个人,如果将u想要让v拿到自己的球看做是u到v有向边,那么可以把换球方案看做是若干个环。
其中,若一个环上有两个人或更少是体力为1的,则可以进行交换。其中有0个易证。有1个的交换方法如下,其中黑箭头表示想让谁拿自己的球,空心箭头表示一次交换
有两个的交换方法如下
知道这个就很好办了
先判断体力为1的分为大小不超过二的若干组
设有i个人体力为1的方案数是F(i)
则有
Fi=Fi−1+Fi−2∗(i−1)
推导:对于新加的一个人,可以让他单独一组(
Fi−1
),或者从前i个里面任选一个配对(
Fi−2∗(i−1)
)
然后对于每一个体力为2的可以让他们站到任意一个已排好的人后面或者单独成环,故有(人数+1)种方案
最终答案就是
Fi∗n!m!
,m是体力为1的人的数量
代码
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
#define maxn 1000007
#define long long long
#define mod 1000000007
int n;
long f[maxn];
long ans;
int m;
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
int x;
scanf("%d",&x);
if(x==1) m++;
}
f[0]=f[1]=1ll;
for(int i=2;i<=m;i++)
f[i]=(f[i-1]+f[i-2]*(i-1)%mod)%mod;
ans=f[m];
for(int i=m+1;i<=n;i++)
ans=(ans*i)%mod;
printf("%lld",ans);
}