题目大意:
输入n,求满足以下要求的n位数字的个数模1e9+7。
1. 它的数字只包含0, 1, 2, 3,且这四个数字都出现过至少一次。
2. 所有的0都出现在所有的1之前,而所有的2都出现在所有的3之前。
3. 最高位数字不为0。
首先我们可以发现,如果不考虑第三号约束的话,0与1的放置,2与3的放置完全可以独立处理。然后再拼接成一个长串。
子问题
假如一个串长度为 x ,全由0或1组成,0和1至少出现一次,并且所有0在所有1前边,有多少个呢。
答案是显然的:x-1个。 只能从最后一位一位的放1,第一位保持0。
同理换成 2和3组成的串 也一样是x-1个。
解题
那么现在的问题就是把两个拼接在一起了,题目要求总长度为n,那么由两部分组成 01串 和 23串,假设01串的长度为x,那么23串长度是n-x,怎么拼呢。
仔细思考可以发现只能是23串的2做开头,所以剩下的n-1个位置就是一个组合数学问题了,n-1个位置取x个放01串。
问题显然了:答案就是枚举01串或者23串的长度,用 组合数乘以上边说的子问题 计算贡献。
(注:求组合数代码使用的快速幂求逆元,可以提前线性递推出逆元,把复杂度降到O(n))
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const ll MAXN = 1e5+5;
const ll mod = 1e9+7;
ll Pow(ll a,ll b) /// 快速幂求逆元
{
ll ret = 1;
while(b){
if(b&1)ret = (ret * a)%mod;
a = (a*a)%mod;
b>>=1;
}
return ret;
}
int main()
{
ll n ;
cin>>n;
ll ans = 0;
ll k = 1;
for(ll i=1;i<=n-1;i++){
k = (k*i)%mod; //预处理出组合数的 分子 部分
}
for(ll i=2;i<=n-2;i++){ ///枚举 23串长度
ll kk = k;
for(ll j=1;j<=i-1;j++){ /// 逆元处理出 分母 部分
kk = ( kk*Pow(j,mod-2) )%mod;
}
for(ll j=1;j<=n-i;j++){ /// 逆元处理出 分母 部分
kk = ( kk*Pow(j,mod-2) )%mod;
}
kk = (kk * (i-1) )%mod; ///乘以 23 串种类 (子问题)
kk = (kk * (n-i-1) )%mod; ///乘以 01串 种类
ans = (ans + kk)%mod;
}
cout<<ans<<endl;
}