[Trie树][落谷]P2580 于是他错误的点名开始了

首先是题面.落谷\(Link\)

解析

就是简单的Trie树模板, 由于为了进度, 就不写Trie树的插入, 搜索等操作了.
这里就是每次搜的时候, 要搜得单词没有在树上, 就是 \(WRONG\) ; 要是搜的搜完了, 都在树上, 就看最后一个节点的附加信息, 如果是初始化时的0, 就是没喊过名, 然后就是 \(OK\) , 再给最后一个节点上一个附加信息 \(1\), 表示已经喊过了. 如果都搜到了, 但是这个最后一个节点的附加信息是 \(1\) , 也就是这个名字已经叫过了, 就是 \(REPEAT\) .
代码里面有注释, 详细的请看代码部分

Code

#include <bits/stdc++.h>
#define ll long long
#define N 1000005
using namespace std;

int n, m;
//分别是题目中的n, m
struct Trie
{
	ll tag, val[N], st[N][30]; 
        //tag是节点号; val是附加信息; st是树的储存, 因为只有小写字母(26位), 所以二维就存30足矣, 这个主要看题目规定
	Trie () // 初始化结构体
	{
		tag = 1;
		memset (st[0], 0, sizeof (st[0]));
		memset (val, 0, sizeof (val));
	}
	void insert (char *s) //插入操作, *s是指针用法的(跟某位大佬学的), 相当于 char s[](留空就是根据给入的字符串长度来定义) 
	{
		ll hhd = 0, len = strlen (s + 1);
		for (ll i = 1; i <= len; ++ i)
		{
			ll c = s[i] - 'a'; //转换, 便于存储(下面的一样)
			if (!st[hhd][c]) //要是没有就插入
			{
				memset (st[tag], 0, sizeof (st[tag]));
				st[hhd][c] = tag ++;
			}
			hhd = st[hhd][c];
		}
	}
	ll search (char *s)
	{
        // 关于返回的数值, 可以看主函数的判断
		ll hhd = 0, len = strlen (s + 1);
		for (ll i = 1; i <= len; ++ i)
		{
			ll c = s[i] - 'a';
			if (!st[hhd][c]) return 0; //没搜到就返回0 (WRONG)
			hhd = st[hhd][c];
		}
		if (!val[hhd])
		{
			val[hhd] = 1; //没读过就返回1并标记读过
			return 1;
		}
		return 2; //读过的情况
	}
}tree; 

int main ()
{
	char s[N];
	scanf ("%d", &n);
	for (int i = 1; i <= n; ++ i)
	{
		scanf ("%s", s + 1);
		tree.insert (s);
	}
	scanf ("%d", &m);
	for (int i = 1; i <= m; ++ i)
	{
		scanf ("%s", s + 1);
		ll ans = tree.search (s);
		if (ans == 0) printf ("WRONG\n");
	    else if (ans == 1) printf ("OK\n");
		else printf ("REPEAT\n");  
	}
	return 0;
}