Week15—字典树应用，字符串包含问题

问题描述

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

问题分析

字典树基本知识
查询集合T中是否存在一个字符串是S的前缀，用构建完的字典树匹配字符串S。
一个字符串和一个字符串集合匹配，因此考虑使用字典树
做法：

1. 往字典树中依次插入每个字符串，假设当前字符串为 S
2. 有两种情况需要判断，S 是否为之前某个字符串的前缀，之前是否有某个字符串是 S 的前缀
3. 往字典树中依次插入每个字符串，假设当前字符串为 S，判断：【1】S 是否为之前某个字符串的前缀，当字符串 S 插入结束后，其最后一个节点是字典树中已经存在的节点，则说明 S 为之前某个字符串的前缀【2】之前是否有某个字符串是 S 的前缀，在字符串 S 插入过程中，如果遇到某一个字典树中的节点为某一个字符串的结尾，则说明存在某个字符串是 S 的前缀

代码

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;

struct Tire
{
	//N表示字典数最多的节点个数，charset表示字符种类数
	//tot表示节点编号，root表示根节点编号
	//child[now][x]表示节点now的第x个儿子
	//flag[now]=1表示节点now是某一个字符串的结尾 
	static const int N=1010,charset=30;
	int tot,root,child[N][charset],flag[N];
	Tire()
	{
		memset(child,-1,sizeof child);
		root=tot=0;
	}
	void clear()
	{
		memset(child,-1,sizeof child);
		root=tot=0;
	}
	int insert(char *str)
	{
		//在字典树中插入字符串str 
		int now=root,jud=0,len=strlen(str);
		for(int i=0;i<len;i++)
		{
			int x=str[i]-'0';
			if(child[now][x]==-1)
			{
				child[now][x]=++tot;
				flag[now]=0;
			}
			else if(i==len-1||flag[child[now][x]]) 
				jud=1;
			//i==len-1且child[now][x]!=-1说明S是之前某个字符串的前缀
			//flag[child[now][x]]==1说明之前某个字符串是S的前缀 
			now=child[now][x];
		}
		flag[now]=1;//标记该节点为字符的结尾
		return jud; 
	}
    bool query(char *str)
    {
    	//查询字典树中是否存在某个完整字符串是str的前缀 
		int now=root; 
		for(int i=0;str[i];i++)
		{
			int x=str[i]-'0';
			if(child[now][x]==-1)
	        	return false;
	        if(flag[now]) return true;	
			now=child[now][x];
		} 
		return false;
	}
};
 
int main()
{
	char str[30];
	Tire T;
	int k=1,flag=0;
	while(cin>>str)
	{
		if(str[0]=='9')
		{
			T.clear();
			if(flag==1) cout<<"Set "<<k<<" is not immediately decodable"<<endl;
			else cout<<"Set "<<k<<" is immediately decodable"<<endl;
			flag=0;
			k++;
			continue;
		} 
		else 
		{
		    int p=T.insert(str);//数组名相当于指针名 
		    if(p==1) flag=1;
	    }
	}
}