Description
定义一个长度为
m
的字符串
1.
a
是
2.存在 1≤i1<i2<...<in≤m ,满足 aj=Sij,j=1,2,...,n ,且至少存在一个 1≤j<i 满足 ij+1−ij>1
现有
k
种字符,问用这些字符构造的字符串中有多少字符串其最长弱子序列长度为
Input
两个整数 k,w(1≤k≤106,2≤w≤109)
Output
输出满足条件的字符串个数,结果模 109+7
Sample Input
2 2
Sample Output
10
Solution
比较直观的可以看出
S
的最长弱子序列必然是字符串
假设串长为 w+L ,显然 L≥1 ,且 L≤k ,否则假设最长弱子序列出现在前缀,那么后面 L 个字符由鸽巢原理必然会出现重复字符,那么将原先的最长弱子序列延长到重复字符出现的第一个位置,然后对应上重复字符出现的第二个位置显然满足弱子序列的定义,与原先弱子序列是最长的矛盾
串长
1
.
选取
L
个不同字符放在前
2
.
选取 L 个不同字符放在前 L 个位置 ALk ,选取 L 个不同字符放在后 L 个位置 ALk ,中间那个位置随便取 k ,但是同样有不合法的情况,即前 L+1 个位置字符两两不同,后 L+1 个位置字符两两不同,即先选出 L+1 个不同的字符放在前 L+1 个位置,再从除中间那个位置所放字符之外的 k−1 个字符中选出 L 个不同的字符,方案数 AL+1k⋅ALk−1 ,故总方案数为 ALk⋅ALk⋅k−AL+1k⋅ALk−1
3
.
选取
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=2000005;
#define mod 1000000007
int fact[maxn],inv[maxn];
void init(int n=2e6)
{
fact[0]=1;
for(int i=1;i<=n;i++)fact[i]=(ll)i*fact[i-1]%mod;
inv[1]=1;
for(int i=2;i<=n;i++)inv[i]=mod-(ll)(mod/i)*inv[mod%i]%mod;
inv[0]=1;
for(int i=1;i<=n;i++)inv[i]=(ll)inv[i-1]*inv[i]%mod;
}
int A(int n,int m)
{
if(m>n||n<0)return 0;
return (ll)fact[n]*inv[n-m]%mod;
}
int Pow(int a,int b)
{
int ans=1;
while(b)
{
if(b&1)ans=(ll)ans*a%mod;
a=(ll)a*a%mod;
b>>=1;
}
return ans;
}
int main()
{
init();
int k,w;
while(~scanf("%d%d",&k,&w))
{
int ans=0;
for(int l=1;l<=k;l++)
{
int res;
if(w-l>=2)
res=((ll)A(k,l)*A(k,l)%mod*k%mod*k%mod-(ll)A(k,l+1)*A(k,l+1)%mod+mod)%mod*Pow(k,w-l-2)%mod;
else if(w-l==1)
res=((ll)k*A(k,l)%mod*A(k,l)%mod-(ll)A(k,l+1)*A(k-1,l)%mod+mod)%mod;
else
res=((ll)A(k,l-w)*A(k-l+w,w)%mod*A(k-l+w,w)%mod-(ll)A(k,l-w+2)*A(k-(l-w+2),w-1)%mod*A(k-(l-w+2),w-1)%mod+mod)%mod;
ans=(ans+res)%mod;
}
printf("%d\n",ans);
}
return 0;
}