题意:
给你几个只有0和1的字符串,要你判断是否存在某个字符串是另一个字符串的前缀串。
题解:
这道题我用的是数组版的字典树,果然喜闻乐见的超内存了,对于数组版的内存不太好把控,所以以后的字典树的题目我会尽量用指针版的(如果习惯的话。。。)
怎么做呢?我把输入的字符串存放起来,再判断,怎么判断呢?在find函数中返回前缀数组vis的值的时候-1就可以了,-1是因为该字符串本身在insert的时候就已经算进去了,所以-1之后判断是否为0,如果不是0就是某个字符串的前缀了。
还有就是如果你是Output Limit Exceeded的话说明你输入的时候是用while(1)输入的,这样是不行的,具体操作看我代码。
还有就是网上的博客的指针版的说法好短。。。想法也比我的好,具体怎么回事可以看指针版的代码。
//数组版
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int MAXN=5000+7;
char c[MAXN][15];
int trie[MAXN][2];
int vis[MAXN];
int tot=1;
void insert(char *s)
{
int rt=0;
for(int i=0;s[i];i++)
{
int x=s[i]-'0';
if(trie[rt][x]==0) trie[rt][x]=tot++;
rt=trie[rt][x];
vis[rt]++;
}
}
int find(char *s)
{
int rt=0;
for(int i=0;s[i];i++)
{
int x=s[i]-'0';
if(trie[rt][x]==0) return -1;
rt=trie[rt][x];
}
return vis[rt]-1;
}
int main()
{
memset(trie,0,sizeof(trie));
memset(vis,0,sizeof(vis));
int k=0,Case=1;
while(~scanf("%s",c[k]))
{
if(c[k][0]=='9')
{
// printf("~~~~~~~~~~~~~%d\n",k);
// for(int i=0;i<k;i++)
// printf("%s\n",c[i]);
bool flag=true;
for(int i=0;i<k;i++)
if(find(c[i])!=0)
{
printf("Set %d is not immediately decodable\n",Case++);
flag=false;
break;
}
if(flag)
printf("Set %d is immediately decodable\n",Case++);
memset(trie,0,sizeof(trie));
memset(vis,0,sizeof(vis));
k=0;
continue;
}
insert(c[k]);
k++;
}
}
//指针版
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
struct node
{
int cnt;
node *next[2];
node(){
cnt=0;
for(int i=0;i<2;i++)
next[i]=0;
}
}*root;
bool flag;
void insert(char *s)
{
node *r=root;
for(int i=0;s[i];i++)
{
int x=s[i]-'0';
if(r->next[x]==0) r->next[x]=new(node);
else{
if(r->next[x]->cnt==1||i==strlen(s)-1)//表示的是在存放字符进字典树时查看该树节点是否已经有一个单词了,有说明该字符串具有前缀,另外就是如果一个单词走完了strlen(s)-1之后还有子节点说明该字符串为某个字符串的前缀。
flag=true;
}
r=r->next[x];
}
r->cnt=1;
}
void del(node *head)
{
for(int i=0;i<2;i++)
if(head->next[i]!=0)
del(head->next[i]);
delete(head);
}
int main()
{
root=new(node);
char s[15];
int k=1;
flag=false;
while(~scanf("%s",s))
{
if(s[0]=='9')
{
if(flag)
printf("Set %d is not immediately decodable\n",k++);
else
printf("Set %d is immediately decodable\n",k++);
flag=false;
del(root);
root=new(node);
continue;
}
insert(s);
}
}