题解 | 小红的01子序列构造（easy）

#include <iostream>
#include <string>

using namespace std;
int cnt[3];

int main() {
    int n;
    long long k;
    cin>>n>>k;
    string s;
    cin>>s;
    long long res = 0;
    int j = 0;
    int l = 0, r = 0;
    for(int i = 0;i < n;i++){
        if(s[i] == '1'){
            res += cnt[0];
            cnt[1]++;
            // cout<<i<<" "<<cnt[1]<<"\n";
        }else{
            cnt[0]++;
            // cout<<i<<" "<<cnt[0]<<"\n";
        }
        while(res > k){
            if(s[j] == '1'){
                cnt[1]--;
                // cout<<i<<" "<<cnt[1]<<"\n";
            }else{
                cnt[0]--;
                res -= cnt[1];
                // cout<<i<<" "<<cnt[0]<<"\n";
            }
            j++;
        }
        if(res == k){
            l = j + 1;
            r = i + 1;
            break;
        }
    }
    if(l == 0 && r == 0){
        cout<<-1;
    }else{
        cout<<l<<" "<<r;
    }
}
// 64 位输出请用 printf("%lld")