[CSP-S模拟测试]:小W的魔术(数学 or 找规律)
题目传送门(内部题130)
输入格式
第一行一个整数$n$,表示字符串的长度。
第二行一个只包含小写字母的字符串$s$。
输出格式
一行一个整数表示答案对$998244353$取模后的结果。
样例
样例输入:
2
a
样例输出:
625
数据范围与提示
对于$20\%$的测试数据,$n\leqslant 4$。
对于$40\%$的测试数据,$n\leqslant 100$。
对于$60\%$的测试数据,$n\leqslant 10,000,000$。
对于另外$20\%$的测试数据,保证$s$只由$a$组成。
对于$100\%$的测试数据,$2\leqslant n\leqslant 10,1\leqslant |s|\leqslant 1,000,000$。
对于$100\%$的测试数据,$|s|\leqslant n$。
题解
他们跟我说,打表找了个规律……
反正我是没有找出来。
其实答案就是:
$$26^n-26^{(n-s)}-s\times 25\times 26^{(n-s-1)}$$
时间复杂度:$\Theta(\log n)$。
期望得分:$100$分。
实际得分:$100$分。
代码时刻
#include<bits/stdc++.h>
using namespace std;
const int mod=998244353;
long long n,s;
char ch[1000001];
long long ans;
long long qpow(long long x,long long y)
{
if(y<0)return 0;
long long res=1;
while(y)
{
if(y&1)res=res*x%mod;
x=x*x%mod;
y>>=1;
}
return res;
}
int main()
{
scanf("%lld%s",&n,ch+1);
s=strlen(ch+1);
ans=((qpow(26,n)-qpow(26,n-s)+mod)%mod-s*25%mod*qpow(26,n-s-1)%mod+mod)%mod;
printf("%lld",ans);
return 0;
}
rp++