求助!调了一上午也没有调出来问题在哪里
查看原帖
求助!调了一上午也没有调出来问题在哪里
383601
shijihong楼主2021/12/16 12:45
#include <iostream>
using namespace std;
int trie[10000005][2];
int pre[10000005][2],pos;
int end1[10000005];
void insert(string s){
	int p=0;
	for(int i=0;i<s.length();i++){
		int n=s[i]-'0';
		if(!trie[p][n]) trie[p][n]=++pos;
		p=trie[p][n];
		pre[p][n]++;
	}
	end1[p]++;
}
int find(string s){
	int p=0,res=0,ans=0;
	for(int i=0;i<s.length();i++){
		int n=s[i]-'0';
		if(end1[p]) ans++;
		if(!trie[p][n])return ans;
		p=trie[p][n];
		res=pre[p][n];
	}
	return res+ans;
}
int main(){
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		int p;
		string s="";
		cin>>p;
		for(int i=1;i<=p;i++){
			int x;
			cin>>x;
			s+=char(x+'0');
		}
		insert(s);
	}
	for(int i=1;i<=m;i++){
		int p;
		cin>>p;
		string s="";
		for(int i=1;i<=p;i++){
			int x;
			cin>>x;
			s+=char(x+'0');
		}
		cout<<find(s)<<endl;
	}
}
2021/12/16 12:45
加载中...