[Jzoj] 3843. 寻找羔羊

最新推荐文章于 2019-11-07 12:02:40 发布

原创最新推荐文章于 2019-11-07 12:02:40 发布 · 337 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种算法，用于在给定的小写字母字符串中查找并计数所有不重复的'agnus'子串。通过遍历字符串并利用标记数组，该算法能高效地计算出每个'agnus'子串前后的字符数量，从而得出总的子串数目。

题目描述

给定一个由小写字母组成的字符串，寻找包含“agnus”（羔羊）的子串的个数。注意：当且仅当两个子串的起始位置和终点不同时，这两个子串属于不同的子串。

题目解析

从头到尾遍历一次字符串，每碰到一次 $“ a g n u s ”$ ，用 $“ a g n u s ”$ 前面的字符数目乘上 $“ a g n u s ”$ 后面的字符数目，即为含有当前 $“ a g n u s ”$ 的字符串数目。

但因为要去重，所以第 $i$ 个 $“ a g n u s ”$ 前面字符数目需要减去第 $i - 1$ 个 $“ a g n u s ”$ 中 $“ a ”$ 的前面的字符数目

代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;
string s; 
ll ans;
int len,ls;
bool f[30005];
int main()
{	
	cin>>s;
	len=s.size();
	for(int i=0;i<len-4;i++)
	 if(s[i]=='a'&&s[i+1]=='g'&&s[i+2]=='n'&&s[i+3]=='u'&&s[i+4]=='s')
	  f[i]=f[i+1]=f[i+2]=f[i+3]=f[i+4]=1;
	for(int i=0;i<=len;i++)
	 if(f[i])
	 {
	   i+=4;
	   ans+=(ls+1)*(len-i);
	   ls=4;
	 }
	 else ls++;
	printf("%lld",ans);
}