D - Boring count HDU - 5056

記柔刀

于 2019-03-12 20:19:25 发布

阅读量82

点赞数

分类专栏：题目

本文链接：https://blog.csdn.net/qq_43813493/article/details/88428245

版权

题目专栏收录该内容

2 篇文章 0 订阅

订阅专栏

从后向前找满足条件的最大子串
因为当满足条件的最大字串为i
那么里面公有i个满足条件的子子串所以只需记录从后向前找的满足条件的最大子串
如果找到了没满足条件的那个数那么这个数找到第k次出现的位置数在向前移一位则满足条件
ac代码如下
运用了vector数组来存储每个字母的个数以用来判断

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<string>
//#include<bits/stdc++.h>
using namespace std;
int main()
{
    int t,k;
    string s;
    scanf("%d",&t);
    while(t--)
    {
        cin>>s;
        scanf("%d",&k);
        int lastpos = 0;
        long long count_ = 0;
        vector<int>letter[26];
        int len = s.size();
        for(int i = 0; i < len; ++i)
        {
            int l = s[i] -'a';
            letter[l].push_back(i);
            int sz = letter[l].size();
            if(sz > k)
                lastpos = max(lastpos,letter[l][sz-k-1]+1);
        count_ = count_ + i-lastpos+1;
        }
        cout<<count_<<'\n';
        for(int i = 0;i < 26;++i)
        letter[i].clear();

    }
    return 0;
}