結果

問題 No.307 最近色塗る問題多くない？
ユーザー mkawa2
提出日時 2020-02-09 11:07:07
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 1,449 bytes
コンパイル時間 98 ms
コンパイル使用メモリ 13,056 KB
実行使用メモリ 18,724 KB
最終ジャッジ日時 2024-10-01 05:47:55
合計ジャッジ時間 5,977 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 7 TLE * 1 -- * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys

sys.setrecursionlimit(10 ** 6)

def II(): return int(sys.stdin.readline())
def MI(): return map(int, sys.stdin.readline().split())
def LI(): return list(map(int, sys.stdin.readline().split()))
def SI(): return sys.stdin.readline()[:-1]
def LLI(rows_number): return [LI() for _ in range(rows_number)]
int1 = lambda x: int(x) - 1
def MI1(): return map(int1, sys.stdin.readline().split())
def LI1(): return list(map(int1, sys.stdin.readline().split()))
p2D = lambda x: print(*x, sep="\n")
dij = [(0, 1), (1, 0), (0, -1), (-1, 0)]

def main():
    def bfs(si, sj, x):
        stack=[(si,sj)]
        while stack:
            i,j=stack.pop()
            for di, dj in dij:
                ni, nj = i + di, j + dj
                if ni < 0 or nj < 0 or ni >= h or nj >= w: continue
                if aa[ni][nj] == x: continue
                aa[ni][nj] = x
                cnt[x]+=1
                cnt[1-x]-=1
                stack.append((ni,nj))

    h, w = MI()
    aa = LLI(h)
    cnt = [0] * 2
    for i in range(h):
        for j in range(w):
            cnt[aa[i][j]] += 1
    q=II()
    rcx=LLI(q)
    for r,c,x in rcx:
        r, c = r - 1, c - 1
        if aa[r][c] == x: continue
        aa[r][c] = x
        bfs(r, c, x)
        if cnt[0]*cnt[1]==0:
            x=rcx[-1][-1]
            for i in range(h):
                for j in range(w):
                    aa[i][j]=x
            break
    for row in aa:print(*row)

main()
0