HDU 1671 : Phone List

最新推荐文章于 2020-06-06 18:01:06 发布

ITKaven

最新推荐文章于 2020-06-06 18:01:06 发布

阅读量993

点赞数 1

CC 4.0 BY-SA版权

分类专栏：字典树 HDU ACM

本文链接：https://blog.csdn.net/qq_37960603/article/details/81164843

ACM 同时被 3 个专栏收录

202 篇文章

订阅专栏

HDU

50 篇文章

订阅专栏

字典树

5 篇文章

订阅专栏

本文介绍了一种利用字典树（Trie）来判断字符串集合中是否存在互为前缀的情况的方法。通过构建字典树并标记字符串结尾，可以在插入新字符串时检查是否已有其前缀或它本身作为其他字符串的前缀。

字典树问题
插入每个字符串前，先判断前面某个字符串是不是该字符串的前缀，或者该字符串是不是前面某个字符串的前缀
详细请看代码：

#include<cstdio>
#include<cstring>
using namespace std;

const int maxn=100000+100;

int trie[maxn][10],tot;
bool vis[maxn];

inline void insert(char* ch){
	
	int len=strlen(ch);
	int root=0;
	for(int i=0;i<len;i++){
		
		int ind=ch[i]-'0';
		if(!trie[root][ind]) trie[root][ind]=++tot;
		root=trie[root][ind];  
	}
	vis[root]=true;  //字符串结尾标记
}

inline bool search(char* ch){
	
	int len=strlen(ch);
	int root=0;
	for(int i=0;i<len;i++){
		
		int ind=ch[i]-'0';
		if(!trie[root][ind]) return false;
		root=trie[root][ind];
		if(vis[root]) return true;//前面某个字符串是该字符串的前缀
	}
	for(int i=0;i<=9;i++) if(trie[root][i]) return true; //该字符串是前面某个字符串的前缀
	return false;
}

int main(){
	
	int T,n;
	bool istrue;
	scanf("%d",&T);
	char ch[20];
	while(T--){
		
		tot=0;
		memset(trie,0,sizeof(trie));
		memset(vis,false,sizeof(vis));
		istrue=true;
		scanf("%d",&n);
		getchar();
		for(int i=1;i<=n;i++){
			
			scanf("%s",ch);
			if(!istrue) continue;
			if(search(ch)) istrue=false;
			insert(ch);
		}
		if(istrue) printf("YES\n");
		else printf("NO\n");
	}
}