题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1671
题目大意:
给你很多电话号码,判断其中是否存在号码是其他号码的前缀,若存在,输出NO.否则输出YES
解题思路:
很简单的字典树的变形。只需要判断2种情况:
1. 9112然后输入911
2.911然后输入9112
通过这道题,认识了一种灰常巧妙的方法:空插法(自己起的名。。。。)
就是如果存在一个号码是另一个号码的前缀,则结果就确定了是NO.但是你还是需要继续输入剩下的号码。当时想用一个break跳出,但是这显然就是不行的。后来看网上的解题报告,发现可以用一个bool型变量标记是否出现这种情况。
如果已经出现前缀,则将bool赋值为false。然后对以后的号码就不需要进行处理,continue即可。若没有出现前缀,则还是插入字典树。这样,就可以非常巧妙的避免前面出现前缀而后面做无用功的情况。
代码如下:
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define MAXN 120100
int count;
struct Dictree //字典树数据结构
{
int num;
Dictree *tire[10];
}*head;
Dictree Head[MAXN];
bool insert(char str[])
{
int len, temp;
len = strlen(str);
head = &Head[0]; //头结点是字典树入口,没有数据
for(int i = 0; i < len; ++i)
{
temp = str[i] - 48;
if(head->tire[temp] == NULL)
head->tire[temp] = &Head[++count]; //字母无出现,则赋予空间
head = head->tire[temp];
if(head->num) //该电话是否含有其他电话
return false;
}
head->num = 1;
for(int i = 0; i < 10; ++i) //该电话是否是其他电话的前缀
if(head->tire[i] != NULL)
return false;
return true;
}
int main()
{
//freopen("Input.txt", "r", stdin);
int ncase;
int num;
bool flag;
char str[12];
scanf("%d", &ncase);
while(ncase--)
{
flag = true; //标记是否有不符合条件的电话
count = 0;
memset(Head, 0, sizeof(Head));
scanf("%d", &num);
for(int i = 1; i <= num; ++i)
{
scanf("%s", str);
if(flag == false) //有则不用处理(空插,太巧妙了~~~~)
continue;
if(insert(str) == false)
flag = false;
}
if(flag) printf("YES\n");
else printf("NO\n");
}
return 0;
}
动态分配内存+释放
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
using namespace std;
struct Tire
{
int num;
Tire *next[10];
}*Head;
void init()
{
Head = new Tire;
for(int i = 0; i < 10; ++i)
Head->next[i] = NULL;
}
bool insert(char str[])
{
Tire *head = Head;
int len, temp;
len = strlen(str);
for(int i = 0; i < len; ++i)
{
temp = str[i] - 48;
if(head->next[temp] == NULL)
{
head->next[temp] = new Tire;
head = head->next[temp];
head->num = 0;
for(int j = 0; j < 10; ++j)
head->next[j] = NULL;
}
else
head = head->next[temp];
if(head->num == 1)
return false;
}
head->num = 1;
for(int i = 0; i < 10; ++i)
if(head->next[i] != NULL)
return false;
return true;
}
void del(Tire *head)
{
for(int i = 0; i < 10; ++i)
if(head->next[i] != NULL)
del(head->next[i]);
free(head);
}
int main()
{
int ncase;
int num;
bool flag;
char str[12];
scanf("%d", &ncase);
while(ncase--)
{
init();
flag = true;
scanf("%d", &num);
for(int i = 1; i <= num; ++i)
{
scanf("%s", str);
if(flag == false)
continue;
if(insert(str) == false)
flag = false;
}
if(flag)
printf("YES\n");
else
printf("NO\n");
del(Head);
}
return 0;
}