結果

問題 No.3756 Udon Network
ユーザー まぬお
提出日時 2026-10-09 18:56:13
言語 PyPy3
(7.3.23 + ACL)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
TLE  
実行時間 -
コード長 1,733 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 63 ms
コンパイル使用メモリ 83,284 KB
実行使用メモリ 277,928 KB
最終ジャッジ日時 2026-10-09 18:57:11
合計ジャッジ時間 52,546 ms
ジャッジサーバーID
（参考情報） 		judge5_0 / judge4_1
このコードへのチャレンジ
（要ログイン）
サブタスク 配点 結果
Example 0 % AC * 8
Subtask $1$ 2 % AC * 15
Subtask $2$ 4 % AC * 22
Subtask $3$ 8 % AC * 9
Subtask $4$ 16 % AC * 10
Subtask $5$ 32 % AC * 10
Subtask $6$ 38 % AC * 52 TLE * 1
合計 4 * 62% = 248 点
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

from collections import deque, defaultdict, Counter
from bisect import bisect_left, bisect_right, insort
from itertools import permutations, combinations, groupby
from heapq import heappop, heappush
import math, sys
input = lambda: sys.stdin.readline().rstrip("\r\n")
def printl(li, sep=" "): print(sep.join(map(str, li)))
def yn(flag): print(Yes if flag else No)
_int = lambda x: int(x)-1
MOD = 998244353 #10**9+7
INF = 1<<60
Yes, No = "Yes", "No"

N, M, Q = map(int, input().split())
A = list(map(int, input().split()))
q = []
for _ in range(M):
    u, v, w = map(int, input().split())
    u, v = u-1, v-1
    q.append((w, u, v))
QI = [[] for _ in range(N)]

ans = [-1]*Q
for qi in range(Q):
    s, c = map(int, input().split())
    if c == 1:
        ans[qi] = 0
    else:
        heappush(QI[s-1], (c, qi))

q.sort()

S = [set([A[i]]) for i in range(N)]
from atcoder.dsu import DSU
D = DSU(N)
for w, u, v in q:
    if D.same(u, v): continue
    pu = D.leader(u)
    pv = D.leader(v)
    su = S[pu]
    sv = S[pv]
    qu = QI[pu]
    qv = QI[pv]
    D.merge(u, v)
    if len(su) > len(sv):
        while sv:
            a = sv.pop()
            su.add(a)
        rs = su
    else:
        while su:
            a = su.pop()
            sv.add(a)
        rs = sv
    if len(qu) > len(qv):
        while qv:
            a = heappop(qv)
            heappush(qu, a)
        rq = qu
    else:
        while qu:
            a = heappop(qu)
            heappush(qv, a)
        rq = qv
    p = D.leader(u)
    S[p] = rs
    QI[p] = rq
    # print(w, u, v, rs, rq)
    while rq:
        c, qi = heappop(rq)
        if len(rs) >= c:
            ans[qi] = w
        else:
            heappush(rq, (c, qi))
            break
printl(ans, "\n")
0