#171384 (C++11) No.515 典型LCP

提出ソース

結果

問題	No.515 典型LCP
ユーザー	__math
提出日時	2017-05-05 22:42:22
言語	C++11 (gcc 11.4.0)
結果	TLE
実行時間	-
コード長	7,102 bytes
コンパイル時間	1,626 ms
コンパイル使用メモリ	165,252 KB
実行使用メモリ	38,504 KB
最終ジャッジ日時	2023-10-12 09:58:22
合計ジャッジ時間	6,297 ms
ジャッジサーバーID （参考情報）	judge13 / judge15

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	TLE	-
testcase_01	--	-
testcase_02	--	-
testcase_03	--	-
testcase_04	--	-
testcase_05	--	-
testcase_06	--	-
testcase_07	--	-
testcase_08	--	-
testcase_09	--	-
testcase_10	--	-
testcase_11	--	-
testcase_12	--	-
testcase_13	--	-
testcase_14	--	-
testcase_15	--	-
testcase_16	--	-

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>

using namespace std;
#define FOR(i,n) for(int i = 0; i < (n); i++)
#define sz(c) ((int)c.size())
#define ten(n) ((int)1e##n)
using ll = long long;
using Pii = pair<int, int>;
using Pll = pair<ll, ll>;

template<typename ...> static inline int getchar_unlocked(void) { return getchar(); }
template<typename ...> static inline void putchar_unlocked(int c) { putchar(c); }
#define mygc(c) (c)=getchar_unlocked()
#define mypc(c) putchar_unlocked(c)
void reader(int& x) { int k, m = 0; x = 0; for (;;) { mygc(k); if (k == '-') { m = 1; break; }if ('0' <= k&&k <= '9') { x = k - '0'; break; } }for (;;) { mygc(k); if (k<'0' || k>'9')break; x = x * 10 + k - '0'; }if (m) x = -x; }
void reader(ll& x) { int k, m = 0; x = 0; for (;;) { mygc(k); if (k == '-') { m = 1; break; }if ('0' <= k&&k <= '9') { x = k - '0'; break; } }for (;;) { mygc(k); if (k<'0' || k>'9')break; x = x * 10 + k - '0'; }if (m) x = -x; }
int reader(char c[]) { int i, s = 0; for (;;) { mygc(i); if (i != ' '&&i != '\n'&&i != '\r'&&i != '\t'&&i != EOF) break; }c[s++] = i; for (;;) { mygc(i); if (i == ' ' || i == '\n' || i == '\r' || i == '\t' || i == EOF) break; c[s++] = i; }c[s] = '\0'; return s; }
int reader(string& c) { int i; for (;;) { mygc(i); if (i != ' '&&i != '\n'&&i != '\r'&&i != '\t'&&i != EOF) break; }c.push_back(i); for (;;) { mygc(i); if (i == ' ' || i == '\n' || i == '\r' || i == '\t' || i == EOF) break; c.push_back(i); }; return sz(c); }
template <class T, class S> void reader(T& x, S& y) { reader(x); reader(y); }
template <class T, class S, class U> void reader(T& x, S& y, U& z) { reader(x); reader(y); reader(z); }
template <class T, class S, class U, class V> void reader(T& x, S& y, U& z, V & w) { reader(x); reader(y); reader(z); reader(w); }
void writer(int x, char c) { int s = 0, m = 0; char f[10]; if (x<0)m = 1, x = -x; while (x)f[s++] = x % 10, x /= 10; if (!s)f[s++] = 0; if (m)mypc('-'); while (s--)mypc(f[s] + '0'); mypc(c); }
void writer(ll x, char c) { int s = 0, m = 0; char f[20]; if (x<0)m = 1, x = -x; while (x)f[s++] = x % 10, x /= 10; if (!s)f[s++] = 0; if (m)mypc('-'); while (s--)mypc(f[s] + '0'); mypc(c); }
void writer(const char c[]) { int i; for (i = 0; c[i] != '\0'; i++)mypc(c[i]); }
void writer(const string& x, char c) { int i; for (i = 0; x[i] != '\0'; i++)mypc(x[i]); mypc(c); }
void writer(const char x[], char c) { int i; for (i = 0; x[i] != '\0'; i++)mypc(x[i]); mypc(c); }
template<class T> void writerLn(T x) { writer(x, '\n'); }
template<class T, class S> void writerLn(T x, S y) { writer(x, ' '); writer(y, '\n'); }
template<class T, class S, class U> void writerLn(T x, S y, U z) { writer(x, ' '); writer(y, ' '); writer(z, '\n'); }
template<class T> void writerArr(T x[], int n) { if (!n) { mypc('\n'); return; }FOR(i, n - 1)writer(x[i], ' '); writer(x[n - 1], '\n'); }
template<class T> void writerArr(vector<T>& x) { writerArr(x.data(), (int)x.size()); }

template<class T> void chmin(T& a, const T& b) { if (a > b) a = b; }
template<class T> void chmax(T& a, const T& b) { if (a < b) a = b; }

template<class T> T gcd(T a, T b) { return b ? gcd(b, a % b) : a; }
template<class T> T lcm(T a, T b) { return a / gcd(a, b) * b; }
ll mod_pow(ll a, ll n, ll mod) {
	ll ret = 1;
	ll p = a % mod;
	while (n) {
		if (n & 1) ret = ret * p % mod;
		p = p * p % mod;
		n >>= 1;
	}
	return ret;
}
template<class T> T extgcd(T a, T b, T& x, T& y) { for (T u = y = 1, v = x = 0; a;) { T q = b / a; swap(x -= q * u, u); swap(y -= q * v, v); swap(b -= q * a, a); } return b; }
template<class T> T mod_inv(T a, T m) { T x, y; extgcd(a, m, x, y); return (m + x % m) % m; }

