提出 #76461946

---

ソースコード 拡げる

import java.io.*;
import java.util.*;
public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        int H = Integer.parseInt(st.nextToken());
        int W = Integer.parseInt(st.nextToken());
        int K = Integer.parseInt(st.nextToken());
        int[][] grid = new int[H][W];
        for (int i = 0; i < H; i++) {
            String s = br.readLine();
            for (int j = 0; j < W; j++) {
                grid[i][j] = s.charAt(j) - '0';
            }
        }
        long count = 0;
        int[] freq = new int[H * W + 1];
        for (int r1 = 0; r1 < H; r1++) {
            int[] colSums = new int[W];
            for (int r2 = r1; r2 < H; r2++) {
                Arrays.fill(freq, 0);
                freq[0] = 1;
                int currentSum = 0;
                for (int c = 0; c < W; c++) {
                    colSums[c] += grid[r2][c];
                    currentSum += colSums[c];
                    if (currentSum >= K) {
                        count += freq[currentSum - K];
                    }
                    freq[currentSum]++;
                }
            }
        }
        
        System.out.println(count);
    }
}

提出情報

ジャッジ結果