HDU 1671 POJ 3630 Phone List

HDU1671 Phone List
传送门
POJ3630 Phone List
传送门
题目大意：给你一些数字，看是否有相同前缀，有则输出NO，否则输出YES。（注意这里的数字可能有前导0，所以建议用字符串做）
典型的trie树
这里简单介绍一下trie树

trie树就是把字符串建成树的形式。
如有字符串
abcds
abcf
edfg
先建立一个根节点，这个根节点一般不包含任何信息，只是作为一个入口
然后建成如下的树

优点是可以快速的查找单词
缺点是空间占用量巨大
具体建树过程可以参考代码（写的丑....）
也可以自己写
trie树还是好写的

这道题在POJ上建动态的trie树会TLE
因为每次都要new，浪费了不少时间
还有我是读完字符串再dfs
这里可以边读边找 能节约大概100ms左右的时间
dfs好想些 所以就写的dfs
动态版

#include<cstdio>
#include<cstring>
const int maxn=10;
int ans;
struct node
{
  int num;//标记出现次数，最后统计num==1的个数，如果和n相同 则无相同前缀
  node *next[maxn];
  node()
  {
    num=0;
    memset(next,NULL,sizeof(next));
  }
}*root;
void trie()
{
  root=new node();
}
void insert(char ss[])
{
  node *p=root,*t;
  for(int i=0;ss[i];i++)
  {
    if(p->next[ss[i]-'0']==NULL)//如果这个字母没有，就加进去
    {
      t=new node();
      p->next[ss[i]-'0']=t;
    }
    p=p->next[ss[i]-'0'];//p向后移动
    p->num++;
  }
}
void del(node *rot)//删树是关键，不删会MLE
{
  for(int i=0;i<maxn;i++)
    if(rot->next[i]!=NULL)
      del(rot->next[i]);
  delete rot;
}
void dfs(node *p)
{
  if(p->num==1){ans++;return;}
  for(int i=0;i<10;i++)
    if(p->next[i]!=NULL)
      dfs(p->next[i]);
}
int t,n;
char s[15];
int main()
{
  scanf("%d",&t);
  while(t--)
  {
    trie();
    ans=0;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
      scanf("%s",s),insert(s);
    dfs(root);
    if(ans==n)puts("YES");
    else puts("NO");
    del(root); 
  }
  return 0; 
}

静态版

#include<cstdio>
#include<cstring>
const int maxn=10;
int ans;
int t,n,idx;
char s[15];
struct node
{
  int num;
  node *next[maxn];
  node()
  {
    num=0;
    memset(next,NULL,sizeof(next));
  }
}*root,New[100005];
void ctrie()//初始化树，如果边读边找，就不需要这步，这里配合动态版 还是先读完再dfs
{
  root->num=0;
  memset(root->next,NULL,sizeof(root->next));
  for(int i=0;i<100000;i++)
  {
    memset(&New[i],NULL,sizeof(New[0]));
    New[i].num=0;
  }
}
void insert(char ss[])
{
  node *p=root;
  for(int i=0;ss[i];i++)
  {
    if(p->next[ss[i]-'0']==NULL)
      p->next[ss[i]-'0']=&New[idx++];
    p=p->next[ss[i]-'0'];
    p->num++;
  }
}
void dfs(node *p)
{
  if(p->num==1){ans++;return;}
  for(int i=0;i<10;i++)
    if(p->next[i]!=NULL)
      dfs(p->next[i]);
}
int main()
{
  root=&New[0];
  scanf("%d",&t);
  while(t--)
  {
    ctrie();
    idx=1;
    ans=0;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
      scanf("%s",s),insert(s);
    dfs(root);
    if(ans==n)puts("YES");
    else puts("NO"); 
  }
  return 0; 
}

同学写的暴力

#include<cstdio>
#include<cstring>
#include<cstdlib>
#define MAXN 10005
typedef long long LL;
char ss[MAXN][11];
int nn;
int cmp(const void *A, const void *B) {
    return strcmp((char *)A, (char *)B);
}
int main() {
    int T, i, j;
    scanf("%d", &T);
    while (T--) {
        scanf("%d", &nn);
        for (i = 0; i < nn; i++) {
            scanf("%s", ss[i]);
        }
        qsort(ss, nn, sizeof(ss[0]), cmp);
        bool flag = true;
        for (i = 1; i < nn; i++) {
            if (strstr(ss[i], ss[i - 1]) != 0) {
                flag = false; break;
            }
        }
        printf("%s\n", flag ? "YES" : "NO");
    }
    return 0;
}

自己最早写的map
但一直TLE，这里附上别人写的map（POJ依旧TLE，HDU能过）
哎 … 自己代码写的丑 什么都要TLE

#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<map>
using namespace std;
long long rec[10005];
void getlonglong(long long &x )
{
  x=1;
  char c;
  while(c=getchar(),c<'0'||c>'9');
  x=x*10+c-'0';
  while(c=getchar(),c>='0'&&c<='9')x=x*10+c-'0';
}
int cmp(const void *a,const void *b)
{
  return *(long long *)a-*(long long *)b;
}
int main()
{
  int T;
  scanf("%d",&T);
  while( T-- )
  {
    map<long long, bool>mp;
    int N,flag=0;
    scanf("%d",&N);
    for(int i=0;i<N;++i)
      getlonglong(rec[i]);
    qsort(rec,N,sizeof(rec[0]),cmp);
    for(int i=0;i<N;++i)
    {
      long long x=rec[i];
      if(mp.count(x)==0)mp[x]=true;
      else
      {
        flag=1;
        break;
      }
      x/=10;
      while(x>1&&!flag)
      {
        if(mp.count(x)==0)x/=10;
        else
        {
          flag=1;
          break;
        }
      }
    }
    puts(flag?"NO":"YES");
  }
  return 0;
}