「日常训练」The Intriguing Obsession(CFR439D2C)

题意（CodeForces 869C）

三堆点，每堆一种颜色；连接的要求是同色不能相邻或距离必须至少3。问对整个图有几种连接方法，对一个数取模。

解析

要求很重要：同色不能相邻很容易理解，但是>=3比较难理解。比较常见的是R->G->B->R，这样能看出来一个重要的结论：对单个节点只能连接某个颜色至多一个点。
那么我们在整个图加入一个新点的时候，只需要考虑它和单个颜色点群的问题（其他颜色的可以同理），也就是这两个颜色间的问题。
设$dp[i][j]$为第一个颜色有i个、第二个颜色有j个的情况下的方法数，那么给第一个颜色+1，
这个新点要么与另一个颜色点群不连接，那么相当于无事发生过；要么与其中的一个点连接。在只有这两个颜色的情况下，这个点只能跟这个新点走到黑了。那么，能够得到转移方程了：

$$dp[i][j]=dp[i-1][j]+dp[i-1][j-1] \times j , dp[i][0]=dp[0][i]=1$$
这是两色的，而两两之间的颜色是互不影响的，根据乘法原理，将三对的方案数相乘即可得解。注意到题目的n不超过5000，可以直接 $O(n^2)$算法走人。

---

代码

#include <bits/stdc++.h>
#define PB push_back
#define MP make_pair
#define fi first
#define se second
#define ZERO(X) memset((X),0,sizeof(X))
#define ALL(X) X.begin(),X.end()
#define F0(X,Y) for(int (X)=0;(X)!=(Y);++(X))
#define F1(X,Y) for(int (X)=1;(X)<=(Y);++(X))
using namespace std;
typedef pair<int,int> PI;
typedef pair<pair<int,int>, int> PII;
typedef pair<pair<pair<int,int>, int>, int> PIII; 
typedef long long ll;
typedef unsigned long long ull;

ll dp[5005][5005];
const ll mod=998244353;
int main()
{
    int a,b,c; cin>>a>>b>>c;
    int maxn=max(a,max(b,c));
    ZERO(dp);
    dp[1][0]=dp[0][1]=dp[0][0]=1;
    for(int i=1;i<=maxn;++i) dp[0][i]=dp[i][0]=1;
    for(int i=1;i<=maxn;++i)
        for(int j=1;j<=maxn;++j)
            dp[i][j]=((dp[i-1][j]%mod)+((dp[i-1][j-1]%mod)*j)%mod)%mod;
    cout<<(((dp[a][b]*dp[b][c])%mod)*dp[a][c])%mod<<endl;
    return 0;
}