题解 | 小红的01子序列构造(easy)

小红的01子序列构造(easy)

https://www.nowcoder.com/practice/ee0b6c6baa2642c182df8b4390126f9a

#include<bits/stdc++.h>
#define int long long

using namespace std;

int n,k;
string s;

signed main()
{
    cin>>n>>k;
    cin>>s;
    int l=0,r=0;
    int cnt0=0,cnt1=0;
    int cnt=0;
    bool flag=false;
    while(r<s.size())
    {
        while(r<s.size() && cnt<k)
        {
            if(s[r]=='0') cnt0++;
            if(s[r]=='1') cnt+=cnt0,cnt1++;
            r++;
        }
        if(cnt==k)
        {
            flag=true;
            break;
        }
        while(l<=r && cnt>k)
        {
            if(s[l]=='0') cnt0--,cnt-=cnt1;
            if(s[l]=='1') cnt1--;
            l++;
        }
        if(cnt==k)
        {
            flag=true;
            break;
        }
    }
    if(flag) cout<<l+1<<' '<<r;
    else cout<<-1;
    return 0;
}

全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务