萌新刚学OI一秒,MLE#11求救
查看原帖
萌新刚学OI一秒,MLE#11求救
220227
我就是天帝楼主2021/7/14 17:02
#include<queue>
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int N=1e3+5,M=5e3+5;
int n,m,tot,Trie[N][M][26],End[N][M];
char s[N];
queue <int> q;
void add(int x)
{
	int p=1,len=strlen(s+1);
	for(int i=1;i<=len;i++)
	{
		int ch=s[i]-'a';
		if(!Trie[x][p][ch]) Trie[x][p][ch]=++tot;
		p=Trie[x][p][ch];
	}
	End[x][p]=1;
}
int find(int x)
{
	int len=strlen(s+1),p=1;
	for(int i=1;i<=len;i++)
	{
		int ch=s[i]-'a';
		if(!Trie[x][p][ch]) return 0;
		p=Trie[x][p][ch];
	}
	return End[x][p];
}
int main()
{
	scanf("%d",&n);
	for(int i=1,L;i<=n;i++)
	{
		tot=1;
		scanf("%d",&L);
		for(int j=1;j<=L;j++)
			scanf("%s",s+1),add(i);
	}
	scanf("%d",&m);
	for(int i=1;i<=m;i++)
	{
		scanf("%s",s+1);
		for(int j=1;j<=n;j++)
			if(find(j)) q.push(j);
		while(q.size())
		{
			printf("%d",q.front());
			q.pop();
			if(q.size()) printf(" ");
		}
		printf("\n");
	}
	return 0;
}

RT , 所以我数组开多大好QAQ

2021/7/14 17:02
加载中...