因为加法和乘法的次数交换也不影响结果,所以我们先对其进行一个排序方便我们后续的处理,假设我们有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;
}