HDU - 1671 Phone List 某一字符串为另一字符串前缀 STL || trie

HDU - 1671 Phone List

题意：

查询是否存在某一字符串为另一字符串的前缀

参考试题评论写的

一开始字符串忘记排序了，WR了好几次，思维太不严谨了。。。

trie

#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<sstream>
#include<cstring>
#include<bitset>
#include<cstdio>
#include<string>
#include<deque>
#include<stack>
#include<cmath>
#include<queue>
#include<set>
#include<map>
#define mod 1000000007
using namespace std;
typedef long long ll;
const int maxn = 1e5+10;
int tr[maxn][20];
//int sum[maxn];  //统计前缀
int tot,root;
bool vis[10*maxn];  //标记节点是否为字符串结尾
int ans;
struct node
{
    char s[20];
    int len;
}ss[maxn];
bool cmp(node a,node b)  //按照长度排序
{
    return a.len<b.len;
}
void init()
{
    ans = 0;
    tot = 0;
    memset(tr,0,sizeof(tr));
    memset(vis,false,sizeof(vis));
}
void insert(char *p)
{
    int len = strlen(p);
    root = 0;   //根节点编号为0
    for(int i=0;i<len;i++)
    {
        int id = p[i]-'0';  //第二种编码
        if(!tr[root][id])   //如果之前没有从root到id的前缀
            tr[root][id] = ++tot;  //插入,tot即为第一种编号
        //sum[tr[root][id]]++;   //前缀保存
        root = tr[root][id];   //顺着字典树往下走
        if(i==len-1)
            vis[root] = true;  //将从0到root点为一个字符串
    }
}
void find(char *p)
{
    root = 0;  //从根节点开始查找
    for(int i=0;p[i];i++)
    {
        int x = p[i]-'0';
        if(!tr[root][x])  //以root为头节点的x的字母不存在
            return ;
        root = tr[root][x];  //为查询下个字母做准备，往下走
        if(vis[root]==true)  //检验从0到root是否为某一字符串
            ans++;
    }

    //return sum[root];  //返回出现的次数
}

int main()
{
    int t;
    int n;
    string str;
    scanf("%d",&t);
    while(t--)
    {
        init();
        scanf("%d ",&n);
        for(int i=0;i<n;i++)
        {
            gets(ss[i].s);
            ss[i].len = strlen(ss[i].s);
        }
        sort(ss,ss+n,cmp);   //排序
        for(int i=0;i<n;i++)   
        {
            if(i)
                find(ss[i].s);
            insert(ss[i].s);
        }
        if(ans)
            printf("NO\n");
        else
            printf("YES\n");

    }
    return 0;
}

vector

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
const int maxn = 1e6+10;
int main()
{
    int t,n;
    char s[10];
    string str;
    scanf("%d",&t);
    while(t--)
    {
        vector<string>v;
        scanf("%d ",&n);
        int flag = 0;
        for(int i=0;i<n;i++)
        {
            gets(s);
            str = s;
            v.push_back(str);
        }
        sort(v.begin(),v.end());
        for(int i=0;i<n-1;i++)
        {                          //注意find不属于vector的成员，而存在于算法中，应加上头文件#include <algorithm>
            if(v[i+1].find(v[i])==0)  //返回首地址   如果一个字符串以另一个字符串为前缀，则这两个字符串一定相邻
            {
                flag = 1;
                break;
            }
        }
        if(flag)
            printf("NO\n");
        else
            printf("YES\n");
    }
    return 0;
}