HDU 1671 字典树

字典树的题目，以下题目分析来自于http://www.cnblogs.com/dolphin0520/archive/2011/10/15/2213752.html

{

题目意思很清楚：就是判断输入的电话号码中是否有号码是其他号码的前缀，很显然要用到字典树。根据分析可知:

  如果字符串Xn=X1X2....Xn是字符串Ym=Y1Y2....Ym的前缀，有在插入的时候有两种情况:Xn在Yn之前插入，Xn在Yn之后插入。

  1)如果Xn在Yn之前插入，那么在插入Yn的时候必然经过Xn的路径，此时可以根据判断在这条路径上是否已经有结点被标记已经构成完成的字符串序列来判断是否存在Yn的前缀；

  2)如果Xn在Yn之后插入，那么插入完成之后当前指针指向的结点的next数组中的元素必定不全为NULL。

}

以字典树的思想来写的代码

#include<iostream>
#include<string>
#include<algorithm>
using namespace std;
typedef struct Tnode
{
    Tnode* node[10];
    Tnode() {
        for(int i = 0 ; i < 10 ; i++) node[i] = NULL;
    }
}Tnode;
Tnode* head;
int inserttree(string str) {
    int ans = 1; Tnode* s = head;
    for(int i = 0 ; i < str.length() ; i++) {
        if(s->node[str[i] - '0'] == NULL) {
            ans = 0;
            Tnode* t = new Tnode;
            s->node[str[i] - '0'] = t;
        }
        s = s->node[str[i] - '0'];
    }
    return ans;
}
void deletetree(Tnode* h) {
    for(int i = 0 ; i < 10 ; i++)
        if(h->node[i] != NULL) deletetree(h->node[i]);
    delete h;
}
int cmp(const string& a, const string& b) {
    return a.length() > b.length();
}
int main()
{
    int T; cin >> T;
    while(T--) {
        int n; cin >> n;
        string s[10050]; int ans = 0;
        head = new Tnode;
        for(int i = 0 ; i < n ; i++) cin >> s[i];
        sort(s, s + n, cmp);
        for(int i = 0 ; i < n ; i++) {
            if(ans) break;
            ans = inserttree(s[i]);
        }
        deletetree(head);
        if(!ans) cout << "YES" << endl;
        else cout << "NO" << endl;
    }
    return 0;
}

实际上，题目测试数据比较水

{

当然这道题目也有另一种思路，就是先对电话号码序列进行排序，然后判断相邻的两个号码是否符合条件即可。这种思路利用了一种性质：

存在若干个字符串序列，对于两个字符串序列X[1....n]，Y[1....m]，如果X是Y的前缀，则在对所有的字符串序列按字典顺序排序后，

X[1..n]和Y[1....m]在位置上必然是直接相邻或者间接相邻的。间接相邻指存在字符串序列Z[1....t]，使得X是Z的前缀(X和Z直接相邻)，Z是Y的前缀(Z和Y直接相邻)，则X和Y也被认为是相邻的。下面证明这条性质:

假设X[1...n]是Y[1....m]的前缀，但是X和Y不相邻，即至少在X和Y之间存在一字符串序列Z[1....t]，其中X不是Z的前缀，Z也不是Y的前缀，且字典顺序为X<Z<Y。

证明：

 X不是Z的前缀，并且Z的字典顺序比X大，那么只存在两种情况:

  1)n>=t，这种情况下，必定存在一个i(0<=i<=t)使得X[i]<Z[i]，而由假设可知X是Y的前缀，则必有X[i]=Y[i]，此时Y的字典顺序小于Z，与条件不符，不成立。

  2)n<t，这种情况下，有两种可能，一种是对于所有的i(0<=i<=n)，X[i]=Z[i]，但是如此一来X是Z的前缀，与条件不符，不成立；

     另一种可能是存在i(0<=i<=n)，使得X[i]<Z[i]，与1)中所述情况一样，与条件不符，不成立。

因此假设不成立，从而性质得证。

因此可以利用此性质解决这道题目，题目只需判断是否存在一个号码是否是另一个号码的前缀即可。所以可以先对所有的号码序列进行字典排序，然后判断直接相邻的两个号码是否符号条件即可。如果所有直接相邻的号码都不存在一个号码是另一个号码的前缀，那么这些号码序列中肯定不存在一个号码是另一个号码的前缀。

}

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;

char phone[10000][11];
int cmp(const void* a,const void* b)
{
    return strcmp((char*)a,(char*)b);
}

int main(void)
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int i,n;
        bool flag=false;
        scanf("%d",&n);
        for(i=0;i<n;i++)
        {
            scanf("%s",phone[i]);
        }
        qsort(phone,n,sizeof(phone[0]),cmp);
        for(i=0;i<n-1;i++)
        {
            if(strncmp(phone[i],phone[i+1],strlen(phone[i]))==0)
            {
                flag=true;
                break;
            }
        }
        if(flag==false)
            printf("YES\n");
        else
            printf("NO\n");
    }
    return 0;
}