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

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

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

CODE:

#include 
#include 
#include 

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;
}