题目大意:给出长短不一的电话号码,问是否可以拨打成功。换而言之就是问号码是否存在完全覆盖的情况。
解题思路:这是字典树的一个简单变形,只需要对每个号码边接收建树边查找标记即可,对于每一个号码建树完成后,标记其最后叶子的值为-1,这样在查找的时候遇见-1或者遍历完成也没有出现不匹配的情况时,则说明存在完全覆盖。code中在建树的时候对结点的计数发生改变,但这里巧妙地避免了标记为-1的情况被改变,因为是先查找判断,再将其作为词源建树的,因此在查找的过程中不会出现交集,那么标记为-1的点就不会被改变。因此这两种结点标记和计数是独立的。详见code。
题目来源:http://acm.hdu.edu.cn/showproblem.php?pid=1671
code:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
using namespace std;
const int MAXN = 10;
int t,n;
typedef struct node{
int cnt;
node* next[MAXN];
}Trie;
Trie root;
void createTrie(char* str){
int len=strlen(str); //计算词的长度
Trie *p=&root,*q; //初始化当前结点为根节点,以及中间变量的指针
for(int i=0;i<len;i++){ //将每个字符依次插入字典树中
int id=str[i]-'0'; //计算下标
if(p->next[id]==NULL){ //如果没有出现过,则增加结点
q=new Trie; //初始化一个新节点,开辟一个空间
q->cnt=1; //将其出现次数初始化为1
for(int j=0;j<MAXN;j++) //将其子结点初始化为NULL
q->next[j]=NULL;
p->next[id]=q; //将其付给p的下一个结点
p=p->next[id]; //指针下移
}
else{ //如果出现过
p->next[id]->cnt++; //将对应的子结点的计数加一
p=p->next[id]; //指针下移
}
}
p->cnt=-1; //将最后的结点置为-1,用来标记,方便查找的时候看是否覆盖完全,这里的标记和上面的计数是完全独立的
}
int findTrie(char* str){
int len=strlen(str); //计算关键词的长度
Trie *p=&root; //初始化当前结点为根节点
for(int i=0;i<len;i++){ //依次查找
int id=str[i]-'0'; //计算下标
p=p->next[id]; //下移指针查找
if(p==NULL) return 0; //如果没有比较完成就不匹配,则说明该字典树中不存在完全覆盖
if(p->cnt==-1) return -1; //如果发现匹配过程中出现结点被标记为-1,则说明完全匹配,返回-1.如1234匹配123,发现4这点标记为-1
}
return -1; //如果遍历结束,则说明存在完全匹配,返回-1.如123匹配1234,已经匹配完成,则说明也是完全覆盖的
}
int deal(Trie* T){ //释放内存空间
if(T==NULL) return 0;
for(int i=0;i<MAXN;i++){
if(T->next[i]!=NULL)
deal(T->next[i]);
}
free(T);
return 0;
}
int main(){
char str[MAXN];
scanf("%d",&t);
while(t--){
int flag=0;
for(int i=0;i<MAXN;i++) //初始化根节点
root.next[i]=NULL;
scanf("%d",&n);
for(int i=0;i<n;i++){ //边接收边查找和标记
scanf("%s",str);
if(findTrie(str)==-1) flag=1; //在输入和建树的过程中,发现存在完全匹配,则标记
if(flag==1) continue; //如果已经找到完全匹配,则不进行建树了,继续接收下一个号码,接收完成输出NO
createTrie(str);
}
if(flag) printf("NO\n");
else printf("YES\n");
deal(&root); //释放内存
}
return 0;
}