AtCoder AGC 038 A - 01 Matrix (Go) - あるエンジニアのAtCoder奮闘記

irisruneです。やはりAGCはB問題が安定しないと手が出し辛いです。

問題

いつものAGC-Aらしく数学パズル的な問題です。嘘解法ではないはずですが実行時間1978msでした。

package main

import (
    "bufio"
    "fmt"
    "os"
    "strconv"
)

var sc *bufio.Scanner

func nextStr() string {
    sc.Scan()
    return sc.Text()
}

func nextInt() int {
    sc.Scan()
    i, e := strconv.Atoi(sc.Text())
    if e != nil {
        panic(e)
    }
    return i
}

func main() {
    sc = bufio.NewScanner(os.Stdin)
    sc.Buffer(make([]byte, 0), 1000000001*3)
    sc.Split(bufio.ScanWords)
    h, w, a, b := nextInt(), nextInt(), nextInt(), nextInt()

    for hi := 0; hi < b; hi++ {
        for wi := 0; wi < a; wi++ {
            fmt.Print("1")
        }
        for wi := a; wi < w; wi++ {
            fmt.Print("0")
        }
        fmt.Println("")
    }
    for hi := b; hi < h; hi++ {
        for wi := 0; wi < a; wi++ {
            fmt.Print("0")
        }
        for wi := a; wi < w; wi++ {
            fmt.Print("1")
        }
        fmt.Println("")
    }
}

入力例1に対しての出力例1がミスリードになっています。 $A,B$ は各行、各列について0,1の少ない方の数となるので、 $B=1$ のときには1列目に1を2個（0を1個）配置することができます。同様に $A=1$ のときには3行目に1を2個（0を1個）配置することができるので、以下のような出力が可能です。