题目描述
给出一个长度为n, 由小写英文字母组成的字符串S, 求在所有由小写英文字母组成且长度为n 且恰好有k 位与S 不同的字符串中,给定字符串T 按照字典序排在第几位。由于答案可能很大,模10^9 + 7 输出。
瞎做
你尝试枚举比T串小的合法字符串最早在第几位和T不同,然后组合数+幂数算一波以这个为前缀的合法字符串数量。
#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fd(i,a,b) for(i=a;i>=b;i--)
using namespace std;
const int maxn=100000+10,mo=1000000007;
typedef long long ll;
int mi[maxn],fac[maxn],inv[maxn];
char a[maxn],b[maxn],ch;
int i,j,k,l,t,n,m,ans;
int quicksortmi(int x,int y){
if (!y) return 1;
int t=quicksortmi(x,y/2);
t=(ll)t*t%mo;
if (y%2) t=(ll)t*x%mo;
return t;
}
int C(int n,int m){
if (n<m) return 0;
return (ll)fac[n]*inv[m]%mo*inv[n-m]%mo;
}
int F(int n,int m){
if (m<0||m>n) return 0;
return (ll)C(n,m)*mi[m]%mo;
}
int main(){
freopen("string.in","r",stdin);freopen("string.out","w",stdout);
scanf("%d%d",&n,&m);
scanf("%s",a+1);
scanf("%s",b+1);
fac[0]=1;
fo(i,1,n) fac[i]=(ll)fac[i-1]*i%mo;
inv[n]=quicksortmi(fac[n],mo-2);
fd(i,n-1,0) inv[i]=(ll)inv[i+1]*(i+1)%mo;
mi[0]=1;
fo(i,1,n) mi[i]=(ll)mi[i-1]*25%mo;
fo(i,1,n){
fo(j,0,25){
ch='a'+j;
if (ch==b[i]) break;
if (ch==a[i]) t=F(n-i,m-l);else t=F(n-i,m-l-1);
(ans+=t)%=mo;
}
if (a[i]!=b[i]) l++;
}
ans++;
(ans+=mo)%=mo;
printf("%d\n",ans);
}