Codeforces 271D Good Substrings 暴力+Trie

最新推荐文章于 2019-08-25 15:30:54 发布

Masamiiiii

最新推荐文章于 2019-08-25 15:30:54 发布

阅读量470

点赞数

CC 4.0 BY-SA版权

分类专栏： CODEFORCES_ 字符串算法

本文链接：https://blog.youkuaiyun.com/Jeremy1149/article/details/70916568

CODEFORCES_ 同时被 2 个专栏收录

232 篇文章

订阅专栏

字符串算法

7 篇文章

订阅专栏

本文介绍了一种使用Trie树优化字符串匹配算法的方法，针对特定问题场景（bad字符限制），通过优化重复子串检测过程，实现了从O(n^2logn)到O(n^2)的时间复杂度降低。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

点击打开链接

题意:string s长度<=1500,字母表a中1代表bad,问s中有多少个不同的substring(s[l]~s[r])满足bad字符不超过k个?
枚举substring 开头直到右端点R非法 map标记重复,O(n^2logn) TLE了
在标记重复上进行优化,用Trie保存所有的substring,枚举开头,遍历Trie,若该结点没访问过,标记后ans++即可 O(n^2)

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e3+20;
string s,a;
int b[N],k;
int ch[N*N][30],val[N*N];//trie
int root,num;
void insert(int l,int r)
{
	int u=0;
	for(int i=l;i<r;i++)
	{
		int id=s[i]-'a';
		if(!ch[u][id])
			ch[u][id]=++num;
		u=ch[u][id];
		val[u]=1;
	}
	val[u]=1;//
}
int query(int l,int r)
{
	int res=0,cnt=0;
	int u=0;
	for(int i=l;i<r;i++)
	{
		int id=s[i]-'a';
		u=ch[u][id];
		if(a[id]=='0')
			cnt++;
		if(cnt<=k)
			res+=val[u],val[u]=0;//val[u]清零 
		else
			return res;	
	}
	return res;
}
int main()
{
 	while(cin>>s>>a>>k)
 	{
 		int ans=0;
 		memset(ch,0,sizeof(ch));
		num=0;
 		int len=s.length();
 		for(int i=0;s[i];i++)//把所有的substring插入到Trie 
 			insert(i,len); 	
		for(int i=0;s[i];i++)//O(n^2)
			ans+=query(i,len);
		cout<<ans<<endl;
	}
	return 0;
}