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;
}