CodeForces 135 E.Weak Subsequence（组合数学）

Description

定义一个长度为$m$的字符串$S$的长度为$n$的弱子序列$a$需要满足以下两个条件：

1.$a$是$S$的子串

2.存在$1\le i_1< i_2<...<i_n\le m$，满足$a_j=S_{i_j},j=1,2,...,n$，且至少存在一个$1\le j<i$满足$i_{j+1}-i_j>1$

现有$k$种字符，问用这些字符构造的字符串中有多少字符串其最长弱子序列长度为$w$

Input

两个整数$k,w(1\le k\le 10^6,2\le w\le 10^9)$

Output

输出满足条件的字符串个数，结果模$10^9+7$

Sample Input

2 2

Sample Output

10

Solution

比较直观的可以看出$S$的最长弱子序列必然是字符串$S$的前缀或者后缀，否则必然不是最长(把前面或者后面的补上即可更长)

假设串长为$w+L$，显然$L\ge 1$，且$L\le k$，否则假设最长弱子序列出现在前缀，那么后面$L$个字符由鸽巢原理必然会出现重复字符，那么将原先的最长弱子序列延长到重复字符出现的第一个位置，然后对应上重复字符出现的第二个位置显然满足弱子序列的定义，与原先弱子序列是最长的矛盾

串长$w+L$固定后，由上面的证明知该字符串前$L$个字符两两不同，且后$L$个字符两两不同，且注意到如果最长弱子序列在前缀，那么在后$L$个字符中必然存在一个字符和第$w$个字符相同（如果不存在，那么第$w$个字符只能对应自己，那么其前面的字符也都只能对应自己，与弱子序列定义矛盾），同理，如果最长弱子序列在后缀，那么在前$L$个字符中必然存在一个字符与第$L+1$个字符相同，知道这三个性质后，我们分类讨论三种情况下的答案

$1$.$w-L\ge 2$，序列分五块，前$L$个，第$L+1$个，中间$w-L-2$个，第$w$个，后$L$个

选取$L$个不同字符放在前$L$个位置$A_k^L$，选取$L$个不同字符放在后$L$个位置$A_k^L$，再选取两个字符放在第$L+1$和第$w+L$个位置$k^2$，但是有不合法的情况，即第$L+1$个字符和前$L$个字符均不相同且第$w$个字符与后$L$个字符均不相同，即前$L+1$个位置字符两两不同，后$L+1$个位置字符两两不同，方案数为$A_k^{L+1}\cdot A_k^{L+1}$，中间$w-L-2$个字符随便取，故总方案数为$k^{w-L-2}\cdot(A_k^L\cdot A_k^L\cdot k^2-A_k^{L+1}\cdot A_k^{L+1})$

$2$.$w-L=1$，序列分三块，前$L$个，中间一个，后$L$个

选取$L​$个不同字符放在前$L​$个位置$A_k^L​$，选取$L​$个不同字符放在后$L​$个位置$A_k^L​$，中间那个位置随便取$k​$，但是同样有不合法的情况，即前$L+1​$个位置字符两两不同，后$L+1​$个位置字符两两不同，即先选出$L+1​$个不同的字符放在前$L+1​$个位置，再从除中间那个位置所放字符之外的$k-1​$个字符中选出$L​$个不同的字符，方案数$A_k^{L+1}\cdot A_{k-1}^L​$，故总方案数为$A_k^L\cdot A_k^L\cdot k-A_k^{L+1}\cdot A_{k-1}^L​$

$3$.$w\le L$，序列分成三块，前$w$个，中间$L-w$个，后$w$个

选取$L-w$个不同的字符放在中间$A_k^{L-w}$，两边分别从剩下的$k-(L-w)$个字符中选$w$个不同的字符$A_{k-(L-w)}^w\cdot A_{k-(L-w)}^w$，不合法的情况，选取$L-w+2$个不同的字符放中间$L-w$个位置以及第$L+1$个位置和第$w$个位置$A_k^{L-w+2}$，再从剩下的$k-(L-w+2)$个字符中分别选取$w-1$个字符放在两边$A_{k-(L-w+2)}^{w-1}\cdot A_{k-(L-w+2)}^{w-1}$，故总方案数为$A_k^{L-w}\cdot A_{k-(L-w)}^w\cdot A_{k-(L-w)}^w-A_k^{L-w+2}\cdot A_{k-(L-w+2)}^{w-1}\cdot A_{k-(L-w+2)}^{w-1}$

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;
}