Codeforces-814E An unavoidable detour for home(DP)

传送门：CF-814E

题意：给出n个点每个点的度数为2或3，对于第i个点和第i+1个点，第i个点到第一个点的最短距离不能大于第i+1个点到第一个点的最短距离，问有多少种建图的方法

根据题意有2个性质：

①有边连接的两个点的深度差不超过1，否则bfs中根到某个节点的距离就不是最短路
②深度相同的点，编号一定是连续的
状态转移时，只考虑当前层和上一层的

设f[i][j]为一共有i个点，且有j个点与第i个点在同一层
g[i][j][k]表示当前层有i个点，上一层有j个度为2的点，k个度为3的点
a[i]表示i个点组成项链的方案数
f[i][j]=∑f[i-j][k]*g[j][c1][c2] k∈[1,i-j]

g的状态转移：
①当i=j=k=0时，g[i][j][k]=0
②当i=j=0,k>0时，g[i][j][k]=∑g[i][j][k-1-l]*c[k-1][l]*a[i] l∈[2,k-1]
  表示从前k-1个度数能变成3的点中选l个点出来与最后一个点构成环，这l+1度数都变成3
③当i=0,j>0或k>0时，g[i][j][k]=g[i][j-2][k]*(j-1)+g[i][j][k-1]*k
  前一部分：表示从j-1个度数能变成2的点中选一个出来与最后一个点相连，这2个点的度数都变成2
  后一部分：表示从k个度数能变成2的点中选一个出来与最后一个点相连，这个点变成度数3，最后一个点变成度数2
④当i>0时，g[i][j][k]=g[i-1][j-1][k]*j+g[i-1][j+1][k-1]*k
  前一部分：表示从j个度数能变成2的点中选一个出来与最后一个点相连，这个点变成度数2
  后一部分：表示从k个度数能变成3的点钟选一个出来与最后一个点相连，这个点变成度数3

#include<bits/stdc++.h>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define x first
#define y second
#define add(x,y) (x+=y)%=mod;
using namespace std;
typedef long long LL;
const int MX = 55;
const int inf = 0x3f3f3f3f;
const LL INF = 0x3f3f3f3f3f3f3f3f;
const LL mod = 1e9 + 7;
int d[MX], n;
LL f[MX][MX], g[MX][MX][MX], a[MX], c[MX][MX];
//g[i][j][k]表示当前层有i个点，上一层有j个度为2的点，k个度为3的点
//a[i]=(i-1)!/2
void init() {
    a[0] = a[1] = 0; a[2] = a[3] = 1;
    for (int i = 4; i <= n; i++) a[i] = a[i - 1] * (i - 1) % mod;
    c[0][0] = 1;
    for (int i = 1; i <= n; i++) {
        c[i][0] = 1;
        for (int j = 1; j <= i; j++) c[i][j] = (c[i - 1][j - 1] + c[i - 1][j]) % mod;
    }
    g[0][0][0] = 1;
    for (int j = 0; j <= n; j++)
        for (int k = 0; k <= n - j; k++) {
            if (j == 0 && k > 0) {
                //因为不存在重边，一个环至少有3个点
                for (int l = 2; l < k; l++) add(g[0][j][k], g[0][j][k - 1 - l] * c[k - 1][l] % mod * a[l + 1]);
            } else {
                if (j >= 2) add(g[0][j][k], g[0][j - 2][k] * (j - 1));
                if (k >= 1) add(g[0][j][k], g[0][j][k - 1] * k);
            }
        }
    for (int i = 1; i <= n; i++)
        for (int j = 0; j <= n - i; j++)
            for (int k = 0; k <= n - i - j; k++) {
                if (j >= 1) add(g[i][j][k], g[i - 1][j - 1][k]*j);
                if (k >= 1) add(g[i][j][k], g[i - 1][j + 1][k - 1]*k);
            }
}

int main() {
    //freopen("in.txt", "r", stdin);
    scanf("%d", &n);
    for (int i = 1; i <= n; i++) scanf("%d", &d[i]);
    init();
    f[d[1] + 1][d[1]] = 1;
    //f[i][j]=∑f[i-j][k]*g[j][c1][c2] k∈[1,i-j]
    for (int i = d[1] + 2; i <= n; i++) {
        for (int j = 1; j <= i - d[1] - 1; j++) {//从第三层开始,假设现在为第x层,有j个点
            int c1 = 0, c2 = 0;
            for (int k = 1; k <= i - j; k++) {  //枚举第x-1层有k个点
                if (d[i - j - k + 1] == 2) c1++;
                else c2++;
                add(f[i][j], f[i - j][k]*g[j][c1][c2]);
            }
        }
    }
    LL ans = 0;
    int c1 = 0, c2 = 0;
    for (int j = 1; j < n; j++) {
        if (d[n - j + 1] == 2) c1++;
        else c2++;
        add(ans, f[n][j]*g[0][c1][c2]);
    }
    printf("%I64d\n", ans);
    return 0;
}