結果
問題 |
No.962 LCPs
|
ユーザー |
|
提出日時 | 2020-08-04 13:40:20 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 26 ms / 2,000 ms |
コード長 | 809 bytes |
コンパイル時間 | 1,213 ms |
コンパイル使用メモリ | 86,952 KB |
最終ジャッジ日時 | 2025-01-12 14:15:06 |
ジャッジサーバーID (参考情報) |
judge2 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 64 |
ソースコード
#include <iostream> #include <vector> #include <string> #include <functional> using lint = long long; void solve() { int n; std::cin >> n; std::vector<std::string> ss(n); for (auto& s : ss) { std::cin >> s; s.push_back('$'); } lint ans = 0; std::function<void(int, int, int)> dfs = [&](int l, int r, int d) -> void { while (l < r && ss[l][d] == '$') ++l; if (l == r) return; int i = l; while (i < r && ss[i][d] == ss[l][d]) ++i; lint len = i - l; ans += len * (len + 1) * (len + 2) / 6; dfs(l, i, d + 1); dfs(i, r, d); }; dfs(0, n, 0); std::cout << ans << "\n"; } int main() { std::cin.tie(nullptr); std::ios::sync_with_stdio(false); solve(); return 0; }