[BZOJ3238][Ahoi2013]差异

[BZOJ3238][Ahoi2013]差异

试题描述

输入

一行,一个字符串S

输出

一行,一个整数,表示所求值

输入示例

cacao

输出示例

54

数据规模及约定

2<=N<=500000,S由小写英文字母组成

题解

我们考虑把两部分分开计算,“∑ len(Ti) + len(Tj)” 显然可以推出公式,现在就是算“∑ lcp(Ti, Tj)”的问题了。

我们可以先把 height 数组求出来,那么 ∑ lcp(Ti, Tj) = ∑l ∑r min{ heightl, heightl+1, ... , heightr };

接下来我们可以令 sufl = ∑r min{ heightl, heightl+1, ... , heightr },于是这个 sufl 就可以递推转移得到了,我们只需要处理出每个位置 i 后边最靠前的位置 p 满足 height[p] < height[i],那么 sufi = sufp + height[i] * (p - i)(想一想,为什么)。

那么最终 ∑ lcp(Ti, Tj) = ∑ sufi,问题解决。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cctype>
#include <algorithm>

#define maxn 500010
#define maxlog 19
#define LL long long

char S[maxn];
int n, rank[maxn], height[maxn], sa[maxn], Ws[maxn];

bool cmp(int* a, int p1, int p2, int l) {
	if(p1 + l > n && p2 + l > n) return a[p1] == a[p2];
	if(p1 + l > n || p2 + l > n) return 0;
	return a[p1] == a[p2] && a[p1+l] == a[p2+l];
}
void ssort() {
	int *x = rank, *y = height;
	int m = 0;
	for(int i = 1; i <= n; i++) Ws[x[i] = S[i]]++, m = std::max(m, x[i]);
	for(int i = 1; i <= m; i++) Ws[i] += Ws[i-1];
	for(int i = n; i; i--) sa[Ws[x[i]]--] = i;
	for(int j = 1, pos = 0; pos < n; j <<= 1, m = pos) {
		pos = 0;
		for(int i = n - j + 1; i <= n; i++) y[++pos] = i;
		for(int i = 1; i <= n; i++) if(sa[i] > j) y[++pos] = sa[i] - j;
		for(int i = 1; i <= m; i++) Ws[i] = 0;
		for(int i = 1; i <= n; i++) Ws[x[i]]++;
		for(int i = 1; i <= m; i++) Ws[i] += Ws[i-1];
		for(int i = n; i; i--) sa[Ws[x[y[i]]]--] = y[i];
		std::swap(x, y); pos = 1; x[sa[1]] = 1;
		for(int i = 2; i <= n; i++) x[sa[i]] = cmp(y, sa[i], sa[i-1], j) ? pos : ++pos;
	}
	return ;
}
void calch() {
	for(int i = 1; i <= n; i++) rank[sa[i]] = i;
	for(int i = 1, j, k = 0; i <= n; height[rank[i++]] = k)
		for(k ? k-- : 0, j = sa[rank[i]-1]; S[j+k] == S[i+k]; k++);
	return ;
}

int mnh[maxlog][maxn], Log[maxn];
void rmq_init() {
	Log[1] = 0;
	for(int i = 2; i <= n; i++) Log[i] = Log[i>>1] + 1;
	for(int i = 1; i <= n; i++) mnh[0][i] = height[i];
	for(int j = 1; (1 << j) <= n; j++)
		for(int i = 1; i + (1 << j) - 1 <= n; i++)
			mnh[j][i] = std::min(mnh[j-1][i], mnh[j-1][i+(1<<j-1)]);
	return ;
}
int qhei(int l, int r) {
	if(r > n) return -1;
	int t = Log[r-l+1];
	return std::min(mnh[t][l], mnh[t][r-(1<<t)+1]);
}

LL f[maxn], ans;

