HDU 1305 Immediate Decodability && hdu 1671 Phone List（字典树查找前缀）

题目思路

这两题解法基本相同 都是先将字符串插入字典树
最后查找前缀
注意这里的前缀是说一个串是否能做能一个串的前缀
之前还以为只要是有相同部分的就算是前缀
查找前缀的方法也很简单
就是遍历每一个字符串 到他的结尾时
判断字典树的flag值是否大于1
如果大于说明有串以它为前缀
标记 并在结尾输出
还有第一题的输入格式需要注意一下

ac代码

HDU 1305 Immediate Decodability

#include <stdio.h>
#include <iostream>
#include <algorithm>
#include <math.h>
#include <string.h>
#include <vector>
#include <stack>
#include <queue>
#include <map>
#include <set>
#include <utility>
#define pi 3.1415926535898
#define ll long long
#define lson rt<<1
#define rson rt<<1|1
#define eps 1e-6
#define ms(a,b) memset(a,b,sizeof(a))
#define legal(a,b) a&b
#define print1 printf("111\n")
using namespace std;
const int maxn = 1e6+10;
const int inf = 0x3f3f3f3f;
const ll llinf =0x3f3f3f3f3f3f3f3f;
const int mod = 1e9+7;

struct Trie
{
    Trie *nex[3];
    int flag;
    Trie()
    {
        flag=1;
        memset(nex,NULL,sizeof(nex));
    }
}*root;

int ans;
void inserts(char *str)
{
    int len=strlen(str);
    Trie *p=root,*q;
    for(int i=0;i<len;i++)
    {
        int id=str[i]-'0';
        if(p->nex[id]==NULL)
        {
            q=new Trie();
            p->nex[id]=q;
            p=p->nex[id];
        }else
        {
            p=p->nex[id];
            ++(p->flag);
        }
    }
}

void query(char *str)
{
    int len=strlen(str);
    Trie *p=root;
    for(int i=0;i<len;i++)
    {
        int id=str[i]-'0';
        p=p->nex[id];//判断应该在这一个语句的后面 因为最开始p在根节点 但根节点上没有值
        if(i==len-1)
        {
            if(p->flag>1)
            {
                ans=1;
                return;
            }

        }

        if(p==NULL)
            return;
    }
}

void free(Trie *t)
{
    if(t==NULL)return;
    for(int i=0;i<3;i++)
    {
        if(t->nex[i])free(t->nex[i]);
    }
    delete(t);
}

char a[maxn][50];

int main()
{
    int ce=1;
	while(~scanf("%s",a[0]))
    {
        root= new Trie();//这里的初始化要记得写 总是忘记
        ans=0;
        inserts(a[0]);
        int n=1;
        while(scanf("%s",a[n])&&a[n][0]!='9')
        {
            inserts(a[n]);
            n++;
        }
        for(int i=0;i<n;i++)
        {
            query(a[i]);

        }

        if(ans==0)
            printf("Set %d is immediately decodable\n",ce++);
        else
            printf("Set %d is not immediately decodable\n",ce++);
        free(root);//记得free不然可能mle
    }
	return 0;
}

hdu 1671 Phone List

#include <stdio.h>
#include <iostream>
#include <algorithm>
#include <math.h>
#include <string.h>
#include <vector>
#include <stack>
#include <queue>
#include <map>
#include <set>
#include <utility>
#define pi 3.1415926535898
#define ll long long
#define lson rt<<1
#define rson rt<<1|1
#define eps 1e-6
#define ms(a,b) memset(a,b,sizeof(a))
#define legal(a,b) a&b
#define print1 printf("111\n")
using namespace std;
const int maxn = 5e5+10;
const int inf = 0x3f3f3f3f;
const ll llinf =0x3f3f3f3f3f3f3f3f;
const int mod = 1e9+7;

char a[10000][20];
int vis;

struct Trie
{
    Trie *nex[20];
    int flag;
    Trie()
    {
        flag=1;
        memset(nex,NULL,sizeof(nex));
    }
}*root;

void inserts(char *str)
{
    int len=strlen(str);
    Trie *p=root,*q;
    for(int i=0;i<len;i++)
    {
        int id=str[i]-'0';
        if(p->nex[id]==NULL)
        {
            q=new Trie();
            p->nex[id]=q;
            p=p->nex[id];
        }else
        {
            p=p->nex[id];
            ++(p->flag);
        }
    }
}

void query(char *str)
{
    int len=strlen(str);
    Trie *p=root;
    vis=0;
    for(int i=0;i<len;i++)
    {
        int id=str[i]-'0';
        p=p->nex[id];
        if(i==len-1)
        {
            if(p->flag>1)
            {
                vis=1;
                return;
            }
            else
            {
                vis=0;
                return;
            }

        }
    }
}

void free(Trie*t)
{
    if(t==NULL)return;
    for(int i=0;i<20;i++)
    {
        if(t->nex[i])
            free(t->nex[i]);
    }
    delete(t);
}
int main()
{
    int t;

    scanf("%d",&t);
    while(t--)
    {
        root=new Trie();
        int n;
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
            scanf("%s",a[i]);
            inserts(a[i]);
        }
        for(int i=1;i<=n;i++)
        {
            query(a[i]);
            if(vis==1)
            {
                break;
            }
        }
        if(vis==1)
            printf("NO\n");
        else
            printf("YES\n");
        free(root);
    }
}