結果

問題 No.1018 suffixsuffixsuffix
ユーザー ats5515ats5515
提出日時 2020-04-03 22:13:00
言語 C++11
(gcc 11.4.0)
結果
WA  
実行時間 -
コード長 3,000 bytes
コンパイル時間 2,317 ms
コンパイル使用メモリ 170,520 KB
実行使用メモリ 13,020 KB
最終ジャッジ日時 2024-07-03 03:23:10
合計ジャッジ時間 7,905 ms
ジャッジサーバーID
（参考情報） 		judge4 / 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 AC 2 ms
5,376 KB
testcase_04 WA -
testcase_05 AC 2 ms
5,376 KB
testcase_06 WA -
testcase_07 WA -
testcase_08 AC 1 ms
5,376 KB
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 AC 98 ms
12,288 KB
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 AC 111 ms
12,280 KB
testcase_19 AC 22 ms
5,376 KB
testcase_20 AC 22 ms
5,376 KB
testcase_21 AC 22 ms
5,376 KB
testcase_22 AC 22 ms
5,376 KB
testcase_23 AC 23 ms
5,376 KB
testcase_24 AC 102 ms
12,000 KB
testcase_25 AC 123 ms
13,004 KB
testcase_26 AC 107 ms
12,124 KB
testcase_27 AC 116 ms
12,288 KB
testcase_28 AC 116 ms
12,032 KB
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 WA -
testcase_33 WA -
testcase_34 WA -
testcase_35 AC 22 ms
5,376 KB
testcase_36 WA -
testcase_37 AC 127 ms
12,800 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++) {
			cout << sa.sa[X[i]] << endl;
		}
		return 0;
	}
	vector<int> res(Q);
	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);
					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];
					cur++;
				} else {
					break;
				}
			}
		}
	}
	for (int i = 0; i < Q; i++) {
		if (i > 0) cout << " ";
		cout << res[i] + 1;
	}
	cout << endl;
	return 0;
}
0