結果

問題 No.765 ukuku 2
ユーザー ei1333333
提出日時 2018-10-30 01:53:14
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 249 ms / 3,000 ms
コード長 5,159 bytes
コンパイル時間 2,326 ms
コンパイル使用メモリ 211,528 KB
最終ジャッジ日時 2025-01-06 15:06:12
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 48
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include <bits/stdc++.h>
using namespace std;
using int64 = long long;
using pi = pair< int, int >;
struct SuffixArray {
  vector< int > SA;
  const string s;
  SuffixArray(const string &str) : s(str) {
    SA.resize(s.size());
    iota(begin(SA), end(SA), 0);
    sort(begin(SA), end(SA), [&](int a, int b) {
      return s[a] == s[b] ? a > b : s[a] < s[b];
    });
    vector< int > classes(s.size()), c(s.begin(), s.end()), cnt(s.size());
    for(int len = 1; len < s.size(); len <<= 1) {
      for(int i = 0; i < s.size(); i++) {
        if(i > 0 && c[SA[i - 1]] == c[SA[i]] && SA[i - 1] + len < s.size() && c[SA[i - 1] + len / 2] == c[SA[i] + len / 2]) {
          classes[SA[i]] = classes[SA[i - 1]];
        } else {
          classes[SA[i]] = i;
        }
      }
      iota(begin(cnt), end(cnt), 0);
      copy(begin(SA), end(SA), begin(c));
      for(int i = 0; i < s.size(); i++) {
        int s1 = c[i] - len;
        if(s1 >= 0) SA[cnt[classes[s1]]++] = s1;
      }
      classes.swap(c);
    }
  }
  int operator[](int k) const {
    return SA[k];
  }
  size_t size() const {
    return s.size();
  }
  bool lt_substr(const string &t, int si = 0, int ti = 0) {
    int sn = (int) s.size(), tn = (int) t.size();
    while(si < sn && ti < tn) {
      if(s[si] < t[ti]) return true;
      if(s[si] > t[ti]) return false;
      ++si, ++ti;
    }
    return si >= sn && ti < tn;
  }
  int lower_bound(const string &t) {
    int low = -1, high = (int) SA.size();
    while(high - low > 1) {
      int mid = (low + high) / 2;
      if(lt_substr(t, SA[mid])) low = mid;
      else high = mid;
    }
    return high;
  }
  pair< int, int > lower_upper_bound(string &t) {
    int idx = lower_bound(t);
    int low = idx - 1, high = (int) SA.size();
    t.back()++;
    while(high - low > 1) {
      int mid = (low + high) / 2;
      if(lt_substr(t, SA[mid])) low = mid;
      else high = mid;
    }
    t.back()--;
    return {idx, high};
  }
  void output() {
    for(int i = 0; i < size(); i++) {
      cout << i << ": " << s.substr(SA[i]) << endl;
    }
  }
};
struct LongestCommonPrefixArray {
  const SuffixArray &SA;
  vector< int > LCP, rank;
  LongestCommonPrefixArray(const SuffixArray &SA) : SA(SA), LCP(SA.size()) {
    rank.resize(SA.size());
    for(int i = 0; i < SA.size(); i++) {
      rank[SA[i]] = i;
    }
    for(int i = 0, h = 0; i < SA.size(); i++) {
      if(rank[i] + 1 < SA.size()) {
        for(int j = SA[rank[i] + 1]; max(i, j) + h < SA.size() && SA.s[i + h] == SA.s[j + h]; ++h);
        LCP[rank[i] + 1] = h;
        if(h > 0) --h;
      }
    }
  }
  int operator[](int k) const {
    return LCP[k];
  }
  size_t size() const {
    return LCP.size();
  }
  void output() {
    for(int i = 0; i < size(); i++) {
      cout << i << ": " << LCP[i] << " " << SA.s.substr(SA[i]) << endl;
    }
  }
};
template< typename T >
struct SparseTable {
  vector< vector< T > > st;
  vector< int > lookup;
  SparseTable(const vector< T > &v) {
    int b = 0;
    while((1 << b) <= v.size()) ++b;
    st.assign(b, vector< T >(1 << b));
    for(int i = 0; i < v.size(); i++) {
      st[0][i] = v[i];
    }
    for(int i = 1; i < b; i++) {
      for(int j = 0; j + (1 << i) <= (1 << b); j++) {
        st[i][j] = min(st[i - 1][j], st[i - 1][j + (1 << (i - 1))]);
      }
    }
    lookup.resize(v.size() + 1);
    for(int i = 2; i < lookup.size(); i++) {
      lookup[i] = lookup[i >> 1] + 1;
    }
  }
  inline T rmq(int l, int r) {
    int b = lookup[r - l];
    return min(st[b][l], st[b][r - (1 << b)]);
  }
};
const int INF = 1 << 30;
int main() {
  string S;
  cin >> S;
  string T = S;
  reverse(begin(T), end(T));
  T = S + "$" + T;
  SuffixArray sa(T);
  LongestCommonPrefixArray lcp(sa);
  SparseTable< int > table(lcp.LCP);
  vector< int > latte(S.size()), malta(S.size());
  for(int i = 0; i < T.size(); i++) {
    int pos = sa[i];
    if(pos == S.size()) continue;
    if(pos > S.size()) {
      pos -= S.size() + 1;
      pos = (int) S.size() - pos - 1;
      malta[pos] = i;
    } else {
      latte[pos] = i;
    }
  }
  auto get_lcp = [&](int x, int y) {
    if(x > y) swap(x, y);
    return table.rmq(x + 1, y + 1);
  };
  int ret = 0;
  for(int i = 0; i < S.size(); i++) {
    auto lcp = get_lcp(latte[i], malta[i]);
    if(i - lcp >= 0) {
      auto add = lcp;
      if(i - lcp - 1 >= 0 && i + lcp < S.size()) add += get_lcp(latte[i + lcp], malta[i - lcp - 1]);
      ret = max(ret, add * 2 - 1);
    }
    if(i + lcp < S.size()) {
      auto add = lcp;
      if(i - lcp >= 0 && i + lcp + 1 < S.size()) add += get_lcp(latte[i + lcp + 1], malta[i - lcp]);
      ret = max(ret, add * 2 - 1);
    }
  }
  for(int i = 1; i < S.size(); i++) {
    auto lcp = get_lcp(latte[i], malta[i - 1]);
    if(i - lcp - 1 >= 0) {
      auto add = lcp;
      if(i - lcp - 2 >= 0 && i + lcp < S.size()) add += get_lcp(latte[i + lcp], malta[i - lcp - 2]);
      ret = max(ret, add * 2);
    }
    if(i + lcp < S.size()) {
      auto add = lcp;
      if(i - lcp - 1 >= 0 && i + lcp + 1 < S.size()) add += get_lcp(latte[i + lcp + 1], malta[i - lcp - 1]);
      ret = max(ret, add * 2);
    }
  }
  cout << ret << endl;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0