Codeforces 316D PE Lesson

我们转变思路，考虑给出一个循环判断它是否合法。如果这个循环里没有$1$，显然合法。如果有一个$1$，将它与旁边的$2$交换得到一个形式相同、规模更小的问题，而当只有两个的时候直接交换即可。如果有两个$1$和上面一样。但是如果有三个$1$，最后剩下构成三角关系的$1$就不能解决了。更多当然更不行。所以我们知道一个循环合法的充要条件是里面的$1$不超过两个。
先考虑安排$1$，每个$1$要不然独立成循环，要不然和之前的一个$1$成循环，可以写出递推式$f_i=f_{i-1}+(i-1)f_{i-2}$。
再考虑把$2$加入，每个加入的$2$可以放在之前加入的任何一个元素后面，所以最终答案是$f_m\prod_{i=m+1}^n i$。

#include<cstdio>
#include<algorithm>
using namespace std;
#define LL long long
const int maxn=1000010,p=1000000007;
int n,m,dp[maxn];
int main()
{
    int x,ans;
    scanf("%d",&n);
    for (int i=1;i<=n;i++)
    {
        scanf("%d",&x);
        if (x==1) m++;
    }
    dp[0]=dp[1]=1;
    for (int i=2;i<=m;i++)
        dp[i]=(dp[i-1]+(LL)dp[i-2]*(i-1))%p;
    ans=dp[m];
    for (int i=m+1;i<=n;i++) ans=(LL)ans*i%p;
    printf("%d\n",ans);
}