标签:
#include <iostream> #include <algorithm> #include <cstdio> #include <cstring> using namespace std; typedef struct node{ int num; node *next[30]; node(){ memset(next,0,sizeof(next)); num=0; ///值得一学的好处理方法 简单实用 初始化节点的漂亮代码 } }Trie; char dir[32],s[32]; void Insert(node *root,char *s){ node *p=root; for(int i=0; s[i]; i++){ int x=s[i]-‘a‘; if(p->next[x]==NULL) p->next[x]=new node; p=p->next[x]; p->num++; } } int Search(node *root,char *s) { node *p=root; for(int i=0;s[i];i++) { int x=s[i]-‘a‘; if(p->next[x]==NULL) return 0; p=p->next[x]; } return p->num; } void Delete(node *root,char *s,int cnt) { node *p=root; for(int i=0;s[i];i++) { int x=s[i]-‘a‘; p=p->next[x]; p->num-=cnt; //删除节点的操作无非就是把当前前缀开头单词数减掉 } for(int i=0;i<30;i++) p->next[i]=0; //再把之后所有的孩子一律杀死(吼吼 好残忍) } int main() { int n; scanf("%d",&n); Trie *root=new node; while(n--) { scanf("%s %s",dir,s); if(dir[0]==‘i‘) { Insert(root,s); } else if(dir[0]==‘s‘) { if(Search(root,s)) printf("Yes\n"); else printf("No\n"); } else { int cnt=Search(root,s); if(cnt) Delete(root,s,cnt); } } return 0; }
标签:
原文地址:http://www.cnblogs.com/chen9510/p/5512652.html