#458787 (C++14) No.1018 suffixsuffixsuffix

• 問題: No.1018 suffixsuffixsuffix
• ユーザー: chocorusk
• 提出日時: 2020-04-04 18:10:36
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: WA
• 実行時間: -
• コード長: 2,344 bytes
• コンパイル時間: 1,343 ms
• コンパイル使用メモリ: 120,604 KB
• 実行使用メモリ: 8,064 KB
• 最終ジャッジ日時: 2024-07-03 07:36:08
• 合計ジャッジ時間: 10,471 ms
• ジャッジサーバーID （参考情報）: judge5 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	WA	-
testcase_04	AC	2 ms; 5,376 KB
testcase_05	AC	2 ms; 5,376 KB
testcase_06	WA	-
testcase_07	AC	2 ms; 5,376 KB
testcase_08	AC	2 ms; 5,376 KB
testcase_09	AC	205 ms; 5,376 KB
testcase_10	AC	217 ms; 5,376 KB
testcase_11	AC	223 ms; 5,376 KB
testcase_12	AC	201 ms; 5,376 KB
testcase_13	AC	212 ms; 5,376 KB
testcase_14	AC	316 ms; 7,552 KB
testcase_15	AC	308 ms; 7,664 KB
testcase_16	AC	324 ms; 8,064 KB
testcase_17	AC	315 ms; 7,552 KB
testcase_18	AC	305 ms; 7,660 KB
testcase_19	WA	-
testcase_20	WA	-
testcase_21	AC	164 ms; 5,376 KB
testcase_22	WA	-
testcase_23	AC	167 ms; 5,376 KB
testcase_24	AC	337 ms; 7,552 KB
testcase_25	AC	349 ms; 7,936 KB
testcase_26	AC	327 ms; 7,532 KB
testcase_27	AC	329 ms; 7,680 KB
testcase_28	AC	310 ms; 7,680 KB
testcase_29	AC	191 ms; 5,376 KB
testcase_30	AC	183 ms; 5,376 KB
testcase_31	AC	187 ms; 5,376 KB
testcase_32	AC	182 ms; 5,376 KB
testcase_33	AC	179 ms; 5,376 KB
testcase_34	AC	2 ms; 5,376 KB
testcase_35	AC	165 ms; 5,376 KB
testcase_36	AC	184 ms; 5,376 KB
testcase_37	AC	333 ms; 7,936 KB

ソースコード

#include <cstdio>
#include <cstring>
#include <iostream>
#include <string>
#include <cmath>
#include <bitset>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <deque>
#include <algorithm>
#include <complex>
#include <unordered_map>
#include <unordered_set>
#include <random>
#include <cassert>
#include <fstream>
#include <utility>
#include <functional>
#include <time.h>
#include <stack>
#include <array>
#define popcount __builtin_popcount
using namespace std;
typedef long long int ll;
typedef pair<int, int> P;
struct SuffixArray{
	int n, k;
	vector<int> rk, tmp, sa;
	string s;
	SuffixArray(string s):n(s.size()), s(s), rk(s.size()+1), tmp(s.size()+1), sa(s.size()+1){
		auto compare_sa=[&](int i, int j){
			if(rk[i]!=rk[j]) return rk[i]<rk[j];
			else{
				int ri, rj;
				if(i+k<=n) ri=rk[i+k];
				else ri=-1;
				if(j+k<=n) rj=rk[j+k];
				else rj=-1;
				return ri<rj;
			}
		};
		for(int i=0; i<=n; i++){
			sa[i]=i;
			if(i<n){
				rk[i]=s[i]-'A';
			}else{
				rk[i]=-1;
			}
		}
		for(k=1; k<=n; k<<=1){
			sort(sa.begin(), sa.end(), compare_sa);
			tmp[sa[0]]=0;
			for(int i=1; i<=n; i++){
				tmp[sa[i]]=tmp[sa[i-1]];
				if(compare_sa(sa[i-1], sa[i])) tmp[sa[i]]++;
			}
			rk=tmp;
		}
	}
	int operator[](const int &k) const{
		return sa[k];
	}
};
vector< int > z_algorithm(const string &s) {
  vector< int > prefix(s.size());
  for(int i = 1, j = 0; i < s.size(); i++) {
    if(i + prefix[i - j] < j + prefix[j]) {
      prefix[i] = prefix[i - j];
    } else {
      int k = max(0, j + prefix[j] - i);
      while(i + k < s.size() && s[k] == s[i + k]) ++k;
      prefix[i] = k;
      j = i;
    }
  }
  prefix[0] = (int) s.size();
  return prefix;
}
int main()
{
	int n, q;
	ll m;
	cin>>n>>m>>q;
	string s;
	cin>>s;
	if(m==1){
		SuffixArray sa(s);
		while(q--){
			ll k; cin>>k;
			cout<<sa[k]+1<<" ";
		}
		cout<<endl;
		return 0;
	}
	auto z=z_algorithm(s);
	int d;
	for(d=1; d<=n; d++){
		if(n%d!=0) continue;
		if(z[d]==n-d) break;
	}
	s=s.substr(0, d);
	m*=n/d;
	n=d;
	ll c=0;
	SuffixArray sa(s+s);
	ll x[200020];
	x[0]=0;
	for(int i=1; i<=2*n; i++){
		x[i]=x[i-1]+((sa[i]<n)?(m-1):1);
	}
	while(q--){
		ll k; cin>>k;
		int j=lower_bound(x, x+2*n+1, k)-x;
		ll ans;
		if(sa[j]>=n){
			ans=sa[j]-n+n*(m-1)+1;
		}else{
			ans=n*(x[j]-k)+1+sa[j];
		}
		cout<<ans<<" ";
	}
	cout<<endl;
	return 0;
}