結果

問題 No.1018 suffixsuffixsuffix
ユーザー ats5515ats5515
提出日時 2020-04-03 22:31:13
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 130 ms / 2,000 ms
コード長 3,349 bytes
コンパイル時間 2,259 ms
コンパイル使用メモリ 155,984 KB
実行使用メモリ 13,076 KB
最終ジャッジ日時 2023-09-16 03:17:12
合計ジャッジ時間 6,948 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge13
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,376 KB
testcase_01 AC 2 ms
4,376 KB
testcase_02 AC 2 ms
4,380 KB
testcase_03 AC 2 ms
4,376 KB
testcase_04 AC 2 ms
4,380 KB
testcase_05 AC 2 ms
4,376 KB
testcase_06 AC 2 ms
4,376 KB
testcase_07 AC 1 ms
4,380 KB
testcase_08 AC 2 ms
4,376 KB
testcase_09 AC 52 ms
7,804 KB
testcase_10 AC 55 ms
7,844 KB
testcase_11 AC 59 ms
8,128 KB
testcase_12 AC 57 ms
7,868 KB
testcase_13 AC 59 ms
7,792 KB
testcase_14 AC 99 ms
12,460 KB
testcase_15 AC 105 ms
12,420 KB
testcase_16 AC 119 ms
12,932 KB
testcase_17 AC 113 ms
12,420 KB
testcase_18 AC 116 ms
12,340 KB
testcase_19 AC 21 ms
4,380 KB
testcase_20 AC 22 ms
4,636 KB
testcase_21 AC 22 ms
4,684 KB
testcase_22 AC 23 ms
4,628 KB
testcase_23 AC 23 ms
4,624 KB
testcase_24 AC 106 ms
12,316 KB
testcase_25 AC 121 ms
12,964 KB
testcase_26 AC 110 ms
12,044 KB
testcase_27 AC 125 ms
12,440 KB
testcase_28 AC 118 ms
12,292 KB
testcase_29 AC 95 ms
13,036 KB
testcase_30 AC 94 ms
12,496 KB
testcase_31 AC 90 ms
12,448 KB
testcase_32 AC 94 ms
12,428 KB
testcase_33 AC 96 ms
12,640 KB
testcase_34 AC 1 ms
4,380 KB
testcase_35 AC 22 ms
4,812 KB
testcase_36 AC 67 ms
12,940 KB
testcase_37 AC 130 ms
13,076 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;
#define int long long
const int MOD = 1000000007;
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);
	}
};
signed main() {
	cin.tie(0);
	ios::sync_with_stdio(false);
	int N, M, Q;
	cin >> N >> M >> Q;
	string S;
	cin >> S;
	vector<int> X(Q);
	for (int i = 0; i < Q; i++) {
		cin >> X[i];
	}
	if (M == 1) {
		SuffixArray sa(S);
		for (int i = 0; i < Q; i++) {
			if (i > 0) cout << " ";
			cout << sa.sa[X[i]] + 1;
		}
		cout << endl;
		return 0;
	}
	vector<int> res(Q);
	string T = S;
	S = S + S;
	SuffixArray sa(S);
	// for (int i = 0; i < sa.rev.size(); i++) {
	// 	cerr << sa.rev[i] << " ";
	// }
	// cerr << endl;
	int cnt = 0;
	vector<int> v;
	int cur = 0;
	for (int i = 1; i < sa.sa.size(); i++) {
		if (sa.sa[i] < N) {
			v.push_back(sa.sa[i]);
			if (i + 1 == (int)sa.sa.size() || sa.sa[i + 1] > sa.sa[i]) {
				while (cur < Q) {
					int t = X[cur];
					int rem = t - (i + cnt) + (int)v.size() - 1;
					if (rem < (M - 1) * v.size()) {
						int rep = rem / v.size();
						res[cur] = v[rem % v.size()] + (M - 2 - rep) * N;
						//cerr << cur << " " << i << " " << v.size() << " " << v[0] << endl;
						cur++;
					} else {
						break;
					}
				}
				cnt += (M - 2) * v.size();
				v.clear();
			}
		} else {
			while (cur < Q) {
				if (i + cnt == X[cur]) {
					res[cur] = sa.sa[i] + (M - 2) * N;
					cur++;
				} else {
					break;
				}
			}
		}
	}
	assert(v.size() == 0);
	assert(cnt == (M - 2) * N);
	for (int i = 0; i < Q; i++) {
		if (i > 0) cout << " ";
		cout << res[i] + 1;
	}
	cout << endl;
	// S = "";
	// for (int i = 0; i < M; i++) {
	// 	S += T;
	// }
	// SuffixArray sa2(S);
	// for (int i = 0; i < Q; i++) {
	// 	if (i > 0) cerr << " ";
	// 	cerr << sa2.sa[X[i]] + 1;
	// }
	// cerr << endl;

	return 0;
}
0