问题描述:
链接: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:借鉴了其他大佬的代码