結果

問題 No.2131 Concon Substrings (COuNt Version)
ユーザー 👑 rin204
提出日時 2022-11-25 21:56:22
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 767 bytes
コンパイル時間 1,944 ms
コンパイル使用メモリ 195,876 KB
最終ジャッジ日時 2025-02-09 00:10:56
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 16
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<bits/stdc++.h>
using namespace std;
const long long MOD = 998244353;

long long modpow(long long a, long long b){
	long long ret = 1;
	a %= MOD;
	while(b){
		if(b & 1){
			ret *= a;
			ret %= MOD;
		}
		a *= a;
		a %= MOD;
		b >>= 1;
	}
	return ret;
}

void solve(){
	int n;
	cin >> n;
	vector<long long> dp(3, 0);
	dp[0] = 1;
	long long ans = 0;
	long long times = modpow(26, n);
	long long inv = modpow(26, MOD - 2);
	for(int i = 0; i < n; i++){
		times *= inv;
		times %= MOD;
		ans += dp[2] * times % MOD;
		ans %= MOD;
		vector<long long> ndp(3, 0);
		for(int j = 0; j < 3; j++){
			if(j == 0) ndp[j] = (dp[j] * 25 + dp[2]) % MOD;
			else ndp[j] = (dp[j] * 25 + dp[j - 1]) % MOD;
		}
		swap(dp, ndp);
		}
	cout << ans << "\n";
}

int main(){
	solve();
}
0