結果
問題 | No.2131 Concon Substrings (COuNt Version) |
ユーザー |
👑 |
提出日時 | 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 |
ソースコード
#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(); }