《字典树》hdu acm 5.2.2 字典树模版

#include<iostream>
#include<cstring>
using namespace std; char str[12]; int flag; const int maxn=10; typedef struct trie { int v;
	trie *next[maxn]; }trie;
trie *root; void insert(char *str) { int i,id,j; int len=strlen(str);
	trie *p=root,*q; for(i=0;i<len;i++) {
		id=str[i]-'0'; if(p->next[id]==NULL) {
			q=(trie*)malloc(sizeof(trie));
			q->v=1; for(j=0;j<maxn;j++) {
				q->next[j]=NULL; }
			p->next[id]=q;
			p=p->next[id]; } else {
            p->next[id]->v=1;
			p=p->next[id]; } }
	p->v=-1; } int findtrie(char *str) { int len=strlen(str);
	trie *p=root; for(int i=0;i<len;i++) { int id=str[i]-'0';
		p=p->next[id]; if(p==NULL) return 0; if(p->v==-1) return -1; } return -1; } int  deal(trie* T) { int i; if(T==NULL) return 0; for(i=0;i<maxn;i++) { if(T->next[i]!=NULL)
		    deal(T->next[i]); }
	free(T); return 0; } int main() { int t,n,i;
	scanf("%d",&t); while(t--) {
		flag=0;
		root=(trie *)malloc(sizeof(trie)); for(i=0;i<10;i++)
			root->next[i]=NULL;
		scanf("%d",&n); while(n--) {
			scanf("%s",str); if(findtrie(str)==-1)
				flag=1; if(flag) continue;
			insert(str); } if(flag)
			printf("NO\n"); else
			printf("YES\n");
		deal(root); } return 0; }