結果

問題 No.962 LCPs
ユーザー lumc_lumc_
提出日時 2019-12-22 18:01:22
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 17 ms / 2,000 ms
コード長 1,712 bytes
コンパイル時間 1,143 ms
コンパイル使用メモリ 110,040 KB
実行使用メモリ 9,644 KB
最終ジャッジ日時 2024-09-14 13:27:29
合計ジャッジ時間 3,346 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 4 ms
7,824 KB
testcase_01 AC 4 ms
7,860 KB
testcase_02 AC 5 ms
7,800 KB
testcase_03 AC 4 ms
7,764 KB
testcase_04 AC 5 ms
7,828 KB
testcase_05 AC 4 ms
7,796 KB
testcase_06 AC 5 ms
7,768 KB
testcase_07 AC 5 ms
7,740 KB
testcase_08 AC 4 ms
7,900 KB
testcase_09 AC 5 ms
7,768 KB
testcase_10 AC 4 ms
7,920 KB
testcase_11 AC 4 ms
7,872 KB
testcase_12 AC 5 ms
7,972 KB
testcase_13 AC 5 ms
7,872 KB
testcase_14 AC 4 ms
7,804 KB
testcase_15 AC 5 ms
7,864 KB
testcase_16 AC 4 ms
7,740 KB
testcase_17 AC 17 ms
9,644 KB
testcase_18 AC 12 ms
8,756 KB
testcase_19 AC 12 ms
8,880 KB
testcase_20 AC 9 ms
8,624 KB
testcase_21 AC 10 ms
8,496 KB
testcase_22 AC 8 ms
8,308 KB
testcase_23 AC 9 ms
8,472 KB
testcase_24 AC 8 ms
8,280 KB
testcase_25 AC 7 ms
8,188 KB
testcase_26 AC 6 ms
8,212 KB
testcase_27 AC 9 ms
8,540 KB
testcase_28 AC 9 ms
8,148 KB
testcase_29 AC 7 ms
8,152 KB
testcase_30 AC 8 ms
8,316 KB
testcase_31 AC 10 ms
8,640 KB
testcase_32 AC 7 ms
8,056 KB
testcase_33 AC 14 ms
8,752 KB
testcase_34 AC 7 ms
8,272 KB
testcase_35 AC 7 ms
8,248 KB
testcase_36 AC 8 ms
8,408 KB
testcase_37 AC 8 ms
8,096 KB
testcase_38 AC 8 ms
8,060 KB
testcase_39 AC 8 ms
8,112 KB
testcase_40 AC 8 ms
7,988 KB
testcase_41 AC 9 ms
7,992 KB
testcase_42 AC 8 ms
8,092 KB
testcase_43 AC 8 ms
7,960 KB
testcase_44 AC 8 ms
8,140 KB
testcase_45 AC 8 ms
8,104 KB
testcase_46 AC 9 ms
7,988 KB
testcase_47 AC 6 ms
7,984 KB
testcase_48 AC 5 ms
8,032 KB
testcase_49 AC 6 ms
8,000 KB
testcase_50 AC 6 ms
8,204 KB
testcase_51 AC 6 ms
8,140 KB
testcase_52 AC 4 ms
8,024 KB
testcase_53 AC 5 ms
7,952 KB
testcase_54 AC 5 ms
7,992 KB
testcase_55 AC 6 ms
8,012 KB
testcase_56 AC 6 ms
8,272 KB
testcase_57 AC 5 ms
7,916 KB
testcase_58 AC 6 ms
7,916 KB
testcase_59 AC 5 ms
7,892 KB
testcase_60 AC 6 ms
7,896 KB
testcase_61 AC 6 ms
7,904 KB
testcase_62 AC 5 ms
7,940 KB
testcase_63 AC 13 ms
9,008 KB
testcase_64 AC 6 ms
8,112 KB
testcase_65 AC 14 ms
9,016 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

// O(S^1.5)
// includes {{{
#include<iostream>
#include<iomanip>
#include<algorithm>
#include<vector>
#include<stack>
#include<queue>
#include<map>
#include<set>
#include<tuple>
#include<cmath>
#include<random>
#include<cassert>
#include<bitset>
#include<cstdlib>
// #include<deque>
// #include<multiset>
// #include<cstring>
// #include<bits/stdc++.h>
// }}}
using namespace std;
using ll = long long;

int main() {
  std::ios::sync_with_stdio(false), std::cin.tie(0);
  int n;
  cin >> n;
  string t;
  cin >> t;
  const int S = 2e5;
  assert(1 <= t.size() && t.size() <= S);
  for(auto c : t) assert('a' <= c && c <= 'z');
  vector<vector<int>> cnt(S + 1);
  ll ans = t.size();
  assert(ans <= S);

  for(int i = 1; i < n; i++) {
    string s;
    cin >> s;
    for(auto c : s) assert('a' <= c && c <= 'z');
    assert(1 <= s.size() && s.size() <= S);
    int f = 0;
    while(f < s.size() && f < t.size() && s[f] == t[f]) f++;
    t = s;
    cnt.at(f).push_back(i);
    ans += s.size();
    assert(ans <= S);
  }
  vector<int> used(n+1);
  used[0] = used[n] = 1;
  for(int i = 0; i <= S; i++) if(cnt[i].size()) {
    int now = 0;
    int last = 0, last2;
    int lasti = -1;
    ll x, x2;
    for(int k = 1; k <= n; k++) {
      if(now < cnt[i].size() && cnt[i].at(now) == k) {
        used[k] = 1;
        x2 = k - last;
        last2 = last;
      }
      if(lasti >= 0 && used[k]) {
        ll y = k - lasti;
        ans += ((x * x * y + x * y * y) / 2 + x * y) * i;
        lasti = -1;
      }
      if(used[k]) last = k;
      if(now < cnt[i].size() && cnt[i].at(now) == k) {
        now++;
        x = x2;
        last = last2;
        lasti = k;
      }
    }
  }
  cout << ans << endl;
  return 0;
}
0