Immediate Decodability HDU - 1305(模板trie)

求这些01串是否有一个是另一个的前缀。。

就是求次数就好了嘛。。。emm。。。

网上竟然都用指针写。。。。

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#define maxn 2000010
#define mem(a, b) memset(a, b, sizeof(a))
using namespace std;
int tot, n, m, rt;
int trie[maxn][], sum[];
char str[][];
//bool vis[maxn];查询整个单词用
void build(char *s)
{
int len = strlen(s);
rt = ;
for(int i=; i<len; i++)
{
int x=s[i]-'';
if(trie[rt][x]==)//现在插入的字母在之前同一节点处未出现过
{
trie[rt][x]=++tot;//字母插入一个新的位置,否则不做处理
}
sum[trie[rt][x]]++; //前缀后移一个位置保存前缀出现的次数
rt=trie[rt][x];//为下个字母的插入做准备
}
/*vis[rt]=true;标志该单词末位字母的尾结点,在查询整个单词时用到*/
}
int qp(char *s)
{
int len = strlen(s);
rt = ;
for(int i=; i<len; i++)
{
int x=s[i]-'';
if(trie[rt][x]==)return ;//以rt为头结点的x字母不存在,返回0
rt=trie[rt][x];//为查询下个字母做准备
}
// cout<< sum[rt] <<endl;
return sum[rt];
//查询整个单词时,应该return vis[rt] , 查询前缀出现的次数时,应该return sum[rt]
} void init()
{
tot = ;
mem(trie, );
mem(sum, );
} int main()
{
init();
int kase = , cnt = ;
while(scanf("%s",str[++cnt])!=EOF)
{
if(strcmp(str[cnt],"")==){
bool flag=true;
for(int i=;i<cnt;i++){
// cout<< temp <<endl;
if(qp(str[i])>)
{
flag=false;
break;
}
}
// cout << flag <<endl;
if(flag) printf("Set %d is immediately decodable\n",kase++);
else printf("Set %d is not immediately decodable\n",kase++); init();
cnt = ;
continue;
}
build(str[cnt]);
}
return ;
}
上一篇:hdu 1305 还是字典树


下一篇:windows下自动关机