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

题目大意:

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


思路:

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


代码:

T r i e Trie Trie
#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 map map
#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;
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值