求优化
查看原帖
求优化
418419
ko_no_lzx_da楼主2022/1/4 13:14
#include<iostream>
#include<cstring>
#include<string>
#include<cstdio>
#include<queue>
#include<cmath>
#include<algorithm>
using namespace std;
struct node{
	long long ch[25];
	long long val[1000],v;
}a[100000];
long long top=1;
void add(string s,long long num){
	long long p=1;
	for(long long j=0;j<s.length();j++){
		long long ch=s[j]-'a';
		//cout <<"---"<<ch<<"---"<<endl;
		if(!a[p].ch[ch])a[p].ch[ch]=++top;
		p=a[p].ch[ch];
	}
	a[p].val[a[p].v++]=num;
}
void find(string s){
	long long p=1;
	for(long long j=0;j<s.length();j++){
		long long ch=s[j]-'a';
		if(!a[p].ch[ch])return;
		p=a[p].ch[ch];
	}
	for(long long j=0;j<a[p].v;j++){
		cout <<a[p].val[j]<<" ";
	}
}
long long n,m;
int main(){
	cin >>n;
	for(long long i=1;i<=n;i++){
		long long len;	
		cin >>len;
		for(long long j=0;j<len;j++){
			string s;
			cin >>s;
			add(s,i);
		}
	}
	cin >>m;
	for(long long j=0;j<m;j++){
		string s;
		cin >>s;
		find(s);
		cout <<endl;
	}
	return 0;
}


2022/1/4 13:14
加载中...