Description
有两个长度为n且仅由小写字母组成的字符串S,T,满足S和T恰好有k位不同。问在所有恰好与S有k位不同的字符串中,T按照字典序排在第几位。由于答案可能很大,模10^9+7输出。
Input
第一行两个整数n,k。
第二行一个字符串S。
第三行一个字符串T。
Output
一行一个整数表示答案。
Sample Input
4 1
abcd
bbcd
Sample Output
76
Hint
对于前30%的数据,n<=5.
对于100%的数据,k<=n<=100000.
做法:
我们就从第一位开始统计:
首先字符串的第一位肯定不能大于T的第一位
那么如果S的第一位比T的第一位要小,那么第一位对答案的贡献就是
C(n-1,m-1)25^(m-1)(T[1]-‘a’-1)+C(n-1,m)*25^m然后要m–
如果S的第一位比T的第一位要大,那么第一位对答案的贡献就是C(n-1,m-1)25^(m-1)(T[1]-‘a’)然后要m–
如果S的第一位等于T的第一位,那么对答案的贡献就是C(n-1,m-1)25^(m-1)(T[1]-‘a’)
然后往下一位一样的接着处理就好了。
代码如下:
#include <iostream>
#include <cstdio>
#include <cstring>
#define maxn 1000000007
using namespace std;
char a[100007],b[100007];
long long f[100007],g[100007],l[100007];
long long n,m;
long long ans;
long long pow4(long long a,long long b)
{
long long r=1,base=a;
while(b)
{
if(b&1) r*=base;
r%=maxn;
base*=base;
base%=maxn;
b>>=1;
}
return r;
}
long long C(long long x,long long y)
{
return f[x]*l[x-y]%maxn*l[y]%maxn;
}
int main()
{
scanf("%lld%lld",&n,&m);
cin>>a+1;
cin>>b+1;
g[0]=1; f[0]=1; l[0]=1;
for (long long i=1;i<=n;i++)
{
f[i]=f[i-1]*i%maxn;
g[i]=g[i-1]*25%maxn;
l[i]=pow4(f[i],maxn-2);
}
for (long long i=1;i<=n;i++)
{
if (a[i]<b[i])
{
ans+=(b[i]-'a'-1)*g[m-1]%maxn*C(n-i,m-1)%maxn;
ans%=maxn;
if (n-i>=m) ans+=C(n-i,m)%maxn*g[m]%maxn;
ans%=maxn;
m--;
}
if (a[i]==b[i])
{
ans+=(b[i]-'a')*g[m-1]%maxn*C(n-i,m-1)%maxn;
ans%=maxn;
}
if (a[i]>b[i])
{
ans+=(b[i]-'a')*g[m-1]%maxn*C(n-i,m-1)%maxn;
ans%=maxn;
m--;
}
}
cout<<ans+1;
}