結果

問題 No.5006 Hidden Maze
ユーザー kyskys
提出日時 2022-06-12 15:56:18
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 684 ms / 2,000 ms
コード長 4,582 bytes
コンパイル時間 287 ms
実行使用メモリ 103,208 KB
スコア 0
平均クエリ数 1000.00
最終ジャッジ日時 2022-06-12 15:59:49
合計ジャッジ時間 66,883 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge14
純コード判定しない問題か言語
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 637 ms
103,152 KB
testcase_01 AC 603 ms
100,668 KB
testcase_02 AC 665 ms
102,344 KB
testcase_03 AC 631 ms
101,088 KB
testcase_04 AC 642 ms
102,096 KB
testcase_05 AC 635 ms
101,968 KB
testcase_06 AC 581 ms
100,928 KB
testcase_07 AC 645 ms
102,376 KB
testcase_08 AC 627 ms
101,608 KB
testcase_09 AC 583 ms
101,064 KB
testcase_10 AC 658 ms
101,568 KB
testcase_11 AC 589 ms
100,988 KB
testcase_12 AC 592 ms
101,392 KB
testcase_13 AC 637 ms
101,508 KB
testcase_14 AC 606 ms
100,840 KB
testcase_15 AC 595 ms
101,156 KB
testcase_16 AC 674 ms
103,172 KB
testcase_17 AC 602 ms
101,196 KB
testcase_18 AC 619 ms
101,492 KB
testcase_19 AC 608 ms
101,336 KB
testcase_20 AC 632 ms
100,704 KB
testcase_21 AC 599 ms
100,928 KB
testcase_22 AC 684 ms
101,864 KB
testcase_23 AC 633 ms
101,664 KB
testcase_24 AC 612 ms
102,432 KB
testcase_25 AC 652 ms
103,152 KB
testcase_26 AC 564 ms
101,164 KB
testcase_27 AC 602 ms
102,268 KB
testcase_28 AC 630 ms
102,376 KB
testcase_29 AC 574 ms
100,264 KB
testcase_30 AC 674 ms
102,280 KB
testcase_31 AC 586 ms
101,968 KB
testcase_32 AC 570 ms
101,332 KB
testcase_33 AC 613 ms
101,780 KB
testcase_34 AC 564 ms
101,548 KB
testcase_35 AC 601 ms
101,456 KB
testcase_36 AC 591 ms
100,760 KB
testcase_37 AC 589 ms
100,972 KB
testcase_38 AC 586 ms
101,552 KB
testcase_39 AC 631 ms
102,776 KB
testcase_40 AC 586 ms
101,728 KB
testcase_41 AC 634 ms
103,156 KB
testcase_42 AC 613 ms
101,808 KB
testcase_43 AC 570 ms
102,644 KB
testcase_44 AC 593 ms
101,060 KB
testcase_45 AC 590 ms
101,184 KB
testcase_46 AC 565 ms
102,272 KB
testcase_47 AC 577 ms
100,248 KB
testcase_48 AC 616 ms
101,756 KB
testcase_49 AC 567 ms
101,132 KB
testcase_50 AC 585 ms
101,364 KB
testcase_51 AC 568 ms
101,064 KB
testcase_52 AC 662 ms
102,780 KB
testcase_53 AC 651 ms
101,812 KB
testcase_54 AC 630 ms
101,432 KB
testcase_55 AC 640 ms
102,136 KB
testcase_56 AC 623 ms
103,208 KB
testcase_57 AC 640 ms
102,228 KB
testcase_58 AC 581 ms
100,588 KB
testcase_59 AC 660 ms
102,784 KB
testcase_60 AC 589 ms
101,016 KB
testcase_61 AC 601 ms
101,516 KB
testcase_62 AC 609 ms
101,336 KB
testcase_63 AC 624 ms
101,452 KB
testcase_64 AC 631 ms
102,240 KB
testcase_65 AC 594 ms
101,616 KB
testcase_66 AC 572 ms
100,828 KB
testcase_67 AC 623 ms
101,620 KB
testcase_68 AC 584 ms
102,148 KB
testcase_69 AC 612 ms
102,076 KB
testcase_70 AC 593 ms
101,604 KB
testcase_71 AC 611 ms
101,276 KB
testcase_72 AC 661 ms
102,272 KB
testcase_73 AC 629 ms
102,068 KB
testcase_74 AC 580 ms
101,804 KB
testcase_75 AC 627 ms
102,288 KB
testcase_76 AC 607 ms
101,212 KB
testcase_77 AC 583 ms
101,956 KB
testcase_78 AC 573 ms
100,916 KB
testcase_79 AC 573 ms
100,660 KB
testcase_80 AC 616 ms
100,776 KB
testcase_81 AC 616 ms
102,788 KB
testcase_82 AC 601 ms
101,836 KB
testcase_83 AC 611 ms
101,728 KB
testcase_84 AC 565 ms
102,332 KB
testcase_85 AC 639 ms
101,532 KB
testcase_86 AC 622 ms
102,856 KB
testcase_87 AC 592 ms
102,508 KB
testcase_88 AC 619 ms
101,444 KB
testcase_89 AC 574 ms
102,212 KB
testcase_90 AC 571 ms
101,144 KB
testcase_91 AC 606 ms
101,352 KB
testcase_92 AC 562 ms
101,916 KB
testcase_93 AC 593 ms
100,988 KB
testcase_94 AC 544 ms
100,996 KB
testcase_95 AC 610 ms
102,240 KB
testcase_96 AC 604 ms
101,532 KB
testcase_97 AC 575 ms
101,992 KB
testcase_98 AC 576 ms
101,680 KB
testcase_99 AC 618 ms
101,848 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict, deque
import sys

