結果

問題 No.755 Zero-Sum Rectangle
ユーザー brthyyjp
提出日時 2021-02-03 02:36:03
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,970 ms / 2,000 ms
コード長 1,081 bytes
コンパイル時間 256 ms
コンパイル使用メモリ 82,264 KB
実行使用メモリ 76,508 KB
最終ジャッジ日時 2024-06-30 00:10:32
合計ジャッジ時間 15,349 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff #

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):
                    if s[k][l] - s[k][j] - s[i][l] + s[i][j] == 0:
                        imos[i][j] += 1
                        imos[i][l] -= 1
                        imos[k][j] -= 1
                        imos[k][l] += 1

    for i in range(m+1):
        for j in range(m):
            imos[i][j+1] += imos[i][j]
    for j in range(m+1):
        for i in range(m):
            imos[i+1][j] += imos[i][j]
    #print(imos)

    ans = []
    for i in range(n):
        x, y = map(int, input().split())
        ans.append(imos[x-1][y-1])
    print(*ans, sep='\n')

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