百练+Trie数+求字符串前缀的经典模板

#include<stdio.h>
#include<stdlib.h>
#include<iostream>
#include<string.h>
#include<cstring>
#include<string>
#include<math.h>
#include<algorithm>
#define LL long long
#define inf 0x3f3f3f3f
#define mod 1e9+7
using namespace std;
struct Node{
struct Node *next[10];///只有0-9这10个
int isCover, isEnd;
};
int ok;
void clean(Node *p)
{
memset(p->next,0,sizeof(p->next));
p->isCover = p->isEnd = 0;
}
void Insert(char *str, Node *root)
{
Node *p = root;
int id;
while(*str){
id = *str - '0';
if(p->next[id] == NULL){
p->next[id] = (Node *)malloc(sizeof(Node));
clean(p->next[id]);
}
p = p->next[id];
if(p->isEnd) ok = 0;///有前缀了，这里标记一波。
++p->isCover;
++str;///字符串移动一波。
}
if(p->isCover > 1) ok = 0;
p->isEnd = 1;///字符串最后一个字符的附加信息标记。
}
void DELETE(Node *p)
{
for(int i = 0; i < 10; ++i)///典型递归模板，10个表示0-9
if(p->next[i]) DELETE(p->next[i]);
free(p);
}
int main()
{
int t, n;
char str[12];
scanf("%d", &t);
while(t--){
Node *root = (Node *)malloc(sizeof(Node));
scanf("%d", &n);
clean(root); ok = 1;
while(n--){
scanf("%s", str);
if(ok) Insert(str, root);
}
printf(ok ? "YES\n" : "NO\n");///YES表示没有前缀，NO表示有前缀。
DELETE(root);
}
return 0;
}