結果
問題 |
No.3025 Chocol∀te
|
ユーザー |
![]() |
提出日時 | 2025-02-14 23:50:43 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 2,627 bytes |
コンパイル時間 | 353 ms |
コンパイル使用メモリ | 82,448 KB |
実行使用メモリ | 139,564 KB |
最終ジャッジ日時 | 2025-02-14 23:51:36 |
合計ジャッジ時間 | 37,051 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge7 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 28 WA * 49 |
ソースコード
import sys # sys.setrecursionlimit(200005) # sys.set_int_max_str_digits(200005) int1 = lambda x: int(x)-1 pDB = lambda *x: print(*x, end="\n", file=sys.stderr) p2D = lambda x: print(*x, sep="\n", end="\n\n", file=sys.stderr) def II(): return int(sys.stdin.readline()) def LI(): return list(map(int, sys.stdin.readline().split())) def LLI(rows_number): return [LI() for _ in range(rows_number)] def LI1(): return list(map(int1, sys.stdin.readline().split())) def LLI1(rows_number): return [LI1() for _ in range(rows_number)] def SI(): return sys.stdin.readline().rstrip() dij = [(0, 1), (-1, 0), (0, -1), (1, 0)] # dij = [(0, 1), (-1, 0), (0, -1), (1, 0), (1, 1), (1, -1), (-1, 1), (-1, -1)] # inf = -1-(-1 << 31) inf = -1-(-1 << 62) # md = 10**9+7 md = 998244353 n, m = LI() r = round(n**0.5) to = [set() for _ in range(n)] for _ in range(m): u, v = LI1() to[u].add(v) to[v].add(u) aa = LI() ss = [0]*n tob = [set() for _ in range(n)] for u in range(n): if len(to[u]) > r: for v in to[u]: tob[v].add(u) else: for v in to[u]: ss[v] += aa[u] for _ in range(II()): t, *uv = LI() if t == 1: u, v = uv[0]-1, uv[1]-1 if v in to[u]: to[u].discard(v) to[v].discard(u) tob[u].discard(v) tob[v].discard(u) if len(to[u]) == r: for k in to[u]: tob[k].discard(u) ss[k] += aa[u] if len(to[v]) == r: for k in to[v]: tob[k].discard(v) ss[k] += aa[v] else: to[u].add(v) to[v].add(u) if len(to[u])>r+1: tob[v].add(u) if len(to[v])<=r:ss[u]+=aa[v] if len(to[v])>r+1: tob[u].add(v) if len(to[u]) <= r: ss[v] += aa[u] if len(to[u]) == r+1: ss[u] = 0 for k in to[u]: tob[k].add(u) if len(to[k]) <= r: ss[u] += aa[k] if len(to[v]) == r+1: ss[v] = 0 for k in to[v]: tob[k].add(v) if len(to[k]) <= r: ss[v] += aa[k] elif t == 2: u, a = uv[0]-1, uv[1] d = a-aa[u] aa[u] = a if len(to[u]) <= r: for v in tob[u]: ss[v] += d else: u = uv[0]-1 ans = 0 if len(to[u]) > r: ans += ss[u] for v in tob[u]: ans += aa[v] else: for v in to[u]: ans += aa[v] print(ans) # print(t,uv,to,tob,ss,aa)