結果

問題 No.5006 Hidden Maze
ユーザー kyskys
提出日時 2022-06-12 16:28:01
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 890 ms / 2,000 ms
コード長 5,294 bytes
コンパイル時間 980 ms
実行使用メモリ 105,136 KB
スコア 0
平均クエリ数 1000.00
最終ジャッジ日時 2022-06-12 16:29:26
合計ジャッジ時間 84,231 ms
ジャッジサーバーID
(参考情報)
judge14 / judge13
純コード判定しない問題か言語
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 781 ms
103,628 KB
testcase_01 AC 686 ms
101,904 KB
testcase_02 AC 774 ms
103,644 KB
testcase_03 AC 799 ms
103,820 KB
testcase_04 AC 706 ms
102,040 KB
testcase_05 AC 738 ms
102,608 KB
testcase_06 AC 727 ms
103,676 KB
testcase_07 AC 821 ms
103,740 KB
testcase_08 AC 692 ms
102,552 KB
testcase_09 AC 707 ms
101,628 KB
testcase_10 AC 738 ms
103,236 KB
testcase_11 AC 693 ms
101,860 KB
testcase_12 AC 715 ms
103,256 KB
testcase_13 AC 732 ms
103,376 KB
testcase_14 AC 784 ms
103,048 KB
testcase_15 AC 763 ms
103,216 KB
testcase_16 AC 735 ms
103,504 KB
testcase_17 AC 743 ms
103,640 KB
testcase_18 AC 694 ms
102,488 KB
testcase_19 AC 703 ms
101,852 KB
testcase_20 AC 739 ms
102,780 KB
testcase_21 AC 793 ms
103,016 KB
testcase_22 AC 728 ms
105,136 KB
testcase_23 AC 763 ms
103,000 KB
testcase_24 AC 777 ms
103,744 KB
testcase_25 AC 798 ms
103,804 KB
testcase_26 AC 761 ms
102,524 KB
testcase_27 AC 871 ms
104,816 KB
testcase_28 AC 780 ms
103,136 KB
testcase_29 AC 684 ms
101,984 KB
testcase_30 AC 760 ms
103,036 KB
testcase_31 AC 703 ms
102,532 KB
testcase_32 AC 757 ms
102,932 KB
testcase_33 AC 775 ms
103,368 KB
testcase_34 AC 795 ms
104,412 KB
testcase_35 AC 750 ms
103,260 KB
testcase_36 AC 766 ms
103,536 KB
testcase_37 AC 726 ms
102,664 KB
testcase_38 AC 746 ms
103,400 KB
testcase_39 AC 778 ms
103,756 KB
testcase_40 AC 769 ms
104,156 KB
testcase_41 AC 824 ms
103,740 KB
testcase_42 AC 743 ms
102,452 KB
testcase_43 AC 801 ms
103,868 KB
testcase_44 AC 759 ms
102,252 KB
testcase_45 AC 746 ms
103,640 KB
testcase_46 AC 799 ms
104,288 KB
testcase_47 AC 750 ms
104,612 KB
testcase_48 AC 760 ms
102,700 KB
testcase_49 AC 715 ms
104,064 KB
testcase_50 AC 744 ms
103,316 KB
testcase_51 AC 855 ms
104,976 KB
testcase_52 AC 748 ms
102,992 KB
testcase_53 AC 793 ms
104,032 KB
testcase_54 AC 743 ms
102,480 KB
testcase_55 AC 730 ms
102,896 KB
testcase_56 AC 758 ms
102,976 KB
testcase_57 AC 756 ms
104,052 KB
testcase_58 AC 785 ms
104,064 KB
testcase_59 AC 706 ms
102,008 KB
testcase_60 AC 750 ms
103,660 KB
testcase_61 AC 890 ms
104,484 KB
testcase_62 AC 791 ms
102,920 KB
testcase_63 AC 846 ms
103,724 KB
testcase_64 AC 862 ms
104,328 KB
testcase_65 AC 710 ms
103,480 KB
testcase_66 AC 802 ms
103,128 KB
testcase_67 AC 716 ms
102,556 KB
testcase_68 AC 789 ms
102,928 KB
testcase_69 AC 779 ms
103,592 KB
testcase_70 AC 818 ms
103,772 KB
testcase_71 AC 693 ms
102,656 KB
testcase_72 AC 820 ms
103,560 KB
testcase_73 AC 829 ms
104,648 KB
testcase_74 AC 679 ms
103,376 KB
testcase_75 AC 744 ms
102,692 KB
testcase_76 AC 709 ms
101,792 KB
testcase_77 AC 689 ms
102,244 KB
testcase_78 AC 776 ms
103,372 KB
testcase_79 AC 789 ms
103,052 KB
testcase_80 AC 830 ms
104,000 KB
testcase_81 AC 855 ms
104,308 KB
testcase_82 AC 721 ms
101,376 KB
testcase_83 AC 832 ms
104,536 KB
testcase_84 AC 877 ms
104,736 KB
testcase_85 AC 720 ms
101,908 KB
testcase_86 AC 857 ms
104,888 KB
testcase_87 AC 789 ms
105,012 KB
testcase_88 AC 768 ms
103,356 KB
testcase_89 AC 782 ms
103,136 KB
testcase_90 AC 755 ms
103,320 KB
testcase_91 AC 803 ms
104,112 KB
testcase_92 AC 807 ms
103,032 KB
testcase_93 AC 810 ms
103,576 KB
testcase_94 AC 824 ms
103,596 KB
testcase_95 AC 850 ms
104,304 KB
testcase_96 AC 793 ms
102,652 KB
testcase_97 AC 755 ms
102,544 KB
testcase_98 AC 798 ms
103,148 KB
testcase_99 AC 825 ms
103,704 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

from collections import defaultdict, deque
import sys
from math import sqrt

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.50] * W for _ in [0] * H], [[0.50] * W for _ in [0] * H], [[0.50] * W for _ in [0] * H], [[0.50] * W for _ in [0] * H]]
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 = []
    seen = [[False] * W for _ in [0] * H]
    seen[0][0] = True
    for _ in [0] * 400:
        prob_tmp = []
        for i in range(4):
            try:
                dr, dc = direction_coord[i]
                nx = place[0] + dr
                ny = place[1] + dc
                if nx < 0 or nx >= H or ny < 0 or ny >= W or seen[nx][ny]:
                    prob_tmp.append(0)
                else:
                    prob_tmp.append(probability[i][place[0]][place[1]])
            except IndexError:
                print('IndexError', place)
        sum_prob = sum(prob_tmp)
        if sum_prob == 0:
            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
        seen[place[0]][place[1]] = True
        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 = sqrt((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()
0