AtCoder Beginner Contest 124 D - Handstand（思维+前缀和）

Posted 2021-02-16 qieqiemin

D - Handstand

---

Time Limit: 2 sec / Memory Limit: 1024 MB

Score : $400400$ points

Problem Statement

$\mathrm{NN}$ people are arranged in a row from left to right.

You are given a string $\mathrm{SS}$ of length $\mathrm{NN}$ consisting of 0 and 1, and a positive integer $\mathrm{KK}$.

The $\mathrm{ii}$-th person from the left is standing on feet if the $\mathrm{ii}$-th character of $\mathrm{SS}$ is 0, and standing on hands if that character is 1.

You will give the following direction at most $\mathrm{KK}$ times (possibly zero):

Direction: Choose integers $\mathrm{ll}$ and $\mathrm{rr}$ satisfying $\mathrm{1\le l\le r\le N1\le l\le r\le N}$, and flip the $\mathrm{ll}$-th, $(l+1)(l+1)$-th, $......$, and $\mathrm{rr}$-th persons. That is, for each $\mathrm{i=l,l+1,...,ri=l,l+1,...,r}$, the $\mathrm{ii}$-th person from the left now stands on hands if he/she was standing on feet, and stands on feet if he/she was standing on hands.

Find the maximum possible number of consecutive people standing on hands after at most $\mathrm{KK}$ directions.

Constraints

• $\mathrm{NN}$ is an integer satisfying $\mathrm{1\le N\le 1051\le N\le 105}$.
• $\mathrm{KK}$ is an integer satisfying $\mathrm{1\le K\le 1051\le K\le 105}$.
• The length of the string $\mathrm{SS}$ is $\mathrm{NN}$.
• Each character of the string $\mathrm{SS}$ is 0 or 1.

---

Input

Input is given from Standard Input in the following format:

$\mathrm{NN}$ $\mathrm{KK}$
$\mathrm{SS}$

Output

Print the maximum possible number of consecutive people standing on hands after at most $\mathrm{KK}$ directions.

---

Sample Input 1 Copy

Copy

5 1
00010

Sample Output 1 Copy

Copy

4

We can have four consecutive people standing on hands, which is the maximum result, by giving the following direction:

• Give the direction with $\mathrm{l=1,r=3l=1,r=3}$, which flips the first, second and third persons from the left.

---

Sample Input 2 Copy

Copy

14 2
11101010110011

Sample Output 2 Copy

Copy

8

---

Sample Input 3 Copy

Copy

1 1
1

Sample Output 3 Copy

Copy

1

No directions are necessary.

 

题意：

给你一个只含有0和1的字符串，并且给你一个数字K，你可以选择最多K次区间，每一个区间L<=R，然后对这个区间的元素进行取反操作。

即0变成1，1变成0，问你在最聪明的操作之后最大可以获得的连续1的串是多长？

 

可以看样例解释理解题意。

 

思路：

我们对字符串的连续1和0串进行计数处理，即把连续的1或者0的个数记录起来，

那么字符串会生成如下的数组

例如字符串是  110000111001101

我们把连续的1和连续0分别放入a和b数组，

那么a的元素是2 3 2 1

       b的元素是4 2 1

我们思考可以发现，我们想要最长的连续1串，那么我们处理的区间肯定是连续的0区间，让他们变成1，然后来增长连续1串的长度。

即处理的0区间一定是连续的。

那么我们可以知道如下，例如我们处理两个区间，那么可以的得到的最长的连续1串就是这两个区间的0串长度的sum和以及这两个0串前后和中间的1串的sum和。

那么我们只需要枚举连续的K个的0串即可，

细节见代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <vector>
#include <iomanip>
#define ALL(x) (x).begin(), (x).end()
#define rt return
#define dll(x) scanf("%I64d",&x)
#define xll(x) printf("%I64d
",x)
#define sz(a) int(a.size())
#define all(a) a.begin(), a.end()
#define rep(i,x,n) for(int i=x;i<n;i++)
#define repd(i,x,n) for(int i=x;i<=n;i++)
#define pii pair<int,int>
#define pll pair<long long ,long long>
#define gbtb ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
#define MS0(X) memset((X), 0, sizeof((X)))
#define MSC0(X) memset((X), ‘‘, sizeof((X)))
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define eps 1e-6
#define gg(x) getInt(&x)
#define db(x) cout<<"== [ "<<x<<" ] =="<<endl;
using namespace std;
typedef long long ll;
ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
ll lcm(ll a,ll b){return a/gcd(a,b)*b;}
ll powmod(ll a,ll b,ll MOD){ll ans=1;while(b){if(b%2)ans=ans*a%MOD;a=a*a%MOD;b/=2;}return ans;}
inline void getInt(int* p);
const int maxn=100010;
const int inf=0x3f3f3f3f;
/*** TEMPLATE CODE * * STARTS HERE ***/
int n;
int k;
char s[maxn];
std::vector<int> v1,v2;
ll sum1[maxn];
ll sum2[maxn];
int main()
{
    //freopen("D:\common_text\code_stream\in.txt","r",stdin);
    //freopen("D:\common_text\code_stream\out.txt","w",stdout);
    gbtb;
    cin>>n>>k;
    cin>>s;
    int cnt=0;
//    int i=0;
    if(s[0]==‘0‘)
    {
        v1.pb(0);
    }
    repd(i,0,n-1)
    {
        if(s[i]==‘0‘)
        {
            while(s[i]==‘0‘)
            {
                cnt++;
                i++;
            }
            v2.push_back(cnt);
            cnt=0;
            i--;
        }
        if(s[i]==‘1‘)
        {
            while(s[i]==‘1‘)
            {
                cnt++;
                i++;
            }
            v1.push_back(cnt);
            cnt=0;
            i--;
        }

    }
    int num=max(sz(v1),sz(v2));
    repd(i,1,maxn-2)
    {
        v1.push_back(0);
        v2.push_back(0);
    }
    repd(i,1,maxn-2)
    {
        sum1[i]+=sum1[i-1]+v1[i-1];
        sum2[i]+=sum2[i-1]+v2[i-1];
    }
    int w=0;
    ll ans=0ll;
    repd(i,1,n-k+4)
    {
        int l=i;
        int r=l+k;
        ll temp=sum1[r]-sum1[l-1];
        temp+=sum2[r-1]-sum2[l-1];
        ans=max(ans,temp);
    }
    cout<<ans<<endl;



    return 0;
}

inline void getInt(int* p) {
    char ch;
    do {
        ch = getchar();
    } while (ch == ‘ ‘ || ch == ‘
‘);
    if (ch == ‘-‘) {
        *p = -(getchar() - ‘0‘);
        while ((ch = getchar()) >= ‘0‘ && ch <= ‘9‘) {
            *p = *p * 10 - ch + ‘0‘;
        }
    }
    else {
        *p = ch - ‘0‘;
        while ((ch = getchar()) >= ‘0‘ && ch <= ‘9‘) {
            *p = *p * 10 + ch - ‘0‘;
        }
    }
}