题解 | 小红的排列构造

package main

import (
    "fmt"
)

func main() {
    var n int
    fmt.Scan(&n)
    var s string
    fmt.Scan(&s)

    if s[n-1] != '1' {
        fmt.Println("-1")
        return
    }

    ans := make([]int, n)
    k := 0

    for i := 0; i < n; i++ {
        if s[i] == '1' {
            for j:=i; j >=0; j-- {
                if ans[j] == 0 {
                    k++
                    ans[j] = k
                } else {
                    break
                }
            }
        }
    }

    for _, v := range ans {
        fmt.Printf("%d ", v)
    }
}