照看小猫(快速乘/快速幂取模)

问题描述:

链接:https://ac.nowcoder.com/acm/contest/11038/C
来源:牛客网

在一个风和日丽的午后,少佐给薇尔莉特伊芙嘉登安排了一个任务。
任务大致是这样的,接下来的一周,薇尔莉特需要照顾 N 只小猫咪。为了方便管理这N 只猫咪,
薇尔莉特准备给每只猫咪取一个独一无二的名字。取名字要征求猫咪本咪的同意才可以。
但这些猫咪都非常有个性,绝不接受很长的名字。
现给出每只猫咪所能容忍名字的长度上限。
请你为所有猫咪取名字,请问共有多少种不同的方案。
名字仅包含小写英文字母。结果可能非常大,所以请将结果对 77797 取模
如果无方案,请输出 -1
输入描述:
第一行,一个正整数 N 。
第二行, N 个正整数,第 i 个数字 a_ia
i

,表示第 i 只猫咪所能容忍名字的长度上限,数字间以空格隔开。
输出描述:
第一行,仅输出 1 个数,取模后的总方案数 x 。PS:如果无方案,请输出 -1
示例1
输入
复制
1
1
输出
复制
26
说明
猫咪的名字可以为 a-z 间的任意一个字母,所以方案数为 26。
示例2
输入
复制
1
2
输出
复制
702
说明
样例 2 中,猫咪的名字的长度可以为 1,也可以为 2。
长度为 1 时,名字可以为 a-z 中任意一个,方案数 26 种。
长度为 2 时,名字可以为 aa-az, ba-bz ,…,za-zz 中任意一个,方案数 26×26=676种。
所以总方案数为 26+676=702 种。
示例3
输入
复制
5
7 5 8 4 3
输出
复制
9416
备注:
100%的数据,1 ≤ N ≤ 10^4,1 ≤ a_i ≤ 101≤N≤10
4
,1≤a
i

≤10


代码如下:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod=77797;
const ll maxn=1e5+100;
ll a[maxn];
ll quick_pow(ll base,ll pow)//快速幂 
{
    
    
	ll res=1;
	while(pow>0)
	{
    
    
		if(pow&1) res=res*base%mod;
		pow>>=1;
		base=base*base%mod;
	}
	return res;
}
ll muti(ll a,ll b)//快速乘 
{
    
    
	ll res=0;
	while(b>0)
	{
    
    
		if(b&1) res=(res+a)%mod;
		a=(a*2)%mod;
		b>>=1;
	}
	return res;
}
ll sum(ll x,ll n)
{
    
    
	ll ans=0;
	for(int i=1;i<=x;i++){
    
    
		ans+=quick_pow(26,i);
		ans%=mod;
	}
	ans-=(n-1)%mod;
	ans%=mod;
	return ans;
}
int main()
{
    
    
	int n;
	cin>>n;
	ll res=1;
	for(int i=1;i<=n;i++) cin>>a[i];
	sort(a+1,a+1+n);//猫的容忍度从小到大排序 
	for(int i=1;i<=n;i++) a[i]=sum(a[i],i);//计算猫一共能有多少种情况 
	for(int i=1;i<=n;i++) res=muti(res,a[i]);//按顺序将小猫的情况相乘 
	if(res) cout<<res<<endl;
	else cout<<"-1"<<endl; 
	return 0;
}

总结:

综合了快速乘、快速幂

取模方面:
(a+b)%mod=(a%mod+b%mod)%mod;
(a-b)%mod=(a%mod-b%mod)%mod;
(ab)%mod=(a%modb%mod)%mod;

PS:借鉴了其他大佬的代码

猜你喜欢

转载自blog.csdn.net/Lzhzl211/article/details/114650051