結果

問題 No.1018 suffixsuffixsuffix
ユーザー beetbeet
提出日時 2020-03-24 10:00:35
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 110 ms / 2,000 ms
コード長 3,399 bytes
コンパイル時間 2,485 ms
コンパイル使用メモリ 214,308 KB
実行使用メモリ 9,524 KB
最終ジャッジ日時 2024-07-03 00:21:05
合計ジャッジ時間 6,985 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 2 ms
6,944 KB
testcase_04 AC 2 ms
6,944 KB
testcase_05 AC 2 ms
6,940 KB
testcase_06 AC 2 ms
6,940 KB
testcase_07 AC 2 ms
6,940 KB
testcase_08 AC 2 ms
6,940 KB
testcase_09 AC 43 ms
6,940 KB
testcase_10 AC 46 ms
6,940 KB
testcase_11 AC 51 ms
6,996 KB
testcase_12 AC 48 ms
6,940 KB
testcase_13 AC 48 ms
6,940 KB
testcase_14 AC 86 ms
8,952 KB
testcase_15 AC 91 ms
9,076 KB
testcase_16 AC 100 ms
9,524 KB
testcase_17 AC 96 ms
8,832 KB
testcase_18 AC 99 ms
9,064 KB
testcase_19 AC 21 ms
6,940 KB
testcase_20 AC 21 ms
6,940 KB
testcase_21 AC 21 ms
6,944 KB
testcase_22 AC 21 ms
6,940 KB
testcase_23 AC 21 ms
6,940 KB
testcase_24 AC 91 ms
8,916 KB
testcase_25 AC 103 ms
9,408 KB
testcase_26 AC 97 ms
8,768 KB
testcase_27 AC 103 ms
8,808 KB
testcase_28 AC 101 ms
8,900 KB
testcase_29 AC 27 ms
6,940 KB
testcase_30 AC 28 ms
6,944 KB
testcase_31 AC 26 ms
6,944 KB
testcase_32 AC 27 ms
6,944 KB
testcase_33 AC 26 ms
6,944 KB
testcase_34 AC 2 ms
6,940 KB
testcase_35 AC 20 ms
6,940 KB
testcase_36 AC 25 ms
6,944 KB
testcase_37 AC 110 ms
9,392 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
template<typename T1,typename T2> inline void chmin(T1 &a,T2 b){if(a>b) a=b;}
template<typename T1,typename T2> inline void chmax(T1 &a,T2 b){if(a<b) a=b;}
using Int = long long;
const char newl = '\n';

// longest common prefix of s and s[i:n]
template<typename T>
vector<int> zalgorithm(vector<T> vs){
  int n=vs.size();
  vector<int> as(n+1,0);
  as[0]=n;
  int i=1,j=0;
  while(i<n){
    while(i+j<n&&vs[j]==vs[i+j]) j++;
    as[i]=j;
    if(j==0){
      i++;
      continue;
    }
    int k=1;
    while(i+k<n&&k+as[k]<j) as[i+k]=as[k],k++;
    i+=k;
    j-=k;
  }
  return as;
}
vector<int> zalgorithm(string s){
  return zalgorithm(vector<char>(s.begin(),s.end()));
}

struct SuffixArray{
  string s;
  vector<int> sa,rev;

  SuffixArray(){}
  SuffixArray(const string &S):s(S){
    int n=s.size();
    s.push_back('$');
    sa.resize(n+1);
    iota(sa.begin(),sa.end(),0);
    sort(sa.begin(),sa.end(),
         [&](int a,int b){
           if(s[a]==s[b]) return a>b;
           return s[a]<s[b];
         });
    vector<int> cs(n+1,0),rs(n+1),cnt(n+1);
    for(int i=0;i<=n;i++) rs[i]=s[i];
    for(int len=1;len<=n;len*=2){
      for(int i=0;i<=n;i++){
        cs[sa[i]]=i;
        if(i>0 &&
           rs[sa[i-1]]==rs[sa[i]] &&
           sa[i-1]+len<=n &&
           rs[sa[i-1]+len/2]==rs[sa[i]+len/2]) cs[sa[i]]=cs[sa[i-1]];
      }
      iota(cnt.begin(),cnt.end(),0);
      copy(sa.begin(),sa.end(),rs.begin());
      for(int i=0;i<=n;i++){
        int s1=rs[i]-len;
        if(s1>=0) sa[cnt[cs[s1]]++]=s1;
      }
      cs.swap(rs);
    }
    rev.resize(n+1);
    for(int i=0;i<=n;i++) rev[sa[i]]=i;
  }
  int operator[](int i) const{return sa[i];}

  bool lt_substr(string &t,int si,int ti){
    int sn=s.size(),tn=t.size();
    while(si<sn&&ti<tn){
      if(s[si]<t[ti]) return 1;
      if(s[si]>t[ti]) return 0;
      si++;ti++;
    }
    return si==sn&&ti<tn;
  }

  int lower_bound(string& t){
    int l=0,r=s.size();
    while(l+1<r){
      int m=(l+r)>>1;
      if(lt_substr(t,sa[m],0)) l=m;
      else r=m;
    }
    return r;
  }

  int upper_bound(string& t){
    t.back()++;
    int res=lower_bound(t);
    t.back()--;
    return res;
  }

  // O(|T|*log|S|)
  int count(string& T){
    return upper_bound(T)-lower_bound(T);
  }
};

//INSERT ABOVE HERE
signed main(){
  cin.tie(0);
  ios::sync_with_stdio(0);

  using ll = long long;
  ll n,m,q;
  cin>>n>>m>>q;

  string s;
  cin>>s;

  vector<ll> ks(q);
  for(int i=0;i<q;i++) cin>>ks[i];

  auto zs=zalgorithm(s+s);
  for(int i=1;i<=n;i++){
    if(n%i) continue;
    if(zs[i]>=n){
      m*=n/i;
      s=s.substr(0,i);
      break;
    }
  }
  n=s.size();

  if(n==1){
    for(int i=0;i<q;i++){
      if(i) cout<<" ";
      cout<<m-ks[i]+1;
    }
    cout<<endl;
    return 0;
  }

  if(m==1){
    SuffixArray sa(s);
    for(int i=0;i<q;i++){
      if(i) cout<<" ";
      cout<<1+sa.sa[ks[i]];
    }
    cout<<endl;
    return 0;
  }

  SuffixArray sa(s+s);
  vector<ll> ans(q,-1);

  ll tmp=0;
  for(int i=1,j=0;i<=n*2;i++){
    if(sa.sa[i]>=n){
      tmp++;
      if(j<q and ks[j]==tmp){
        ans[j]=n*(m-2)+1+sa.sa[i];
        j++;
      }
    }else{
      tmp+=m-1;
      while(j<q and ks[j]<=tmp){
        ans[j]=1+sa.sa[i]-(ks[j]-tmp)*n;
        j++;
      }
    }
  }

  for(int i=0;i<q;i++){
    if(i) cout<<" ";
    cout<<ans[i];
  }
  cout<<endl;
  return 0;
}
0