题目大意:给出若干电话号码,判断是否出现某个电话是另外一个电话的前缀,有就输出NO,没有则输出YES。
思路:字典树,稍微修改一下插入节点的判断条件即可。另外暴力搜索应该也可以A,由于当时刚学了字典树,所以利用字典树写了一下。
#include <iostream>
using namespace std;
#include <stdio.h>
#define MAX 10 //字符集大小
typedef struct TrieNode {
int endFlag;
struct TrieNode *next[MAX];
}TrieNode;
TrieNode Memory[1000010];
int allocp =0;
int flag;
/*初始化*/
void InitTrieRoot(TrieNode **pRoot) {
*pRoot = NULL;
}
/*创建新节点*/
TrieNode *CreateTrieNode() {
int i;
TrieNode *p;
p =&Memory[allocp++];
p->endFlag = 0;
for(i =0 ; i < MAX ; i++) {
p->next[i] = NULL;
}
return p;
}
//插入节点
void InsertTrie(TrieNode **pRoot , char*s) {
int i , k;
TrieNode *p;
if(!(p =*pRoot)) {
p =*pRoot = CreateTrieNode();
}
i =0;
while(s[i]) {
k = s[i++] -'0'; //确定branch
if(p->next[k]) {
//判断是否存在前缀
if (p->next[k]->endFlag==1||s[i]=='\0') {
flag=1;
return;
}
}
else
p->next[k] = CreateTrieNode();
p = p->next[k];
}
p->endFlag=1;
}
int main()
{
int t,n,i;
char str[12];
TrieNode *Root;
scanf("%d",&t);
while(t--) {
flag=allocp=0;
InitTrieRoot(&Root);
scanf("%d",&n);
getchar();
for (i=0;i<n;i++) {
scanf("%s",&str);
if (flag==0)
InsertTrie(&Root,str);
}
if (flag==0)
printf("YES\n");
else
printf("NO\n");
}
return 0;
}