結果

問題 No.1607 Kth Maximum Card
ユーザー あかしあみどりあかしあみどり
提出日時 2023-02-20 22:39:21
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 2,450 ms / 3,500 ms
コード長 2,040 bytes
コンパイル時間 389 ms
コンパイル使用メモリ 82,592 KB
実行使用メモリ 257,592 KB
最終ジャッジ日時 2024-07-21 12:51:11
合計ジャッジ時間 25,906 ms
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 39 ms
52,960 KB
testcase_01 AC 37 ms
52,876 KB
testcase_02 AC 38 ms
53,264 KB
testcase_03 AC 37 ms
52,004 KB
testcase_04 AC 37 ms
53,560 KB
testcase_05 AC 39 ms
52,588 KB
testcase_06 AC 42 ms
53,308 KB
testcase_07 AC 39 ms
52,556 KB
testcase_08 AC 1,616 ms
134,988 KB
testcase_09 AC 1,028 ms
126,128 KB
testcase_10 AC 2,450 ms
155,336 KB
testcase_11 AC 199 ms
89,840 KB
testcase_12 AC 1,255 ms
128,352 KB
testcase_13 AC 189 ms
84,224 KB
testcase_14 AC 255 ms
86,828 KB
testcase_15 AC 1,020 ms
128,856 KB
testcase_16 AC 239 ms
85,616 KB
testcase_17 AC 110 ms
78,464 KB
testcase_18 AC 771 ms
114,912 KB
testcase_19 AC 431 ms
95,792 KB
testcase_20 AC 593 ms
105,192 KB
testcase_21 AC 652 ms
107,164 KB
testcase_22 AC 1,600 ms
150,340 KB
testcase_23 AC 1,599 ms
149,828 KB
testcase_24 AC 584 ms
94,748 KB
testcase_25 AC 275 ms
87,792 KB
testcase_26 AC 431 ms
90,028 KB
testcase_27 AC 584 ms
94,688 KB
testcase_28 AC 511 ms
91,172 KB
testcase_29 AC 1,336 ms
114,648 KB
testcase_30 AC 2,155 ms
140,964 KB
testcase_31 AC 1,239 ms
109,204 KB
testcase_32 AC 386 ms
172,400 KB
testcase_33 AC 383 ms
166,504 KB
testcase_34 AC 639 ms
257,592 KB
testcase_35 AC 492 ms
208,384 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

def oi(): return int(input())
def os(): return input()
def mi(): return list(map(int, input().split()))

import sys
input = sys.stdin.readline
# import sys
# sys.setrecursionlimit(10**8)
# import pypyjit
# pypyjit.set_param('max_unroll_recursion=-1')
input_count = 0

input_count = 0
N,M,K = mi()
NODE = N
LOOP_NUM = M
maps = {i:[] for i in range(NODE)}
cost_dict = []
for i in range(LOOP_NUM):
    a,b,c = mi()
    maps[a-1].append((b-1, i))
    maps[b-1].append((a-1,i))
    cost_dict.append(c)
# 以下の問題にてテスト済み
# https://atcoder.jp/contests/abc213/tasks/abc213_e
def zero_one_dfs(target, K):
    zero_deq = [0]
    count=0
    use_node = [0] * M

    while True:
        one_deq = []
        
        while zero_deq:
            node = zero_deq.pop()
                
            # 目的地に到達したら
            if node == N-1:
                if count < K:
                    return True

            # コスト0の時の移動パターン
            for next_node, ind in maps[node]:
                if use_node[ind] == 0:
                    use_node[ind] = -1
                    if cost_dict[ind] > target:
                        one_deq.append(next_node)
                    else:
                        zero_deq.append(next_node)

        count+=1
        zero_deq = one_deq[:]
        if count >= K:
            return False
def is_ok(arg):
    # 条件を満たすかどうか?問題ごとに定義
    # 初めてFalseからTrueになるTrueの地点を取得
    return zero_one_dfs(arg, K)

def meguru_bisect(ng, ok):
    '''
    初期値のng,okを受け取り,is_okを満たす最小(最大)のokを返す
    まずis_okを定義すべし
    ng ok は  とり得る最小の値-1 とり得る最大の値+1
    最大最小が逆の場合はよしなにひっくり返す
    '''
    while (abs(ok - ng) > 1):
        mid = (ok + ng) // 2
        if is_ok(mid):
            ok = mid
        else:
            ng = mid
    return ok
mins = meguru_bisect(-1, 2*10**5 + 1)
print(mins)
0