結果

問題 No.179 塗り分け
ユーザー hang_hang_cln
提出日時 2018-06-09 11:03:05
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 2,207 bytes
コンパイル時間 123 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 11,008 KB
最終ジャッジ日時 2024-06-30 12:25:45
合計ジャッジ時間 6,519 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4 WA * 2
other AC * 29 WA * 11
権限があれば一括ダウンロードができます

ソースコード

diff # 

H, W = map(int,input().split())
    
field = []
for h in range(H):
    field.append(list(input()))

def painting(dx, dy,black_cnt):
    # ある平行移動量で塗れるか判定
    paintCnt = [0, 0] # 赤または青で塗った枚数
    canPaintByThiWay = True # この平行移動量で塗り分け可能か
    paint = [[0 for w in range(W)] for h in range(H)] # 0:塗ってない 1:赤 2:青
    for h in range(H):
        for w in range(W):
            if field[h][w] == '#' and paint[h][w] == 0: # 赤で塗ることができる
                paint[h][w] = 1 # 赤で塗る
                paintCnt[0] += 1
                # 平行移動先に青でも塗ってみる
                if h+dy >= H or w+dx >= W: # フィールドの外に移動しようとしていたら
                    return False
                elif field[h+dy][w+dx] == '#' and paint[h+dy][w+dx] == 0: # 青で塗ることができる
                    paint[h+dy][w+dx] = 2 # 青で塗る
                    paintCnt[1] += 1
                else: # 青で塗ることが出来ない = 失敗
                    return False # この平行移動量ではむり
            else: # 赤で塗ることができない = 失敗とは限らない
                if paint[h][w] == 2: # 青ですでに塗られていたら
                    return False # この平行移動量ではむり

            if paintCnt[0]==paintCnt[1] and sum(paintCnt) == black_cnt:
                return True

def check():
    black_cnt = 0
    for h in range(H):
        for w in range(W):
            #print(field[h][w])
            if field[h][w] == '#':
                black_cnt += 1
                
    else:
        # 平行移動量を決める
        for dx in range(W):
            for dy in range(H):
                if dx == 0 and dy == 0:
                    canPaint = False
                    continue
                
                if painting(dx,dy,black_cnt):
                    canPaint = True
                    break
                else:
                    canPaint = False
            if canPaint:
                break
        
        return canPaint

if check():
    print('YES')
else:
    print('NO')
0