結果

問題 No.86 TVザッピング(2)
コンテスト
ユーザー Mao-beta
提出日時 2024-03-05 21:37:32
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 49 ms / 5,000 ms
コード長 2,481 bytes
コンパイル時間 170 ms
コンパイル使用メモリ 82,596 KB
実行使用メモリ 58,384 KB
最終ジャッジ日時 2025-06-20 13:56:05
合計ジャッジ時間 2,563 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
import math
import bisect
from heapq import heapify, heappop, heappush
from collections import deque, defaultdict, Counter
from functools import lru_cache
from itertools import accumulate, combinations, permutations, product

sys.setrecursionlimit(1000000)
MOD = 10 ** 9 + 7
MOD99 = 998244353

input = lambda: sys.stdin.readline().strip()
NI = lambda: int(input())
NMI = lambda: map(int, input().split())
NLI = lambda: list(NMI())
SI = lambda: input()
SMI = lambda: input().split()
SLI = lambda: list(SMI())
EI = lambda m: [NLI() for _ in range(m)]


def main():
    H, W = NMI()
    A = [SI() for _ in range(H)]
    total = sum(a.count(".") for a in A)
    # DRUL
    DH = [1, 0, -1, 0]
    DW = [0, 1, 0, -1]

    INF = 10**5
    for sh in range(H):
        for sw in range(W):
            if A[sh][sw] == "#":
                continue
            steps = [[-INF]*W for _ in range(H)]
            h, w = sh, sw
            R = 0
            d = 0
            steps[sh][sw] = 1
            while True:
                # print(h, w, d)
                rd = (d-1) % 4
                nrh, nrw = h+DH[rd], w+DW[rd]
                if 0 <= nrh < H and 0 <= nrw < W:
                    if A[nrh][nrw] == ".":
                        if R == 1:
                            print("NO")
                            return
                        else:
                            R += 1
                            d = rd

                nh, nw = h+DH[d], w+DW[d]
                if 0 <= nh < H and 0 <= nw < W:
                    if A[nh][nw] == "#":
                        d = (d+1) % 4
                else:
                    d = (d+1) % 4

                nh, nw = h+DH[d], w+DW[d]
                # print(nh, nw)
                if 0 <= nh < H and 0 <= nw < W:
                    if nh == sh and nw == sw:
                        # print(*steps, sep="\n")

                        if steps[h][w] == total:
                            print("YES")
                            return
                        else:
                            print("NO")
                            return
                    elif steps[nh][nw] >= 0:
                        print("NO")
                        return
                    else:
                        steps[nh][nw] = steps[h][w] + 1
                        h = nh
                        w = nw

                else:
                    print("NO")
                    return


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