题意:
ZJM 收到了 Q老师 送来的生日礼物,但是被 Q老师 加密了。只有 ZJM 能够回答对 Q老师 的问题,Q老师 才会把密码告诉 ZJM。
Q老师 给了 ZJM 一些仅有 01 组成的二进制编码串, 他问 ZJM:是否存在一个串是另一个串的前缀.
输入格式:
多组数据。每组数据中包含多个仅有01组成的字符串,以一个9作为该组数据结束的标志。
输出格式:
对于第 k 组数据(从1开始标号),如果不存在一个字符串使另一个的前缀,输出"Set k is immediately decodable",否则输出"Set k is not immediately decodable"。
每组数据的输出单独一行
样例输入:
01
10
0010
0000
9
01
10
010
0000
9
样例输出:
Set 1 is immediately decodable
Set 2 is not immediately decodable
思路:
(1)题意分析:一个字符串和一个字符串集合匹配,因此考虑使用字典树
先记录一下字典树的模板:
struct Trie{
static const int N=1010,num=2;
int tot,root,child[N][num],flag[N];
Trie(){
memset(child,-1,sizeof(child));
root=tot=0;
}
void init(){
memset(child,-1,sizeof(child));
root=tot=0;
}
void insert(string s){
int now=root;
bool jud=false;
for(int i=0;s[i];i++){
int x=s[i]-'a';
if(child[now][x]==-1){
child[now][x]=++tot;
flag[now]=0;
}
now=child[now][x];
}
flag[now]=1;
}
bool query(string s){
int now=root;
for(int i=0;s[i];i++){
int x=s[i]-'a';
if(child[now][x]==-1) return false;
if(flag[now]) return true;
now=child[now][x];
}
return false;
}
};
(2)本题做法:
往字典树中依次插入每个字符串,假设当前字符串为 S
有两种情况需要判断:
S 是否为之前某个字符串的前缀
之前是否有某个字符串是 S 的前缀
对于 S 是否为之前某个字符串的前缀 :
当字符串 S 插入结束后,其最后一个节点是字典树中已经存在的节点,则说明 S 为之前某个字符串的前缀
对于之前是否有某个字符串是 S 的前缀:
在字符串 S 插入过程中,如果遇到某一个字典树中的节点为某一个字符串的结尾,则说明存在某个字符串是 S 的前缀
这里修改一下上述模板中的insert代码即可
bool insert(string s){
int now=root,l=s.size();
bool jud=false;
for(int i=0;i<l;i++){
int x=s[i]-'0';
if(child[now][x]==-1){
child[now][x]=++tot;
flag[now]=0;
}
else if(i==l-1||flag[child[now][x]]) jud=true;
now=child[now][x];
}
flag[now]=1;
return jud;
}
代码:
#include<iostream>
#include<cstring>
#include<string>
using namespace std;
struct Trie{
static const int N=1010,num=2;
int tot,root,child[N][num],flag[N];
Trie(){
memset(child,-1,sizeof(child));
root=tot=0;
}
void init(){
memset(child,-1,sizeof(child));
root=tot=0;
}
bool insert(string s){
int now=root,l=s.size();
bool jud=false;
for(int i=0;i<l;i++){
int x=s[i]-'0';
if(child[now][x]==-1){
child[now][x]=++tot;
flag[now]=0;
}
else if(i==l-1||flag[child[now][x]]) jud=true;
now=child[now][x];
}
flag[now]=1;
return jud;
}
};
int k=1;string str;Trie t;
int main(){
while(cin>>str){
if(str=="9") continue;
t.init();
bool j=t.insert(str);
int fflag=0;
while(cin>>str){
if(str=="9") break;
if(t.insert(str)) fflag=1;
}
if(fflag) cout<<"Set "<<k<<" is not immediately decodable"<<endl;
else cout<<"Set "<<k<<" is immediately decodable"<<endl;
k++;
}
return 0;
}