[2018JSCPC] HDU 6279 Circular Coloring dp

给出$n$个黑球，$m$个白球，并且$m,n\le 5000$。然后呈环状排列，相同的颜色看作连续的一段，答案是这些段的长度之积。求问所有可能的排列答案之和是多少。
最终构成环状的形态一定是白球和黑球都分成了$i$段，$i\le min(n,m)$。枚举最终分成了多少段，预处理出白球和黑球分成$i$段的答案。那么$ans=(n+m)\sum _{i=1}^{min(n,m)}{F}_i{G}_i/i$.
接下来要求$F_i,G_i$，考虑$f_{i,j}$表示前$j$个数分成$i$段的答案。$f_{i,j}=\sum _{k=i-1}^{j-1}{f}_{i-1,k}(j-k)=j\sum _{k=i-1}^{j-1}{f}_{i-1,k}-\sum _{k=i-1}^{j-1}k{f}_{i-1,k}$.
维护两个前缀和转移。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int mod=1e9+7; 
const int N=5e3+7;
int f[N][N];  
int inv[N];
int main() {
	inv[1]=1;
	for(int i=2;i<=5000;i++)
		inv[i]=1LL*(mod-mod/i)*inv[mod%i]%mod;  
	f[0][0]=1; 
	for(int i=1;i<=5000;i++) {
		int s1=f[i-1][i-1];
		int s2=1LL*f[i-1][i-1]*(i-1)%mod;
		for(int j=i;j<=5000;j++) {
			f[i][j]=(1LL*s1*j%mod-s2+mod)%mod;
			s1+=f[i-1][j],s1%=mod;
			s2+=(1LL*j*f[i-1][j])%mod,s2%=mod; 
		}
	}
	int n,m;
	while(scanf("%d%d",&n,&m)!=EOF) {
		int lim=min(n,m);
		int ans=0;
		for(int i=1;i<=lim;i++) {
			ans+=1LL*f[i][n]*f[i][m]%mod*inv[i]%mod;
			ans%=mod;
		}
		printf("%d\n",1LL*ans*(n+m)%mod);
	}
	return 0;
}