結果

問題 No.755 Zero-Sum Rectangle
コンテスト
ユーザー brthyyjp
提出日時 2021-02-03 02:23:42
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,988 ms / 2,000 ms
コード長 1,188 bytes
コンパイル時間 248 ms
コンパイル使用メモリ 82,204 KB
実行使用メモリ 76,480 KB
最終ジャッジ日時 2024-06-30 00:07:41
合計ジャッジ時間 16,092 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
import io, os
input = io.BytesIO(os.read(0,os.fstat(0).st_size)).readline

from itertools import accumulate

def main():

    n, m = map(int, input().split())
    s = [[0]*(m+1)]+[[0]+[a for a in accumulate(map(int,input().split()))] for _ in range(m)]

    for j in range(m):
        for i in range(m):
            s[i+1][j+1] += s[i][j+1]

    #print(s)

    imos = [[0]*(m+1) for i in range(m+1)]

    # [i, k), [j, l)
    for i in range(m):
        for j in range(m):
            for k in range(i+1, m+1):
                for l in range(j+1, m+1):
                    temp = s[k][l] - s[k][j] - s[i][l] + s[i][j]
                    if temp == 0:
                        imos[i][j] += 1
                        imos[i][l] -= 1
                        imos[k][j] -= 1
                        imos[k][l] += 1

    for i in range(1, m+1):
        for j in range(m+1):
            imos[i][j] += imos[i-1][j]

    for j in range(1, m+1):
        for i in range(m+1):
            imos[i][j] += imos[i][j-1]

    #print(imos)

    for i in range(n):
        x, y = map(int, input().split())
        x, y = x-1, y-1
        print(imos[x][y])

if __name__ == '__main__':
    main()
0