結果
問題 | No.1195 数え上げを愛したい(文字列編) |
ユーザー | ianCK |
提出日時 | 2020-09-25 20:34:47 |
言語 | C++11 (gcc 11.4.0) |
結果 |
AC
|
実行時間 | 2,398 ms / 3,000 ms |
コード長 | 1,946 bytes |
コンパイル時間 | 569 ms |
コンパイル使用メモリ | 64,812 KB |
実行使用メモリ | 233,100 KB |
最終ジャッジ日時 | 2023-09-10 14:45:46 |
合計ジャッジ時間 | 67,920 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge12 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2,367 ms
232,804 KB |
testcase_01 | AC | 2,372 ms
232,808 KB |
testcase_02 | AC | 2,375 ms
232,928 KB |
testcase_03 | AC | 2,347 ms
232,832 KB |
testcase_04 | AC | 2,348 ms
232,904 KB |
testcase_05 | AC | 2,349 ms
232,888 KB |
testcase_06 | AC | 2,391 ms
232,320 KB |
testcase_07 | AC | 2,352 ms
232,444 KB |
testcase_08 | AC | 2,321 ms
232,760 KB |
testcase_09 | AC | 2,375 ms
233,080 KB |
testcase_10 | AC | 2,339 ms
233,016 KB |
testcase_11 | AC | 2,369 ms
232,808 KB |
testcase_12 | AC | 2,386 ms
232,856 KB |
testcase_13 | AC | 2,354 ms
232,976 KB |
testcase_14 | AC | 2,322 ms
232,992 KB |
testcase_15 | AC | 2,348 ms
232,868 KB |
testcase_16 | AC | 2,331 ms
232,856 KB |
testcase_17 | AC | 2,331 ms
232,832 KB |
testcase_18 | AC | 2,366 ms
232,912 KB |
testcase_19 | AC | 2,347 ms
233,100 KB |
testcase_20 | AC | 2,391 ms
233,000 KB |
testcase_21 | AC | 2,398 ms
232,788 KB |
testcase_22 | AC | 2,352 ms
233,064 KB |
testcase_23 | AC | 2,390 ms
232,612 KB |
testcase_24 | AC | 2,353 ms
232,592 KB |
testcase_25 | AC | 2,354 ms
232,540 KB |
ソースコード
#include <iostream> #include <vector> using namespace std; #define PB push_back typedef long long int ll; constexpr int kN = 1 << 20, kMod = 998244353; ll Pow(ll a, ll b) { ll ans = 1; while (b) { if (b & 1) ans = ans * a % kMod; a = a * a % kMod; b >>= 1; } return ans; } ll Rev(ll n) {return Pow(n, kMod - 2);} void Ntt(vector<ll> &v, bool on, int size); int cnt[26]; ll f[kN], inf[kN]; vector<ll> v[26]; void pre() { f[0] = f[1] = inf[0] = inf[1] = 1; for (int i = 2; i < kN; i++) f[i] = f[i - 1] * i % kMod; inf[kN - 1] = Rev(f[kN - 1]); for (int i = kN - 1; i > 2; i--) inf[i - 1] = inf[i] * i % kMod; for (int i = 0; i < 26; i++) v[i].resize(kN, 0); return ; } int main() { string s; ll ans = kMod - 1; cin >> s; for (char i : s) cnt[i - 'a']++; pre(); for (int i = 0; i < 26; i++) for (int j = 0; j <= cnt[i]; j++) v[i][j] = inf[j]; for (int i = 0; i < 26; i++) Ntt(v[i], false, kN); for (int i = 1; i < 26; i++) for (int j = 0; j < kN; j++) v[0][j] = v[0][j] * v[i][j] % kMod; Ntt(v[0], true, kN); for (int i = 0; i < kN; i++) ans += v[0][i] * f[i] % kMod; cout << ans % kMod << '\n'; } void Ntt( vector<long long int> &v, bool on, int size) { long long int wn, u, t, w, inv; for (int i = 1, j = size >> 1, k; i < (size - 1); i++) { if (i < j) swap(v[i], v[j]); k = size >> 1; while (j & k) { j ^= k; k >>= 1; } j |= k; } for (int i = 2; i <= size; i <<= 1) { wn = on ? Pow(3, (kMod - 1) / i) : Pow(3, kMod - 1 - (kMod - 1) / i); for (int j = 0; j < size; j += i) { w = 1; for (int k = j; k < j + (i >> 1); k++) { u = v[k]; t = (w * v[k + (i >> 1)]) % kMod; v[k] = (u + t) % kMod; v[k + (i >> 1)] = (u - t + kMod) % kMod; w = (w * wn) % kMod; } } } if (on) { inv = Rev(size); for (int i = 0; i < size; i++) v[i] = (v[i] * inv) % kMod; } return; }