BZOJ 2251.[2010Beijing Wc]外星联络(Trie)

最新推荐文章于 2018-09-23 09:54:46 发布

原创最新推荐文章于 2018-09-23 09:54:46 发布 · 746 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种使用Trie树统计01串中所有出现次数大于一次的子串的方法，并提供了完整的C++实现代码。通过将字符串的后缀依次加入到Trie树中，可以确保所有子串都被正确地表示出来且按照字典序排列。

给定一个01串，输出子串出现次数大于1次的次数，按子串的字典序输出。

首先应该知道：把一个字符串的后缀依次加入Trie树中，用Trie树就能不重不漏的表示这个字符串的所有后缀，而且按这棵Trie树dfs的顺序得到的后缀就是字典序的。这样直接统计就可以了。

CODE:

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int MAX_N = 3005;

int n;
char s[MAX_N];

void init()
{
	scanf("%d%s", &n, s + 1); 
}

struct Trie {
	int ch[2], cnt;
} t[MAX_N * MAX_N];
int sz = 0;

void insert(char *s)
{
	int u = 0, l = strlen(s);
	for (int i = 0; i < l; i ++) {
		int c = s[i] - '0';
		if (!t[u].ch[c]) t[u].ch[c] = ++ sz;
		u = t[u].ch[c];
		t[u].cnt ++;
	}
}

void dfs(int u)
{
	if (t[u].cnt > 1) printf("%d\n", t[u].cnt);
	if (t[t[u].ch[0]].cnt > 1) dfs(t[u].ch[0]);
	if (t[t[u].ch[1]].cnt > 1) dfs(t[u].ch[1]);
}

void doit()
{
	for (int i = 1; i <= n; i ++) 
		insert(s + i);
	dfs(0);
}

int main()
{
	//freopen("bzoj2251.in", "r", stdin);
	init();
	doit();
	return 0;
}