JZOJ 3843. 寻找羔羊(agnus)

题目

Description

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

 

Input

只有一个字符串，表示题中所述的字符串。

Output

仅一个数字，表示满足题意的子串个数。

 

Sample Input

agnusbgnus

Sample Output

6
【样例解释】
    6个子串分别是：agnus、agnusb、agnusbg、agnusbgn、agnusbgnu、agnusbgnus。

 

Data Constraint

对于 40%的数据，字符串长度<=1000
对于 100%的数据，字符串长度<=30000

分析

 

• 首先，显然算一个前面*后面
• 但是我们需要去重
• 直接记录上一个的位置
• 少算就OK

代码

#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define ll long long
using namespace std;
int main()
{
    int ans=0,last=-1;
    string s;
    cin>>s;
    for (int i=0;i<=s.size()-4;i++)
    {
        if(s[i]=='a'&&s[i+1]=='g'&&s[i+2]=='n'&&s[i+3]=='u'&&s[i+4]=='s')
        {
            if (last!=-1) ans+=(i-last)*(s.size()-i-4);
            else ans+=(i+1)*(s.size()-i-4);
            last=i;
        } 
    } 
    cout<<ans;
    return 0;
}