二分图染色(弱化版)（牛客4.10 组合 容斥）

题目链接
题解链接

题目描述
给定一个完全二分图，图的左右两边的顶点数目相同。我们要给图中的每条边染成红色、蓝色、或者绿色，并使得任意两条红边不共享端点、同时任意两条蓝边也不共享端点。
计算所有满足条件的染色的方案数，并对 $10^9+7$ 取模。
(ps：本题数据量与实际比赛中数据量相比，少了一些)
输入描述:
二分图单边的顶点数目 $n(n\le 10^7)$
输出描述:
输出一个整数,即所求的答案。
输入
2
输出
35

思路：
二分图转化成 $n\times n$ 的棋盘，行和列分别代表左边的点集和右边的点集，棋盘上的点表示边，比如 $i$ 行 $j$ 列就表示左边第 $i$ 个点和右边第 $j$ 个点连边，那么棋盘上任意一个点都是不同的边。绿边其实是无所谓的，它相当于是空着不连的边。
这个题使用容斥的思想，好像涉及二项式反演，菜鸡我不会😭

Code：

#include <bits/stdc++.h>
using namespace std;
const int mod = 1e9 + 7;
const int N = 1e7 + 5;
int n, f[N], fac[N], inv[N], ans;
int qpow(int a, int b, int p)
{
    int res = 1;
    while (b)
    {
        if (b & 1)
            res = 1LL * res * a % p;
        a = 1LL * a * a % p;
        b >>= 1;
    }
    return res;
}
int C(int n, int m)
{
    return 1LL * fac[n] * inv[m] % mod * inv[n - m] % mod;
}
int A(int n, int m)
{
    return 1LL * fac[n] * inv[n - m] % mod;
}
int main()
{
    scanf("%d", &n);
    fac[0] = 1;
    for (int i = 1; i <= n; i++)
        fac[i] = 1LL * fac[i - 1] * i % mod;
    inv[n] = qpow(fac[n], mod - 2, mod);
    for (int i = n - 1; i >= 0; i--)
        inv[i] = 1LL * inv[i + 1] * (i + 1) % mod;
    f[0] = 1, f[1] = 2;
    for (int i = 2; i <= n; i++)
        f[i] = (2LL * i * f[i - 1] % mod - 1LL * (i - 1) * (i - 1) % mod * f[i - 2] % mod) % mod;
    for (int i = 0; i <= n; i++)
    {
        if (i % 2)
            ans = (ans - 1LL * C(n, i) * A(n, i) % mod * f[n - i] % mod * f[n - i] % mod) % mod;
        else
            ans = (ans + 1LL * C(n, i) * A(n, i) % mod * f[n - i] % mod * f[n - i] % mod) % mod;
    }
    ans = (ans + mod) % mod;
    printf("%d", ans);
    return 0;
}