input = sys.stdin.readline
def iinput(): return int(input())
def sinput(): return input().rstrip()
def i0input(): return int(input()) - 1
def linput(): return list(input().split())
def liinput(): return list(map(int, input().split()))
def miinput(): return map(int, input().split())
def li0input(): return list(map(lambda x: int(x) - 1, input().split()))
def mi0input(): return map(lambda x: int(x) - 1, input().split())
INF = 10**20
MOD = 1000000007
from random import random, seed
seed(101010)

class UnionFindTree:
    def __init__(self, initial_size:int) -> None:
        self.n_nodes = initial_size
        self.parents = [i for i in range(initial_size)]
        self.ranks = [0 for i in range(initial_size)]
        self.size = [1 for i in range(initial_size)]
        self.n_roots = initial_size
    def root(self, n:int) -> int:
        if self.parents[n] == n:
            return n
        else:
            self.parents[n] = self.root(self.parents[n])
            return self.parents[n]
    def same(self, n:int, m:int) -> bool:
        return (self.root(n) == self.root(m))
    def unite(self, n:int, m:int) -> None:
        if self.same(n, m):
            return
        self.n_roots -= 1
        n, m = self.root(n), self.root(m)
        if self.ranks[n] < self.ranks[m]:
            self.parents[n] = m
            self.size[m] += self.size[n]
        else:
            self.parents[m] = n
            self.size[n] += self.size[m]
            if self.ranks[n] == self.ranks[m]:
                self.ranks[n] += 1
    def get_roots(self) -> set:
        return set([self.root(x) for x in range(self.n_nodes)])
    def count_roots(self) -> int:
        return self.n_roots
    def get_tree_size(self, n:int) -> int:
        return self.size[self.root(n)]

def random_direction(p):
    for i in range(3):
        p[i+1] += p[i]
    r = random()
    for i in range(4):
        if p[i] > r:
            return i

H, W, p = liinput()
# URDL
probability = [[[0.75] * W for _ in [0] * H] for _ in [0] * 4]
p /= 100
direction_inv = {'U': 0, 'R': 1, 'D': 2, 'L': 3}
direction = 'URDL'
direction_coord = [(-1, 0), (0, 1), (1, 0), (0, -1)]
uf = UnionFindTree(H * W)

for i in range(W):
    probability[0][0][i] = 0
    probability[2][-1][i] = 0

for i in range(H):
    probability[1][i][-1] = 0
    probability[3][i][0] = 0

for _ in [0] * 1000:
    place = [0, 0]
    res = []
    for _ in [0] * 400:
        prob_tmp = []
        for i in range(4):
            try:
                prob_tmp.append(probability[i][place[0]][place[1]])
            except IndexError:
                print('IndexError', place)
        sum_prob = sum(prob_tmp)
        for i in range(4):
            prob_tmp[i] /= sum_prob
        d = random_direction(prob_tmp)
        dr, dc = direction_coord[d]
        place[0] += dr
        place[1] += dc
        res.append(direction[d])
    print(''.join(res), flush=True)
    thr = iinput()
    if thr == -1:
        exit()
    place = [0, 0]
    for i in range(thr):
        d = direction_inv[res[i]]
        probability[d][place[0]][place[1]] = 1
        dr, dc = direction_coord[d]
        uf.unite(place[0] * W + place[1], (place[0] + dr) * W + place[1] + dc)
        place[0] += dr
        place[1] += dc
        probability[(d+2)%4][place[0]][place[1]] = 1
    if len(res) > thr:
        d = direction_inv[res[thr]]
        q = probability[d][place[0]][place[1]]
        if q < 1:
            new_p = (p * q) / (p * q + 1 - q)
            probability[d][place[0]][place[1]] = new_p
            dr, dc = direction_coord[d]
            place[0] += dr
            place[1] += dc
            probability[(d+2)%4][place[0]][place[1]] = new_p
    if uf.same(0, H*W-1):
        seen = [[None] * W for _ in [0] * H]
        seen[0][0] = ''
        que = deque()
        que.append((0, 0))
        while que:
            x, y = que.popleft()
            for i in range(4):
                dx, dy = direction_coord[i]
                nx = x + dx
                ny = y + dy
                if nx < 0 or nx >= H or ny < 0 or ny >= W or not uf.same(x * W + y, nx * W + ny):
                    continue
                if seen[nx][ny] is not None:
                    continue
                seen[nx][ny] = seen[x][y] + direction[i]
                que.append((nx, ny))
                if nx == H-1 and ny == W-1:
                    print(seen[nx][ny], flush=True)
                    g = iinput()
                    if g == -1:
                        exit()
        for s in seen:
            print(s)
0