「美团 CodeM 初赛 Round A」二分图染色 容斥原理

题目链接点这里

首先绿色不用管

然后只有2种颜色，很显然需要用容斥原理求，先算全部的数量，然后容斥减去不合法的数目

我们n个点设单种颜色的方案数为F(n)，所以2种颜色的方案数就是F（n）^2（包含不合法），

先看直接列出来的式子：

#include<iostream>
#include<cstdio>
#include<math.h>
#include<algorithm>
#include<map>
#include<set>
#include<bitset>
#include<unordered_map>
#include<stack>
#include<queue>
#include<string.h>
#include<cstring>
#include<vector>
#include<time.h>
#include<stdlib.h>
using namespace std;
#define INF 0x3f3f3f3f
#define INFLL 0x3f3f3f3f3f3f3f3f
#define FIN freopen("input.txt","r",stdin)
#define mem(x,y) memset(x,y,sizeof(x))
typedef unsigned long long ULL;
typedef long long LL;
#define fuck(x) cout<<"q"<<endl;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
typedef pair<pair<int,int>,int> PIII;
typedef pair<int,int> PII;
const double eps=1e-8;
const int P=1e9+7;
#define MX 11111111//最大范围
int n,A[MX];
int F[MX],invF[MX];
int quick_pow(int a,int x)
{
    int ans=1;
    while(x)
    {
        if(x&1)ans=(LL)ans*a%P;
        a=(LL)a*a%P;
        x>>=1;
    }
    return ans;
}
void init(int n)
{
    F[0] = 1;
    for(int i = 1; i <=n; i++)  F[i] = ((LL)F[i - 1] * i) % P;
    invF[n] = quick_pow(F[n], P - 2);
    for(int i = n-1; i >= 0; i--)  invF[i] = (LL)invF[i + 1] * (i + 1) % P;
    A[0]=1;
    A[1]=2;
    for(int i=2; i<=n; i++)A[i]=((LL)2*i*A[i-1]%P-(LL)A[i-2]*(i-1)%P*(i-1)%P+P)%P;
}
int main()
{
    FIN;
    while(cin>>n)
    {
        init(n);
        int ans=0;
        for(int i=0; i<=n; i++)
            ans=(ans+(i&1?-1:1)*((LL)F[n]*F[n]%P*invF[i]%P*invF[n-i]%P*invF[n-i]%P*A[n-i]%P*A[n-i]%P)+P)%P;
        cout<<ans<<endl;
    }
    return 0;
}