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

package main

import (
    "fmt"
)

func main() {
    var n, k int
    fmt.Scan(&n, &k)

    cnt0 := 0
    cnt1 := 0
    l := 0
    r := 0
    
    var s string
    fmt.Scan(&s)
    num := 0
    for l < n && r < n {
        if num == k {
            l = l+1
            break
        }
        if num < k {
            if s[r] == '0' {
                cnt0++
            } else {
                num += cnt0
                cnt1++
            }
            r++
        } else {
            if s[l] == '0' {
                num -= cnt1
                cnt0--
            } else {
                cnt1--
            }
            l++
        }

    }
    if num != k {
        fmt.Println(-1)
    } else {
        fmt.Printf("%d %d", l, r)
    }
}