w15作业--ZJM 与生日礼物（选做）

题意：

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;
}