結果

問題 No.5002 stick xor
ユーザー GrayCoder
提出日時 2018-05-29 01:34:50
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 1,096 bytes
コンパイル時間 23,832 ms
実行使用メモリ 5,972 KB
スコア 24,502
最終ジャッジ日時 2018-05-29 01:35:16
ジャッジサーバーID
(参考情報)
judge9 /
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 21 WA * 11
権限があれば一括ダウンロードができます

ソースコード

diff #

def main():
    grid = [[0] * (N + 2)]
    apnd = grid.append
    for n in A:
        apnd([0] + n + [0])
    apnd([0] * (N + 2))

    for l in L:
        case1 = search(l, grid)
        _, row1, col1, row2, col2 = case1
        for i in range(row1, row2 + 1):
            line = grid[i][col1:col2]
            grid[i][col1:col2] = list(map(lambda x: int(not x), line))
        print(row1, col1, row2, col2 - 1)

def search(l, grid):
    maxb = -1
    for i in range(1, N + 1):
        for j in range(1, N + 2 - l):
            black = grid[i][j:j+l].count(1)
            if l == 1 and black == 0 and grid[i][j-1] == grid[i][j+1] == 1:
                maxb = 0
                return (maxb, i, j, i, j)
            elif black == l and grid[i][j-1] == grid[i][j+l] == 0:
                maxb = black
                return (maxb, i, j, i, j + l)
            elif maxb < black:
                maxb = black
                ret = (maxb, i, j, i, j + l)
    return ret

N, K = map(int, input().split())
L = tuple(map(int, input().split()))
A = [list(map(int, list(input()))) for _ in [0] * N]
main()
0