【洛谷P2580】于是他错误的点名开始了【Trie】

最新推荐文章于 2024-09-20 12:18:53 发布

原创最新推荐文章于 2024-09-20 12:18:53 发布 · 317 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#洛谷 #Trie #map

字符串同时被 2 个专栏收录

9 篇文章

订阅专栏

Trie

6 篇文章

订阅专栏

本文介绍了一种使用Trie树和map解决字符串匹配问题的方法。针对一组预定义的字符串集合，当遇到新的字符串时，能够判断该字符串是否曾出现过、是否为首次出现，还是完全未出现。提供两种实现方式：一种基于Trie树，另一种则利用STL map。

题目大意：

题目链接：https://www.luogu.org/problemnew/show/P2580
先给出 $n$ 个字符串，之后再给出 $m$ 个串 $s$ ，若 $s [i]$ 在那 $n$ 个字符串中出现过且在 $m$ 串中第一次出现，输出"OK"，如果出现在 $n$ 串单在 $m$ 串中出现过，输出"REPEAT"，如果没有出现在 $n$ 串输出 $W R O N G$ 。

思路：

这道题正解是 $T r i e$ ，当然如果你想用 $S T L$ 的 $m a p$ 做也没有人拦你。
首先将前 $n$ 个串插入到 $T r i e$ 里，之后读入的 $m$ 个串就在字典树里查找，如果找到并且是第一次出现，那么就返回"OK"，否则返回"REPEAT"，如果根本找不到或就返回"WRONG"。
用 $m a p$ 的话就将前 $n$ 个串放进数组，然后每读入一个串就在数组里查找。跟字典树思路是基本一样的。

代码：

$T r i e$ ：

#include <cstdio>
#include <string>
#include <iostream>
#define N 300000
using namespace std;

int n,trie[N+1][31],num;
string read;
bool end[N+1],sum[N+1];

void Insert(string s)  //插入
{
	int p=1;
	for (int i=0;i<s.size();i++)
	{
		if (!trie[p][s[i]-'a'+1])  //没有这个点
		 trie[p][s[i]-'a'+1]=++num;  //新建节点
		p=trie[p][s[i]-'a'+1];  //继续搜
	}
	end[p]=true;  //记录
	return;
}

string find (string s)  //查找
{
	int p=1;
	for (int i=0;i<s.size();i++)
	 if (!(p=trie[p][s[i]-'a'+1])) return "WRONG";  //找不到这个串
	if (!end[p]) return "WRONG";  //找到但是没有在这里结束就退出
	  //就像n串有一个abcde，m串有一个abc，你能在Trie里找到abc，但是实际上是abcde的一部分，不算找到。
	if (!sum[p])  //找到且为第一次
	{
		sum[p]=true;
		return "OK";
	}
	return "REPEAT";  //多次，之前找到过
}

int main()
{
	scanf("%d\n",&n);
	for (int i=1;i<=n;i++)
	{
		cin>>read;
		Insert(read);
	}
	scanf("%d\n",&n);
	for (int i=1;i<=n;i++)
	{
		cin>>read;
		cout<<find(read)<<endl;
	}
	return 0;
}

$m a p$ ：

#include <cstdio>
#include <string>
#include <map>
#include <iostream>
using namespace std;

string s;
int n;
map<string,int> a;  //map

int main()
{
    cin>>n;
    for (int i=1;i<=n;i++)
    {
        cin>>s;
        a[s]=1;  //记录，类似hash
    }
    cin>>n;
    for (int i=1;i<=n;i++)
    {
        cin>>s;
        if (a[s]==1)
        {
            puts("OK");
            a[s]=2;
        }
        else if (a[s]==2)
        {
            puts("REPEAT");
        }
        else puts("WRONG");
    }
    return 0;
}