AtCoder Beginner Contest 171 F Strivore 排列组合+难在去重+快速幂+阶乘逆元+组合数

AtCoder Beginner Contest 171   比赛人数10532  比赛开始后14分钟看到A题,之后,每过一分钟看到一道题

AtCoder Beginner Contest 171  F   Strivore   排列组合+难在去重+快速幂+阶乘逆元+组合数

总目录详见https://blog.csdn.net/mrcrack/article/details/104454762

在线测评地址https://atcoder.jp/contests/abc171/tasks/abc171_f

题目大意:给定字符串s,数字k,可以往字符串s里插入k个小写字母,插入位置不限,问生成新的字符串的数量。

涉及排列组合,怎么选取这一下,比较考验人。

为了验证下面算法正确,还专门针对相应的数据,编写了检验代码,感兴趣的读者,可以阅读AC代码之后的内容。

如下数据模拟如下

2
ab

3851

以下生成过程(aabb,aaab,abbb等不会重复生成)


原始字符长度是2,还需插入2个字符,最后字符串总长度是2+2=2
1.原始字符a占据新字符串串首的位置
a( )( )( )
剩下4个空位,原始字符的后1个字符还需占据1个位置,对应的选择数目是C(3,1)=3
a(b)( )( )
a( )(b)( )
a( )( )(b)
1.1以如下形式为例
a(b)( )( )
剩下的两个空位,可以插入字符,插入的字符都不能与前一个字符雷同,每个位置都有25种形式
a(b)(不能是字母b)(不能是字母b)
对应的组合数量25*25

1.2以如下形式为例
a( )(b)( )
剩下的两个空位,可以插入字符,插入的字符都不能与前一个字符雷同,每个位置都有25种形式
a(不能是字母a)(b)(不能是字母b)
对应的组合数量25*25

1.3以如下形式为例(注:abbb只在此时生成)
a( )( )(b)
剩下的两个空位,可以插入字符,插入的字符都不能与前一个字符雷同,每个位置都有25种形式
a(不能是字母a)(不能是字母a)(b)
对应的组合数量25*25

总的组合数量25*25*3=25^2*C(3,1)=26^0*25^2*C(3,1)

2.原始字符a占据新字符串第二个位置
( )a( )( )
新字符串第二个位置之后只剩下2个空位,原始字符的后2个字符还需占据1个位置,对应的选择数目是C(2,1)=2
( )a(b)( )
( )a( )(b)
2.1以如下形式为例
( )a(b)( )
剩下的a前的1个空位,可以插入字符,插入的字符可以是任意字符,每个位置都有26种形式,
剩下的b后的1个空位,可以插入字符,插入的字符可以是不能与b雷同,每个位置都有25种形式,
(任意字符)a(b)(不能是字母b)
对应的组合数量26*25

2.2以如下形式为例(注:aabb只在此时生成)
( )a( )(b)
剩下的a前的1个空位,可以插入字符,插入的字符可以是任意字符,每个位置都有26种形式,
剩下的a后的1个空位,可以插入字符,插入的字符可以是不能与a雷同,每个位置都有25种形式,
(任意字符)a(不能是字母a)(b)
对应的组合数量26*25

总的组合数量26*25*2=26*25*C(2,1)=26^1*25^1*C(2,1)

3.原始字符a占据新字符串第三个位置
( )( )a( )
新字符串第三个位置之后只剩下1个空位,原始字符的后1个字符还需占据1个位置,对应的选择数目是C(1,1)=1
( )( )a(b)
3.1以如下形式为例(注:aaab只在此时生成)
( )( )a(b)
剩下的a前的2个空位,可以插入字符,插入的字符可以是任意字符,每个位置都有26种形式,
(任意字符)(任意字符)a(b)
对应的组合数量26*26=26^2*C(1,1)=26^2*25^0*C(1,1)

结合上述所有情况,对应的组合数量是26^0*25^2*C(3,1)+26^1*25^1*C(2,1)+26^2*25^0*C(1,1)=3851

根据上述模拟过程,对应的算法如下:

原字符串长度是m,需插入n个字符,在新的字串空间n+m中,

1.先放入第一个原始字符,该原始字符前有i(0<=i<=n)个空位,每个空位均有26种摆法,对应,26^i.

2.此时,总长度是n+m,已占据了i+1个字符,剩下n+m-(i+1)个空位,在剩下的空位中选择m-1个位置给原始字符串中剩下的m-1个字母使用,有C(n+m-i-1,m-1)种选择方式。

3.此时还剩下n+m-(i+1)-(m-1)=n-i个空位,在这剩下每个空位对应着25种选法(26个字母,扣除自左向右,最靠近该位的原字串字母,故剩下25个可选字母)。

综上所述,对应公式如下:

AC代码如下

#include <stdio.h>
#include <string.h>
#define mod 1000000007
#define LL long long
#define maxn 1000010
char s[maxn];
LL fact[maxn<<1],inv[maxn<<1];
int m,n;
LL quick_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%mod;
}
void init(){
	int i;
	fact[0]=1;
	for(i=1;i<=n+m;i++)fact[i]=fact[i-1]*i%mod;//阶乘
	inv[n+m]=quick_pow(fact[n+m],mod-2);
	for(i=n+m-1;i>=0;i--)inv[i]=inv[i+1]*(i+1)%mod;//阶乘的乘法逆元
}
LL C(int n,int m){//组合数计算
	return fact[n]*inv[n-m]%mod*inv[m]%mod;
}
int main(){
	int i;
	LL ans=0;
	scanf("%d%s",&n,s);
	m=strlen(s);
	init();
	for(i=0;i<=n;i++)
		ans=(ans+quick_pow(26,i)*C(n+m-i-1,m-1)%mod*quick_pow(25,n-i)%mod)%mod;
	printf("%lld\n",ans);
	return 0;
}

 

为了验证下面算法正确,还专门针对相应的数据,编写了检验代码

验证代码只针对如下数据。

2
ab

3851

验证代码如下,该验证代码编译运行后,输出3851

#include <cstdio>
#include <set>
using namespace std;
set<int> st;
int main(){
	char a,b,c,d;
	int A,B,C,D,ans;
	for(a='a';a<='z';a++)
		for(b='a';b<='z';b++)
			for(c='a';c<='z';c++)
				for(d='a';d<='z';d++){
					A=a-'a',B=b-'a',C=c-'a',D=d-'a';
					ans=A+B*100+C*10000+D*1000000;
					if(b=='b'&&a=='a')st.insert(ans);
					else if(c=='b'&&(a=='a'||b=='a'))st.insert(ans);
					else if(d=='b'&&(a=='a'||b=='a'||c=='a'))st.insert(ans);
				}
	printf("%d\n",st.size());
	return 0;
}

 

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值