【Codeforces316D3】【CF316D3】PE Lesson

题目大意

有n个小朋友手里各有1个球，其中每个人只能把手里的球和别人交换1或2次。输入每个人可交换的次数，求可能的交换方案数。

题解

每个人都把球给一个人，如果将u想要让v拿到自己的球看做是u到v有向边，那么可以把换球方案看做是若干个环。
其中，若一个环上有两个人或更少是体力为1的，则可以进行交换。其中有0个易证。有1个的交换方法如下，其中黑箭头表示想让谁拿自己的球，空心箭头表示一次交换

有两个的交换方法如下

知道这个就很好办了
先判断体力为1的分为大小不超过二的若干组
设有i个人体力为1的方案数是F(i)
则有$F_i = F_{i-1} + F_{i-2}*(i-1)$
推导：对于新加的一个人，可以让他单独一组（$F_{i-1}$），或者从前i个里面任选一个配对（$F_{i-2}*(i-1)$）
然后对于每一个体力为2的可以让他们站到任意一个已排好的人后面或者单独成环，故有（人数+1）种方案
最终答案就是$F_i * \frac{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);
}