template<class val_t>
struct min_t {
	val_t val;
	static min_t zero() { return min_t(); }
	min_t() : val(numeric_limits<val_t>::max()) {}
	min_t(val_t val) : val(val) {}
	operator val_t () const { return val; }
	bool is_zero() const { return val == zero().val; }
	min_t operator+ (const min_t& r) const {
		if (is_zero()) return r;
		else if (r.is_zero()) return *this;

		min_t ret(min(val, r.val));
		return ret;
	}
};

//0-indexed segment tree
template<class P>
class seg_tree {
public:
	vector<P> dat;
	int n;

	void propagate(int i) {
		dat[i] = dat[i * 2 + 1] + dat[i * 2 + 2];
	}

	void init(P* d, int size) {
		n = 1;
		while (n < size) n <<= 1;
		dat.resize(2 * n - 1);
		fill(dat.begin(), dat.end(), P::zero());
		for (int i = 0; i < size; i++) dat[n - 1 + i] = d[i];
		for (int i = n - 2; i >= 0; i--) propagate(i);
	}

	void update(int k, P val) {
		k += n - 1;
		dat[k] = val;
		// dat[k] = dat[k] + val;
		while (k > 0) {
			k = (k - 1) / 2;
			propagate(k);
		}
	}

	//[a,b)
	P query(int a, int b) {
		return query(a, b, 0, 0, n);
	}

	P query(int a, int b, int k, int l, int r) {
		if (r <= a || b <= l) return P::zero();
		if (a <= l && r <= b) return dat[k];

		int md = (l + r) / 2; //[l,md),[md,r)
		int nl = k * 2 + 1, nr = nl + 1;
		P lval = query(a, b, nl, l, md);
		P rval = query(a, b, nr, md, r);

		return lval + rval;
	}
};

struct SAComp {
	const int h;
	const vector<int>& g;
	SAComp(int h, vector<int>& g) : h(h), g(g) {}
	bool operator() (int a, int b) {
		return a == b ? false : g[a] != g[b] ? g[a] < g[b] : g[a + h] < g[b + h];
	}
};
vector<int> buildSA(string& t) {
	const int n = sz(t);
	vector<int> g(n + 1), b(n + 1), v(n + 1);
	FOR(i, n + 1) v[i] = i, g[i] = t[i];
	b[0] = 0; b[n] = 0;

	sort(v.begin(), v.end(), SAComp(0, g));
	for (int h = 1; b[n] != n; h *= 2) {
		SAComp comp(h, g);
		sort(v.begin(), v.end(), comp);
		FOR(i, n) b[i + 1] = b[i] + comp(v[i], v[i + 1]);
		FOR(i, n + 1) g[v[i]] = b[i];
	}
	return v;
}

// Kasai-Lee-Arimura-Arikawa-Park's simple LCP computation: O(n)
// lcp[i] == (sa[i] , sa[i+1]) の共通prefixの長さ
vector<int> buildLCP(string& t, vector<int>& sa) {
	int h = 0;
	const int n = sz(t);
	vector<int> b(n + 1), lcp(n + 1);
	FOR(i, n + 1) b[sa[i]] = i;
	FOR(i, n + 1) {
		if (b[i]) {
			for (int j = sa[b[i] - 1]; j + h<n && i + h<n && t[j + h] == t[i + h]; ++h);
			lcp[b[i] - 1] = h;
		} else lcp[n] = -1;
		if (h > 0) --h;
	}
	return lcp;
}

class lcp_seg_tree {
	string s;
	vector<int> sa, isa, lcp;
	seg_tree<min_t<int>> seg;
public:
	lcp_seg_tree(string& s) : s(s) {
		sa = buildSA(s);
		lcp = buildLCP(s, sa);
		vector<min_t<int>> vp;
		FOR(i, sz(lcp)) vp.emplace_back(lcp[i]);
		seg.init(&vp[0], sz(vp));
		isa.resize(sz(sa));
		FOR(i, sz(sa)) isa[sa[i]] = i;
	}
	int query(int l, int r) {
		int il = isa[l], ir = isa[r];
		if (il > ir) swap(il, ir);
		return seg.query(il, ir);
	}
};

int main() {
	int n; reader(n);
	string s;
	vector<int> idx;
	vector<int> orilen;
	FOR(i, n) {
		idx.push_back(sz(s));

		string a; reader(a);
		for (auto c : a) s.push_back(c);
		s.push_back(';');
		orilen.push_back(sz(a));
	}

	lcp_seg_tree lst(s);

	int m;
	ll x, d; reader(m, x, d);
	const ll nmod = ll(n) * (n - 1);

	ll ans = 0;
	FOR(_, m) {
		int i = (x / (n - 1)) + 1;
		int j = (x % (n - 1)) + 1;
		if (i > j) swap(i, j);
		else j++;

		x = (x + d) % nmod;

		i--;
		j--;
		int len = lst.query(idx[i], idx[j]);
		len = min(len, min(orilen[i], orilen[j]));

		ans += len;
	}
	writerLn(ans);

	return 0;
}

yukicoder

結果

テストケース

ソースコード