제출 #1304799

#제출 시각아이디문제언어결과실행 시간메모리
1304799Faisal_SaqibSelling RNA Strands (JOI16_selling_rna)C++20
0 / 100
279 ms325608 KiB
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
const int M=2e6+10,N=1e5+10;
string s[N];
int nxt[2][M][26];
vector<int> cur[2][M];
int lst[10];
void insert(string&s,int ind,bool p=0)
{
	int rt=0;
	cur[p][rt].push_back(ind);
	for(int i=0;i<s.size();i++)
	{
		if(!nxt[p][rt][s[i]-'a'])nxt[p][rt][s[i]-'a']=++lst[p];
		rt=nxt[p][rt][s[i]-'a'];
		cur[p][rt].push_back(ind);
	}
}
pair<int,int> rg(string& s)
{
	bool p=0;
	int rt=0;
	for(int i=0;i<s.size();i++)
	{
		if(!nxt[p][rt][s[i]-'a']) return {1,0};
		rt=nxt[p][rt][s[i]-'a'];
	}
	return {*begin(cur[p][rt]),*rbegin(cur[p][rt])};
}
int count(string& s,int ql,int qr)
{
	bool p=1;
	int rt=0;
	for(int i=0;i<s.size();i++)
	{
		if(!nxt[p][rt][s[i]-'a']) return 0;
		rt=nxt[p][rt][s[i]-'a'];
	}
	return upper_bound(begin(cur[p][rt]),end(cur[p][rt]),qr)-lower_bound(begin(cur[p][rt]),end(cur[p][rt]),ql);
}
int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	int n,m;
	cin>>n>>m;
	for(int i=0;i<n;i++)cin>>s[i];
	sort(s,s+n);
	for(int i=0;i<n;i++)
	{
		insert(s[i],i);
		reverse(begin(s[i]),end(s[i]));
		insert(s[i],i,1);
	}
	for(int j=0;j<m;j++)
	{
		string p,q;
		cin>>p>>q;
		auto rag=rg(p);
		cout<<count(q,rag.first,rag.second)<<endl;
	}
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...