Luogu P2580 于是他错误的点名开始了

 传送门

P2580 于是他错误的点名开始了

题目背景

XS中学化学竞赛组教练是一个酷爱炉石的人。

他会一边搓炉石一边点名以至于有一天他连续点到了某个同学两次，然后正好被路过的校长发现了然后就是一顿欧拉欧拉欧拉（详情请见已结束比赛CON900）。

题目描述

这之后校长任命你为特派探员，每天记录他的点名。校长会提供化学竞赛学生的人数和名单，而你需要告诉校长他有没有点错名。（为什么不直接不让他玩炉石。）

输入输出格式

输入格式：

 

第一行一个整数 n，表示班上人数。接下来 n 行，每行一个字符串表示其名字（互不相同，且只含小写字母，长度不超过 50）。第 n+2 行一个整数 m，表示教练报的名字。接下来 m 行，每行一个字符串表示教练报的名字（只含小写字母，且长度不超过 50）。

 

输出格式：

 

对于每个教练报的名字，输出一行。如果该名字正确且是第一次出现，输出“OK”，如果该名字错误，输出“WRONG”，如果该名字正确但不是第一次出现，输出“REPEAT”。（均不加引号）

 

输入输出样例

输入样例#1：

5  
a
b
c
ad
acd
3
a
a
e

 

输出样例#1： 

OK
REPEAT
WRONG

　　

说明

对于 40%的数据，n≤1000,m≤2000；

对于 70%的数据，n≤10000,m≤20000；

对于 100%的数据， n≤10000，m≤100000。

 

---

 

Trie树最重要的东西就是设计结构体的成员。

题目要求判断这个字符串是否在点名表里并且要求是一个完整的名字，但是一般Trie树只能查找前缀。

所以在结构体中就不用再去开计数的成员，只需要bool 一个vis和一个signal，signal用来判断到当前节点是否是一个完整的名字。

vis用来判断以当前节点未结束字符的名字是否被点到过。如果点到过，就balabala，如果没点到过，balabala。

#include <iostream>
#include <cstring>
#include <cstdio>

using namespace std;

struct node {
    node *next[30];
    bool signal;
    bool vis;
}*root;
int n, m;
char s[50];

node *build() {
    node *k = new(node);
    k->signal = false;
    k->vis = false;
    memset(k->next, NULL, sizeof(k->next));
    return k;
}

void insert(char *s) {
    node *p = root;
    char *word = s;
    while(*word) {
        int x = *word-'a';
        if(p->next[x] == NULL) {
            p->next[x] = build();
        }
        p = p->next[x];
        word++;
    }
    p->signal = true;
}

bool find(char *s) {
    char *word = s;
    node *rt = root;
    while(*word) {
        int id = *word-'a';
        if(rt->next[id] == NULL) {
            return 0;
        }
        rt = rt->next[id];
        word++;
    }
    if(rt->signal == true&&rt->vis == false) {
        rt->vis = true;
        printf("OK\n");
        return true;
    }
    if(rt->signal == true&&rt->vis == true) {
        printf("REPEAT\n");
        return true;
    }
    return true;
}

int main() {
    root = build();
    scanf("%d", &n);
    for(int i=1; i<=n; i++) {
        scanf("%s", s);
        insert(s);
    }
    scanf("%d", &m);
    for(int i=1; i<=m; i++) {
        scanf("%s", s);
        if(!find(s)) {
            printf("WRONG\n");
        }
    }
}

作者：wlz

出处：http://www.cnblogs.com/bljfy/p/8698195.html 

本文版权归作者和博客园共有，欢迎转载，但未经作者同意必须保留此段声明，且在文章页面明显位置给出原文连接，否则保留追究法律责任的权利。