Babelfish poj2503

2503 -- Babelfish

题目大意：有10000个外语单词和10000个英语单词对应，现给出10000个外语单词，问对应的英语单词是什么，没有对应则输出eh

思路：构建字典树，每一个节点都是一个字母，trie[i][j]表示树上的第i个节点是26个字幕中的第j个字母，值表示这个字母所在单词的下一个字母的节点编号,end[p]表示一个单词在p号节点结束，在查询时如果遍历要查找的字符串后正好在p点，则说明有这个单词

#include<__msvc_all_public_headers.hpp>//VS代码
#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
const int N = 5e5 + 5;
char s[N][30];
int trie[N*10][30];
int val[N*10], en[N*10];
int tot = 0;
void insert(string s, int k)
{//将第k个字典中的字符串插入字典树中
	int len = s.length(), p = 1;
	for (int i = 0; i < len; i++)
	{
		int ch = s[i] - 'a';//将字符转为0-25的数字
		if (!trie[p][ch])
			trie[p][ch] = ++tot;//树的节点的编号
		p = trie[p][ch];//指针指向下一个节点
	}
	val[p] = k;//每一个字母都已经插入了树中
	en[p] = 1;//这里是一个合法单词的结尾
}
int query(char *s)
{//查询单词
	int len = strlen(s), p = 1;
	for (int i = 0; i < len; i++)
	{
		int ch = s[i] - 'a';
		p = trie[p][ch];
		if (!p)
			return 0;//没有匹配的字母了，说明没有这个单词
	}
	if (en[p])
	{//查找到了对应单词
		return val[p];
	}
	return 0;
}
int main()
{
	cin.tie(0);
	ios::sync_with_stdio(false);
	char c[30];
	int cnt = 0;
	s[0][0] = 'e';
	s[0][1] = 'h';
	while (gets_s(c) && c[0] != '\0')
	{//gets字符串读入空格不读入换行
		cnt++;
		bool flag = 1;
		string temp2;
		int j = 0;
		for (int i = 0; i < strlen(c); i++)
		{
			if (c[i] != ' ' && flag)
			{
				s[cnt][j++] = c[i];
			}
			else if (c[i] == ' ')
			{
				flag = 0;
				continue;
			}
			if (!flag)
			{
				temp2 += c[i];
			}			
		}
		insert(temp2, cnt);		
	}
	char temp3[30];
	while (gets_s(temp3))
	{
		printf("%s\n", s[query(temp3)]);
	}
	return 0;
}