結果
問題 |
No.962 LCPs
|
ユーザー |
|
提出日時 | 2020-01-28 17:44:43 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 218 ms / 2,000 ms |
コード長 | 2,243 bytes |
コンパイル時間 | 1,806 ms |
コンパイル使用メモリ | 180,688 KB |
実行使用メモリ | 31,488 KB |
最終ジャッジ日時 | 2024-09-15 16:18:54 |
合計ジャッジ時間 | 6,565 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 64 |
ソースコード
#define _USE_MATH_DEFINES #include "bits/stdc++.h" using namespace std; #define FOR(i,j,k) for(int (i)=(j);(i)<(int)(k);++(i)) #define rep(i,j) FOR(i,0,j) #define each(x,y) for(auto &(x):(y)) #define mp make_pair #define MT make_tuple #define all(x) (x).begin(),(x).end() #define debug(x) cout<<#x<<": "<<(x)<<endl #define smax(x,y) (x)=max((x),(y)) #define smin(x,y) (x)=min((x),(y)) #define MEM(x,y) memset((x),(y),sizeof (x)) #define sz(x) (int)(x).size() #define RT return using ll = long long; using pii = pair<int, int>; using vi = vector<int>; using vll = vector<ll>; struct RollingHash { static const int C = 31, C2 = 29, P = 1000000007, P2 = 1000000009; int _n; vector<long long> pw, pw2, hs, hs2; RollingHash() {} RollingHash(const string &st) :_n((int)st.size()), pw(_n + 1), pw2(_n + 1), hs(_n + 1), hs2(_n + 1) { pw[0] = pw2[0] = 1; rep(i, _n) { pw[i + 1] = (pw[i] * C) % P; pw2[i + 1] = (pw2[i] * C2) % P2; hs[i + 1] = (C*hs[i] + st[i]) % P; hs2[i + 1] = (C2*hs2[i] + st[i]) % P2; } } // [l, r) pii getHash(int l, int r) { int res, res2; res = (hs[r] - hs[l] * pw[r - l]) % P; res2 = (hs2[r] - hs2[l] * pw2[r - l]) % P2; if (res < 0)res += P; if (res2 < 0)res2 += P2; return mp(res, res2); } }; void solve() { int N; cin >> N; map<pii, vi> indices; rep(i, N) { string S; cin >> S; RollingHash rh(S); int M = sz(S); for (int len = 1; len <= M; ++len) { indices[rh.getHash(0, len)].push_back(i); } } ll ans = 0; each(idsids, indices) { auto &ids = idsids.second; int M = sz(ids); for (int i = 0; i < M;) { int j = i + 1; while (j < M && ids[j - 1] + 1 == ids[j]) { j++; } // [i, j)は連続している ll m = j - i; ans += m * (m + 1)*(m + 1) / 2 - m * (m + 1)*(2 * m + 1) / 6; i = j; } } cout << ans << endl; } int main() { ios::sync_with_stdio(false); cin.tie(0); cout << fixed << setprecision(15); solve(); return 0; }