字典树模板题目。
字典树又可以叫做,前缀树,从根遍历到叶结点的一条路,便是字符串的所有前缀。
我们按照这个性质,来进行这道题的模拟。
首先是输入,用一个二维数组来存储,一旦==9,就开始建树和索搜。
其他的是模板。
#include<iostream>
#include<string>
#include<algorithm>
#include<cmath>
#include<cstdlib>
#include<climits>
#include<stack>
#include<vector>
#include<queue>
#include<set>
#include<map>
//#include<regex>
#include<cstdio>
#define up(i,a,b) for(int i=a;i<b;i++)
#define dw(i,a,b) for(int i=a;i>b;i--)
#define upd(i,a,b) for(int i=a;i<=b;i++)
#define dwd(i,a,b) for(int i=a;i>=b;i--)
//#define local
typedef long long ll;
const double esp = 1e-6;
const double pi = acos(-1.0);
const int INF = 0x3f3f3f3f;
const int inf = 1e9;
using namespace std;
typedef pair<int, int> pir;
char s[10][12];
int root = 0;
int cnt = 0;
struct tire {
int next[2];
bool flag;
tire() {
flag = false;
}
}tree[100];
void insert(char s[10])
{
int j = 0;
int n = strlen(s);
for (int i = 0; i < n; i++)
{
if (!tree[j].next[s[i]-'0'])
{
tree[j].next[s[i] - '0'] = ++ cnt;
}
j = tree[j].next[s[i] - '0'];
}
tree[j].flag = true;
}
bool search1(int x)
{
int n = strlen(s[x]);
int j = root;
up(i, 0, n)
{
j = tree[j].next[s[x][i] - '0'];
if (tree[j].flag == true)
{
if (tree[j].next[1]||tree[j].next[0])
{
return false;
}
}
}
return true;
}
int main()
{
int t = 0;
int xx = 0;
while (cin >> s[t++])
{
if (s[t-1][0] == '9')
{
xx++;
t--;
cnt = 0;
root = 0;
memset(tree, 0, sizeof(tree));
up(i, 0, t)
{
insert(s[i]);
}
bool flag = true;
up(i, 0, t)
{
bool temp = search1(i);
if (!temp)
{
printf("Set %d is not immediately decodable\n", xx);
flag = false;
break;
}
}
if (flag)
{
printf("Set %d is immediately decodable\n", xx);
}
t = 0;//注意要清零
}
}
return 0;
}