int main() {
	scanf("%s", S + 1);
	
	n = strlen(S + 1);
	ssort();
	calch();
//	for(int i = 1; i <= n; i++) printf("%d%c", height[i], i < n ? ' ' : '\n');
	rmq_init();
	for(int i = n; i; i--) {
		int l = i, r = n + 2;
		while(r - l > 1) {
			int mid = l + r >> 1;
//			printf("[%d, %d]: %d\n", i, mid, qhei(i, mid));
			if(qhei(i, mid) >= height[i]) l = mid;
			else r = mid;
		}
		if(l <= n) l++;
//		printf("%d\n", l);
		f[i] = f[l] + (LL)(l - i) * height[i];
		ans += f[i];
	}
	ans <<= 1;
	
//	LL all = ((LL)n * (n + 1) >> 1) * (n - 1);
	printf("%lld\n", ((LL)n * (n + 1) >> 1) * (n - 1) - ans);
	
	return 0;
}

上面代码中我求每个位置的后继时用的二分,下面有一个(看上去?)暴力的求法,交到大视野上发现更快了,不知道这个暴力做法靠不靠谱。。。

我们令位置 i 的后继为 nxt[i],那么暴力的做法就是检查位置 height[i+1] < height[i] 是否成立,如果成立那么显然 nxt[i] = i + 1,否则沿着 nxt 数组往下走一直碰到 height 值更小的位置。

填坑:这个做法其实就是单调栈。。。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cctype>
#include <algorithm>

#define maxn 500010
#define maxlog 19
#define LL long long

char S[maxn];
int n, rank[maxn], height[maxn], sa[maxn], Ws[maxn];

bool cmp(int* a, int p1, int p2, int l) {
	if(p1 + l > n && p2 + l > n) return a[p1] == a[p2];
	if(p1 + l > n || p2 + l > n) return 0;
	return a[p1] == a[p2] && a[p1+l] == a[p2+l];
}
void ssort() {
	int *x = rank, *y = height;
	int m = 0;
	for(int i = 1; i <= n; i++) Ws[x[i] = S[i]]++, m = std::max(m, x[i]);
	for(int i = 1; i <= m; i++) Ws[i] += Ws[i-1];
	for(int i = n; i; i--) sa[Ws[x[i]]--] = i;
	for(int j = 1, pos = 0; pos < n; j <<= 1, m = pos) {
		pos = 0;
		for(int i = n - j + 1; i <= n; i++) y[++pos] = i;
		for(int i = 1; i <= n; i++) if(sa[i] > j) y[++pos] = sa[i] - j;
		for(int i = 1; i <= m; i++) Ws[i] = 0;
		for(int i = 1; i <= n; i++) Ws[x[i]]++;
		for(int i = 1; i <= m; i++) Ws[i] += Ws[i-1];
		for(int i = n; i; i--) sa[Ws[x[y[i]]]--] = y[i];
		std::swap(x, y); pos = 1; x[sa[1]] = 1;
		for(int i = 2; i <= n; i++) x[sa[i]] = cmp(y, sa[i], sa[i-1], j) ? pos : ++pos;
	}
	return ;
}
void calch() {
	for(int i = 1; i <= n; i++) rank[sa[i]] = i;
	for(int i = 1, j, k = 0; i <= n; height[rank[i++]] = k)
		for(k ? k-- : 0, j = sa[rank[i]-1]; S[j+k] == S[i+k]; k++);
	return ;
}

int nxt[maxn];
LL f[maxn], ans;

int main() {
	scanf("%s", S + 1);
	
	n = strlen(S + 1);
	ssort();
	calch();
//	for(int i = 1; i <= n; i++) printf("%d%c", height[i], i < n ? ' ' : '\n');
	nxt[n+1] = n + 1; height[n+1] = -1;
	for(int i = n; i; i--) {
		if(height[i+1] < height[i]) nxt[i] = i + 1;
		else {
			nxt[i] = nxt[i+1];
			while(height[nxt[i]] >= height[i]) nxt[i] = nxt[nxt[i]];
		}
		f[i] = f[nxt[i]] + (LL)(nxt[i] - i) * height[i];
		ans += f[i];
	}
	ans <<= 1;
	
//	LL all = ((LL)n * (n + 1) >> 1) * (n - 1);
	printf("%lld\n", ((LL)n * (n + 1) >> 1) * (n - 1) - ans);
	
	return 0;
}

求 hack 啊。。。。。

posted @ 2017-03-07 16:19  xjr01  阅读(165)  评论(0编辑  收藏  举报