二分图染色（dp 数学）

二分图染色（弱化版）

给定一个完全二分图，图的左右两边的顶点数目相同。我们要给图中的每条边染成红色、蓝色、或者绿色，并使得任意两条红边不共享端点、同时任意两条蓝边也不共享端点。
计算所有满足条件的染色的方案数，并对10^9+7取模。

输入描述:

二分图单边的顶点数目n(n ≤ 10^7)

输出描述:

输出一个整数,即所求的答案。

输入

2

输出

35

二分图转换为棋盘问题

两边不共享端点的问题可以转化为棋盘模型，即在 $n$$\times$$n$ 的棋盘上放红蓝棋子，每个格子至多放一次，同行同列的棋子不能同色，显然，对于每一种颜色我们有 $F_n$ =${\sum }_{i=0}^n$$C_n^i$ $A_n^i$ 种放置方案。

考虑两种颜色同时存在，此时 $F_n$$\times$$F_n$ 中包含这两种颜色同方格的情况，因此我们考虑容斥减掉这部分，最终结果为：${\sum }_{i=0}^n$$(-1{)}_{}^n$$C_n^i$ $A_n^i$$(F_{n-i}{)}_{}^2$。

随后考虑 $F_n$ 从前一个状态 $F_{n-1}$ 的转移，即在格子的第 $n$ 行的 $n$ 个空中，我们可以染或者不染某一个方格，该方案数为 $2n$$\times$$F_{n-1}$ ，但是第 $n$ 行的染色操作可能会与之前 $n-1$ 行中的某个方案冲突，横向 $n-1$ 个格子，纵向 $n-1$ 行，总方案数 $(n-1{)}_{}^2$$\times$$F_{n-2}$。

因此对于$F_n$，有递推公式 $F_n$=$2n$$\times$$F_{n-1}$$-$$(n-1{)}_{}^2$$\times$$F_{n-2}$。

代码

#include <cstdio>
#include <cstring>
#include <string>
#include <iostream>
#include <regex>

using namespace std;

#define ll long long
const int maxn = 1e7 + 10;
const int mod = 1e9 + 7;
ll mul[maxn];
ll inv[maxn];
ll f[maxn];

void init() {
    mul[0] = 1;
    for (int i = 1; i < maxn; i++) {
        mul[i] = (mul[i - 1] * i) % mod;
    }
    inv[0] = inv[1] = 1;
    for (int i = 2; i < maxn; i++) {
        inv[i] = (ll) (mod - mod / i) * inv[mod % i] % mod;
    }
    for (int i = 1; i < maxn; i++) {
        inv[i] = (inv[i - 1] * inv[i]) % mod;
    }
    f[0] = 1;
    f[1] = 2;
    for (int i = 2; i < maxn; i++) {
        f[i] = 2ll * i * f[i - 1] % mod - 1ll * (i - 1) * (i - 1) % mod * f[i - 2] % mod;
        f[i] = (f[i] % mod + mod) % mod;
    }
}

ll C(int n, int m) {
    return mul[n] * inv[m] % mod * inv[n - m] % mod;
}

ll A(int n, int m) {
    return mul[n] * inv[n - m] % mod;
}

void solve(int n) {
    ll ans = 0;
    for (int i = 0; i <= n; i++) {
        ans += ((i & 1) ? -1LL : 1LL) * C(n, i) * A(n, i) % mod * f[n - i] % mod * f[n - i] % mod;
        ans = (ans % mod + mod) % mod;
    }
    cout << ans << endl;
}

int main() {
    init();
    int n;
    cin >> n;
    solve(n);
    return 0;
}