#412217 (C++11) No.962 LCPs

提出ソース

結果

問題	No.962 LCPs
ユーザー	ianCK
提出日時	2019-12-27 06:52:10
言語	C++11 (gcc 11.4.0)
結果	AC
実行時間	92 ms / 2,000 ms
コード長	2,315 bytes
コンパイル時間	697 ms
コンパイル使用メモリ	59,612 KB
実行使用メモリ	13,412 KB
最終ジャッジ日時	2024-04-16 05:53:55
合計ジャッジ時間	3,779 ms
ジャッジサーバーID （参考情報）	judge3 / judge5

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,812 KB
testcase_01	AC	2 ms 6,940 KB
testcase_02	AC	2 ms 6,944 KB
testcase_03	AC	2 ms 6,940 KB
testcase_04	AC	3 ms 6,944 KB
testcase_05	AC	2 ms 6,940 KB
testcase_06	AC	2 ms 6,944 KB
testcase_07	AC	2 ms 6,944 KB
testcase_08	AC	2 ms 6,944 KB
testcase_09	AC	3 ms 6,944 KB
testcase_10	AC	2 ms 6,944 KB
testcase_11	AC	2 ms 6,944 KB
testcase_12	AC	2 ms 6,944 KB
testcase_13	AC	2 ms 6,944 KB
testcase_14	AC	2 ms 6,944 KB
testcase_15	AC	2 ms 6,940 KB
testcase_16	AC	2 ms 6,944 KB
testcase_17	AC	92 ms 13,412 KB
testcase_18	AC	48 ms 8,320 KB
testcase_19	AC	47 ms 8,448 KB
testcase_20	AC	34 ms 8,064 KB
testcase_21	AC	33 ms 8,192 KB
testcase_22	AC	26 ms 6,016 KB
testcase_23	AC	27 ms 6,016 KB
testcase_24	AC	23 ms 6,016 KB
testcase_25	AC	23 ms 5,888 KB
testcase_26	AC	20 ms 5,888 KB
testcase_27	AC	34 ms 6,144 KB
testcase_28	AC	25 ms 5,888 KB
testcase_29	AC	22 ms 5,376 KB
testcase_30	AC	30 ms 6,016 KB
testcase_31	AC	39 ms 8,064 KB
testcase_32	AC	20 ms 5,376 KB
testcase_33	AC	56 ms 8,320 KB
testcase_34	AC	25 ms 5,760 KB
testcase_35	AC	24 ms 5,760 KB
testcase_36	AC	28 ms 6,016 KB
testcase_37	AC	15 ms 5,376 KB
testcase_38	AC	15 ms 5,376 KB
testcase_39	AC	15 ms 5,376 KB
testcase_40	AC	15 ms 5,376 KB
testcase_41	AC	15 ms 5,376 KB
testcase_42	AC	15 ms 5,376 KB
testcase_43	AC	16 ms 5,376 KB
testcase_44	AC	16 ms 5,376 KB
testcase_45	AC	15 ms 5,376 KB
testcase_46	AC	15 ms 5,376 KB
testcase_47	AC	5 ms 5,376 KB
testcase_48	AC	4 ms 5,376 KB
testcase_49	AC	5 ms 5,376 KB
testcase_50	AC	5 ms 5,376 KB
testcase_51	AC	6 ms 5,376 KB
testcase_52	AC	4 ms 5,376 KB
testcase_53	AC	5 ms 5,376 KB
testcase_54	AC	4 ms 5,376 KB
testcase_55	AC	4 ms 5,376 KB
testcase_56	AC	6 ms 5,376 KB
testcase_57	AC	5 ms 5,376 KB
testcase_58	AC	5 ms 5,376 KB
testcase_59	AC	5 ms 5,376 KB
testcase_60	AC	5 ms 5,376 KB
testcase_61	AC	5 ms 5,376 KB
testcase_62	AC	6 ms 5,376 KB
testcase_63	AC	60 ms 8,704 KB
testcase_64	AC	6 ms 5,376 KB
testcase_65	AC	58 ms 8,704 KB

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

ソースコード

raw source code

bool debug = false;
#include <iostream>
#include <utility>
using namespace std;

#define F first
#define S second

constexpr int kN = int(4E5 + 10), kInf = int(1E9 + 10);

struct seg_tree {
	pair<int, int> p[kN << 2];
	void pull(int n) {
		if (p[(n << 1) + 1].F < p[(n << 1) + 2].F) p[n] = p[(n << 1) + 1];
		else if (p[(n << 1) + 1].F == p[(n << 1) + 2].F) {
			if (p[(n << 1) + 1].S & 1) p[n] = p[(n << 1) + 2];
			else p[n] = p[(n << 1) + 1];
		}
		else p[n] = p[(n << 1) + 2];
		return ;
	}
	void init(int n, int l, int r, int a[]) {
		if (l == r) p[n] = {a[l], l};
		else {
			int mid = (l + r) >> 1;
			init((n << 1) + 1, l, mid, a);
			init((n << 1) + 2, mid + 1, r, a);
			pull(n);
		}
		return ;
	}
	pair<int, int> ask(int n, int l, int r, int L, int R) {
		if (L <= l && r <= R) return p[n];
		else if (l > R || L > r) return {kInf, l};
		else {
			int mid = (l + r) >> 1;
			pair<int, int> lans = ask((n << 1) + 1, l, mid, L, R), rans = ask((n << 1) + 2, mid + 1, r, L, R);
			if (lans.F < rans.F) return lans;
			else if (lans.F == rans.F) {
				if (lans.S & 1) return rans;
				else return lans;
			}
			else return rans;
		}
	}
};

int LCP(string &a, string &b) {
	int sz = min(int(a.size()), int(b.size()));
	for (int i = 0; i < sz; i++) if (a[i] != b[i]) return i;
	return sz;
}

seg_tree sg;
int a[kN], tot;

long long int dp(int l, int r) {
	if (l > r) return 0;
	pair<int, int> tmp = sg.ask(0, 1, tot, l, r);
	long long int fl, fr, L, R, ans = 0;
	if (!(tmp.S & 1)) L = tmp.S - 1, R = tmp.S + 1;
	else L = R = tmp.S;
	if (debug) printf("l = %d, r = %d, tmp = (%d, %d), L = %lld, R = %lld, ", l, r, tmp.F, tmp.S, L, R);
	fl = l >> 1;
	fr = r >> 1;
	L >>= 1;
	R >>= 1;
	ans += (L - fl + 1) * (fr - R + 1) * (fr + R - fl - L + 2) / 2;
	if (debug) printf("ans = %lld\n", ans * tmp.F); 
	if (tmp.S & 1) return ans * tmp.F + dp(l, tmp.S - 2) + dp(tmp.S + 2, r);
	return ans * tmp.F + dp(l, tmp.S - 1) + dp(tmp.S + 1, r);
}

int main() {
	int n;
	string s[2];
	cin >> n >> s[1];
	a[1] = int(s[1].size());
	for (int i = 2; i <= n; i++) {
		cin >> s[i & 1];
		a[(i << 1) - 2] = LCP(s[0], s[1]);
		a[(i << 1) - 1] = int(s[i & 1].size());
	}
	tot = (n << 1) - 1;
	sg.init(0, 1, tot, a);
	printf("%lld\n", dp(1, tot));
}

yukicoder

結果

テストケース

ソースコード