package main import ( "bufio" "fmt" "math/bits" "os" "strconv" ) func configure(scanner *bufio.Scanner) { scanner.Split(bufio.ScanWords) scanner.Buffer(make([]byte, 1000005), 1000005) } func getNextString(scanner *bufio.Scanner) string { scanned := scanner.Scan() if !scanned { panic("scan failed") } return scanner.Text() } func getNextInt(scanner *bufio.Scanner) int { i, _ := strconv.Atoi(getNextString(scanner)) return i } func getNextInt64(scanner *bufio.Scanner) int64 { i, _ := strconv.ParseInt(getNextString(scanner), 10, 64) return i } func getNextFloat64(scanner *bufio.Scanner) float64 { i, _ := strconv.ParseFloat(getNextString(scanner), 64) return i } func main() { fp := os.Stdin wfp := os.Stdout extra := 0 if os.Getenv("I") == "IronMan" { fp, _ = os.Open(os.Getenv("END_GAME")) extra = 100 } scanner := bufio.NewScanner(fp) configure(scanner) writer := bufio.NewWriter(wfp) defer func() { r := recover() if r != nil { fmt.Fprintln(writer, r) } writer.Flush() }() solve(scanner, writer) for i := 0; i < extra; i++ { fmt.Fprintln(writer, "-----------------------------------") solve(scanner, writer) } } func solve(scanner *bufio.Scanner, writer *bufio.Writer) { h := getNextInt(scanner) w := getNextInt(scanner) k := getNextInt(scanner) sq := makeGrid(h, w) for y := 0; y < h; y++ { s := getNextString(scanner) for x := 0; x < w; x++ { sq[y][x].b[0] = sq[y][x].b[0] | uint32(1< lx { ly := lx - i c := types(ly, lx, rx-lx, sq) if c < k+lw { break } lx++ } lowers[ry-1][rx-1][lw] = lx ans += int64(lowers[ry-1][rx-1][0] - lowers[ry-1][rx-1][lw]) } } } for i := 1; i < h; i++ { for lw := 0; lw < 2; lw++ { ly := i for ry := i + 1; ry < h+1; ry++ { rx := ry - i for ry > ly { lx := ly - i c := types(ly, lx, ry-ly, sq) if c < k+lw { break } ly++ } lowers[ry-1][rx-1][lw] = ly ans += int64(lowers[ry-1][rx-1][0] - lowers[ry-1][rx-1][lw]) } } } fmt.Fprintln(writer, ans) } func types(y, x, m int, sq [][]square) int { z := 32 - bits.LeadingZeros32(uint32(m)) zz := 1 << uint(z-1) b := sq[y][x].b[z-1] | sq[y+m-zz][x].b[z-1] | sq[y][x+m-zz].b[z-1] | sq[y+m-zz][x+m-zz].b[z-1] c := bits.OnesCount32(b) return c } func min(a, b int) int { if a < b { return a } return b } type square struct { b [16]uint32 } func makeGrid(h, w int) [][]square { index := make([][]square, h, h) data := make([]square, h*w, h*w) for i := 0; i < h; i++ { index[i] = data[i*w : (i+1)*w] } return index }