HDU7095 - Add or Multiply 1 (百度之星复赛）

因为加法和乘法的次数交换也不影响结果，所以我们先对其进行一个排序方便我们后续的处理，假设我们有n个加号，m个乘号。(n<=m)交换也同理

if(n>m) swap(n,m);

之后我们就要考虑情况的种类了，首先假设我们把n个加号可以分成1~n段（也就是把n个“不同的”加号分成k个集合，1 <= k <= n），然后我们对这k个集合进行全排列答案也就是A(k,k)，这k段可以把m个乘号分成k-1，k，k+1段（需要注意的是如果n=m，则当k=n时，最多分成n段）然后对这k-1，k，k+1段进行全排列，这里同样需要注意的是如果把乘号也分成k段时，需要乘2。
如图所示的是 k=2 时，可以分成以下情况。

我们设F[i][j]表示把i个数分成j个集合，具体见 [集合划分问题（贝尔数）]，我觉得这篇博客写的挺好的。

long long S(ll n,ll m){
	if(F[n][m]) return F[n][m];
	if(n==0||m==0) return 0;
    if(m==1 || m==n)
        F[n][m] = 1;
    else{
    	F[n][m] = S(n-1,m-1)%mod + ((S(n-1,m)%mod)*m)%mod;
	}
    return F[n][m]%mod;
}
 

设n个元素的集合可以划分为F(n,m)个不同的由m个非空子集组成的集合。
考虑3个元素的集合，可划分为
① 1个子集的集合：{{1，2，3}}
② 2个子集的集合：{{1，2}，{3}}，{{1，3}，{2}}，{{2，3}，{1}}
③ 3个子集的集合：{{1}，{2}，{3}}
∴F(3,1)=1;F(3,2)=3;F(3,3)=1;
如果要求F(4,2)该怎么办呢？
A.往①里添一个元素{4}，得到{{1，2，3}，{4}}
B.往②里的任意一个子集添一个4，得到
{{1，2，4}，{3}}，{{1，2}，{3，4}}，
{{1，3，4}，{2}}，{{1，3}，{2，4}}，
{{2，3，4}，{1}}，{{2，3}，{1，4}}
∴F(4,2)=F(3,1)+2F(3,2)＝1+23＝7
推广，得F(n,m) = F(n-1,m-1) + m*F(n-1,m)

ans += mul(F[n][k],A[k],F[m][k-1],A[k-1]);//用k个加号把乘号分成k-1段
ans += 2*mul(F[n][k],A[k],F[m][k],A[k]);//用k个加号把乘号分成k段（注意有两种情况）
ans += mul(F[n][k],A[k],F[m][k+1],A[k+1]);//用k个加号把乘号分成k+1段

最后再加一个特判n == m时

if(n == m){//如果k=n && n=m的情况下只能把乘号分成 k-1 和 k 段
	ans += 2*A[n]*A[n])%mod + mul(F[n][n],A[n],F[m][n-1],A[n-1];
	ans %= mod;
}
else{//否则还是可以分成三部分进行处理
	ans += mul(F[n][n],A[n],F[m][n-1],A[n-1]);
	ans += 2*mul(F[n][n],A[n],F[m][n],A[n]);
	ans += mul(F[n][n],A[n],F[m][n+1],A[n+1]);
	ans %= mod;			
}

为了取模的时候方便，所以自己写了一个乘法函数，方便操作

long long mul(ll a,ll b,ll c,ll d){
	ll ans = 1;
	ans = a*b%mod;
	ans = ans*c%mod;
	ans = ans*d%mod;
	return ans;
}

完整代码如下：

#include<iostream>
using namespace std;
typedef long long ll;
ll F[3050][3050]={0};
long long A[3030];
const long long mod = 1e9+7;
long long S(ll n,ll m){
	if(F[n][m]) return F[n][m];
	if(n==0||m==0) return 0;
    if(m==1 || m==n)
        F[n][m] = 1;
    else{
    	F[n][m] = S(n-1,m-1)%mod + ((S(n-1,m)%mod)*m)%mod;
	}
    return F[n][m]%mod;
}
 
void init(){
    ll n;
    n = 3010;
    for(int i=1; i<=n; i++){
    	for(int j=1;j<=i;j++){
    		F[i][j] = S(i,j);
		}
	}
	A[0] = 1;
	for(int i=1;i<n;i++){
		A[i] = (A[i-1]*i)%mod;	
	}
}
long long mul(ll a,ll b,ll c,ll d){
	ll ans = 1;
	ans = a*b%mod;
	ans = ans*c%mod;
	ans = ans*d%mod;
	return ans;
}
int main(){
	int t;
	cin >> t;
	init();
	while(t--){
		long long n,m;
		scanf("%lld%lld",&n,&m);
		if(n>m) swap(n,m);
		ll ans = 0;
		for(int i=1;i<n;i++){
			ans += mul(F[n][i],A[i],F[m][i-1],A[i-1]);
			ans += 2*mul(F[n][i],A[i],F[m][i],A[i]);
			ans += mul(F[n][i],A[i],F[m][i+1],A[i+1]);
			ans %= mod;
		}
		if(n == m){
			ans = ans + (2*A[n]*A[n])%mod + mul(F[n][n],A[n],F[m][n-1],A[n-1]);
			ans %= mod;
		}
		else{
			ans += mul(F[n][n],A[n],F[m][n-1],A[n-1]);
			ans += 2*mul(F[n][n],A[n],F[m][n],A[n]);
			ans += mul(F[n][n],A[n],F[m][n+1],A[n+1]);
			ans %= mod;			
		}
		printf("%lld\n",ans);
	}
	return 0;
}