Trie树和并查集

Trie树

作用：

高效地存储和查找字符串集合的数据结构

基本原理：

利用树的原理来存储每一个字符串，字符串中的每个字符都作为树中的节点。查找是否存在该字符串时，就是从树的根节点出发，判断是否存在该字符串。

代码

#include <iostream>
using namespace std;

const int N = 100010;

int son[N][26], cnt[N], idx;//下标是0的点既是根节点，也是空节点
char s[N];

void insert(char str[])
{
	int p = 0;
	for (int i = 0; str[i]; i++) {
		int u = str[i] - 'a';
		if (!son[p][u]) son[p][u] = ++idx;
		p = son[p][u];
	}

	cnt[p]++;//表示以这个点结尾的单词数量多了一个
}

//返回这个字符串出现多少次
int query(char str[])
{
	int p = 0;
	for (int i = 0; str[i]; i++) {
		int u = str[i] - 'a';
		if (!son[p][u]) return 0;
		p = son[p][u];
	}

	return cnt[p];
}

int main()
{
	int n;
	cin >> n;
	while (n--) {
		char op[2];
		cin >> op >> s;
		if (op[0] == 'I') insert(s);
		else cout << query(s) << endl;
	}

	return 0;
}

并查集

作用：

1.将两个集合合并

2.询问两个元素是否在一个集合中

基本原理：

每个集合用一棵树表示。树根的编号就是整个集合的编号。每个节点存储他的父节点，p[x]表示它的父节点

问题：

问题1：如何判断树根：if（p[x] == x）

问题2：如何求x的集合编号：while（p[x] != x）x = p[x]

问题3：如何合并两个集合：px是x的集合编号，py是y的集合编号。p[x] = y(也就是把集合x插到集合y中)

代码

#include <iostream>
using namespace std;

const int N = 100010;

int n, m;
int p[N];

int find(int x)//返回x的祖宗节点 + 路径压缩
{
	if (p[x] != x) p[x] = find(p[x]);
	return p[x];
}

int main()
{
	cin >> n >> m;
	for (int i = 1; i <= n; i++) p[i] = i;

	while (m--) {
		char op[2];
		int a, b;
		cin >> op >> a >> b;

		if (op[0] == 'M') p[find(a)] = find(b);//将a，b集合合并
		else {//判断a，b是否在同一个集合中
			if (find(a) == find(b))cout << "Yes" << endl;
			else cout << "No" << endl;
		}
	}
	return 0;
}