Trie字典树

#include <bits/stdc++.h>
using namespace std;
const int N = 1000;//N个串
const int Z = 27;//字符集大小（此处以26个小写英文字母为例）
int ch[N][Z], ans;//ch表示字典树，ans表示总节点数
bool bo[N];//第几个节点是否是实际字符串集合中的元素

bool insert(char *s)
{
	int len = strlen(s);
	int u = 1;
	for (int i = 0; i < len; i++)
	{
		char c = s[i] - 'a';
		if (!ch[u][c])
			ch[u][c] = ++ans;
		u = ch[u][c];
	}
	bo[u] = 1;
	return 1;
}

bool find(char *s)
{
	int len = strlen(s);
	int u = 1;
	for (int i = 0; i < len; i++)
	{
		int c = s[i] - 'a';
		if (!ch[u][c])return 0;
		u = ch[u][c];
	}
	if(bo[u])return true;
	else return false;
}

int main()
{
	printf("输入要插入的字符串(按0表示结束插入)：");
	char temp[1000];
	ans = 1;
	while (scanf("%s", temp))
	{
		if (temp[0] == '0')break;
		printf("%s\n",insert(temp)?"插入成功":"插入失败");
	}
	printf("输入要查询的字符串(按0表示结束查询）：");
	while (scanf("%s", temp))
	{
		if (temp[0] == '0')break;
		printf("%s\n", find(temp) ? "Yes" : "No");
	}

	return 0;
}