2020牛客寒假算法基础集训营1：nozomi和字符串：字符串贪心

原创于 2020-02-21 09:03:09 发布 · 276 阅读

0 ·

CC 4.0 BY-SA版权

模拟 stl 专栏收录该内容

10 篇文章

订阅专栏

本文介绍了一种优化的字符串处理算法，通过记录1和0的下标，能够快速计算出最大连续相同字符的数量，适用于需要频繁处理二进制字符串的场景。

**[题目链接](https://imgconvert.csdnimg.cn/aHR0cHM6Ly9hYy5ub3djb2Rlci5jb20vYWNtL2NvbnRlc3QvMzAwMi9I?x-oss-process=image/format,png)** 在这里插入图片描述
这里有一个很重要的思想要学会：分别记录1，0下标，方便后续直接处理
AC代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f3f
vector<ll>v1,v0;
int main()
{
 ll n,k;
 cin>>n>>k;
 string s;
 cin>>s;
 v1.push_back(-1);
 v0.push_back(-1);
 for(ll i=0;i<n;i++)
 {
     if(s[i]=='0') v0.push_back(i);
     else v1.push_back(i);
 }
 v1.push_back(n);
 v0.push_back(n);
 ll ans=0;
 if(v1.size()-2<=k) ans=n;
 else
 {
   for(ll i=1,j=k;j<v1.size()-1;i++,j++)
   {
       ans=max(ans,v1[j+1]-v1[i-1]-1);
   }
 }
 if(v0.size()-2<=k) ans=n;
 else
 {
   for(ll i=1,j=k;j<v0.size()-1;i++,j++)
   {
       ans=max(ans,v0[j+1]-v0[i-1]-1);
   }
 }
 cout<<ans<<endl;
}