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

来源：互联网发布：js正则判断数字范围编辑：程序博客网时间：2024/05/29 14:30

点击打开链接

#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+7using 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;}

0 0