[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++

posted @ 2019-11-09 08:08  HEOI-动动  阅读(236)  评论(0编辑  收藏  举报