poj 3630 找重复前缀 trie树

题目:点击打开链接

题意:

给你一些电话号码,如果其中一个电话号码是另一个电话号码的前缀,那么输出NO,否则YES

分析:

这题就是找最小前缀问题,以前我都是用string然后排序,然后取相等的一段判断就行。今天主要是学习一下trie树。

如果是动态节点的话,会TLE,只能初始化数组,但是别忘了一开始trie数组清0.

思路是在建树的时候,在单词的最后一个节点标记一些结尾,然后查询的时候,如果遍历树的过程中遇到了这种标记,而且还没到单词尾,那么说明这个单词有前缀,直接返回0,否则一直遍历到结尾,返回1.

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int N=1e4+5;
char a[N][11];
struct node
{
    bool ok;
    node *next[11];
    node()
    {
        ok=0;
        memset(next,0,sizeof(next));
    }
}trie[100000]; //最多10000*10个节点
int num;
bool Insert(node *rt,char *s)
{
    int i=0;
    node *p=rt;
    while(s[i]){
        int k=s[i++]-'0';
        if(p->next[k]==NULL)
            p->next[k]=&trie[num++];
        p=p->next[k];
        //if(p->ok)return 0;
    }
    p->ok=1;
    return 1;
}
bool Search(node *rt,char *s)
{
    int i=0;
    node *p=rt;
    while(s[i]){
        int k=s[i++]-'0';
        p=p->next[k];
        if(p->ok&&s[i])return 0;
    }
    //if(cnt>1)return false;
    return 1;
}
int main()
{
    int T;
    //freopen("f.txt","r",stdin);
    scanf("%d",&T);
    while(T--){
        int n;
        memset(trie,0,sizeof(trie)); //别忘了
        node *rt=&trie[0];
        num=1;
        scanf("%d",&n);
        for(int i=0;i<n;i++){
            scanf("%s",a[i]);
            Insert(rt,a[i]);
        }
        bool flag=true;
        for(int i=0;i<n;i++){
           // cout<<i<<endl;
            if(Search(rt,a[i]))continue;
            flag=false;break;
        }
        if(flag)printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}


  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值