字典树（一）

1.通过一道简单的题比较了一下指针和数组的写法...两种做法都是参考网上的..数组的写法好像更简洁，耗时间短一点，这里相当于是以后的模板了

题目链接：http://acm.hust.edu.cn/vjudge/contest/123402#problem/A

<span style="font-size:18px;">//指针写法，140ms

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn=26;

struct Node
{
   int cnt;
   Node *nextn[maxn];
   Node()
   {
      memset(nextn,NULL,sizeof(nextn));
      cnt=0;
   }
};

Node *root=new Node(),*p;

void Insert(char *str)
{
   int i,id;
   p=root;
   for(i=0;str[i];i++)
   {
      id=str[i]-'a';
      if(!p->nextn[id])p->nextn[id]=new Node();
      p=p->nextn[id];
      p->cnt++;
   }
}

int Find(char *str)
{
   int i,id;
   p=root;
   for(i=0;str[i];i++)
   {
      id=str[i]-'a';
      if(!p->nextn[id])return 0;
      p=p->nextn[id];
   }
   return p->cnt;
}

void release(Node *root)
{
   int i;
   for(i=0;i<maxn;i++)
   {
      if(root->nextn[i])release(root->nextn[i]);
   }
   delete root;
}

int main()
{
   char str[15];
   while(gets(str))
   {
      if(!strlen(str))break;
      Insert(str);
   }
   while(gets(str)) printf("%d\n",Find(str));
   return 0;
}</span>

<span style="font-size:18px;">//数组写法，62ms

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

const int inf=450000,maxn=26;
int tree[inf][maxn],cnt[inf],num=0;
char str[15];

void Insert()
{
   int i,id,u=0;
   for(i=0;str[i];i++)
   {
      id=str[i]-'a';
      if(!tree[u][id]) tree[u][id]=++num;
      u=tree[u][id];
      cnt[u]++;
   }
}

int Q()
{
   int u=0,i,id;
   for(i=0;str[i];i++)
   {
      id=str[i]-'a';
      if(!tree[u][id])return 0;
      u=tree[u][id];
   }
   return cnt[u];
}

int main()
{
   while(gets(str))
   {
      if(!strlen(str))break;
      Insert();
   }
   while(gets(str))printf("%d\n",Q());
   return 0;
}
</span>

2.第二题，简单应用

<span style="font-size:18px;">/********************************************************************************************
字典树
题目链接 http://acm.hdu.edu.cn/showproblem.php?pid=1247
数组的写法比指针的写法更简单一点，就用数组写了
首先在字典树中找出一个单词中前半部分符合的，然后存下来所有能够切下的地方放在cut数组里面
然后遍历cut数组，从该位置开始在字典树中找符合的单词，相当于是在字典树上找两次
v[inf]是用来标记一个单词的末尾的，如果到了这个单词的结尾就设成-1，如果查找的过程中
出现了-1 ，相当于这个单词作为了另一个单词的前缀
*********************************************************************************************/
#include <iostream>
#include <cstdio>
#include <cstring>
#include <stack>
using namespace std;
const int inf=50005,maxn=26;
int tree[inf][maxn],num,v[inf],cut[105];
char s[105],str[inf][105];

void Insert()
{
   int u=0,i,id;
   for(i=0;s[i];i++)
   {
      id=s[i]-'a';
      if(!tree[u][id]) tree[u][id]=++num;
      u=tree[u][id];
   }
   v[u]=-1;
}

bool is_ok(char *ss)
{
   int u=0,i,j,id,lc=0,flag;
   for(i=0;ss[i];i++)
   {
      id=ss[i]-'a';
      u=tree[u][id];
      if(v[u]==-1 && ss[i+1]) cut[lc++]=i+1;
   }
   for(i=0;i<lc;i++)
   {
      u=flag=0;
      for(j=cut[i];ss[j];j++)
      {
         id=ss[j]-'a';
         if(!tree[u][id]){flag=1;break;}
         u=tree[u][id];
      }
      if(flag)continue;
      if(v[u]==-1)return true;
   }
   return false;
}

int main()
{
   int i,k=0;
   while(scanf("%s",s)!=EOF)
   {
      Insert();
      strcpy(str[k++],s);
   }
   for(i=0;i<k;i++)
   {
      if(is_ok(str[i]))printf("%s\n",str[i]);
   }
   return 0;
}
</span>