結果
問題 | No.2618 除霊 |
ユーザー |
![]() |
提出日時 | 2024-01-27 15:29:15 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,835 bytes |
コンパイル時間 | 337 ms |
コンパイル使用メモリ | 82,188 KB |
実行使用メモリ | 163,576 KB |
最終ジャッジ日時 | 2024-09-28 09:41:27 |
合計ジャッジ時間 | 11,508 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 13 TLE * 1 -- * 29 |
ソースコード
import sysinput = sys.stdin.readlineclass Fenwick_Tree:def __init__(self, n):self._n = nself.data = [0] * ndef add(self, p, x):assert 0 <= p < self._np += 1while p <= self._n:self.data[p - 1] += xp += p & -pdef sum(self, l, r):assert 0 <= l <= r <= self._nreturn self._sum(r) - self._sum(l)def _sum(self, r):s = 0while r > 0:s += self.data[r - 1]r -= r & -rreturn sdef get_size(self, x):x = self.find(x)while r > 0:s += self.data[r - 1]r -= r & -rreturn sdef get(self, k):k += 1x, r = 0, 1while r < self._n:r <<= 1len = rwhile len:if x + len - 1 < self._n:if self.data[x + len - 1] < k:k -= self.data[x + len - 1]x += lenlen >>= 1return xN = int(input())G = [[] for i in range(N)]for i in range(N - 1):u, v = map(int, input().split())u, v = u - 1, v - 1G[u].append(v)G[v].append(u)M = int(input())V = list(map(int, input().split()))for i in range(M):V[i] -= 1for i in range(N):G[i].append(i)V = set(V)T = Fenwick_Tree(N)cnt = [0] * Nfor i in V:for u in G[i]:cnt[u] += 1if cnt[u] == 1:T.add(u, 1)for i in range(N):for u in G[i]:if u in V:for v in G[u]:cnt[v] -= 1if cnt[v] == 0:T.add(v, -1)print(T.sum(0, N))for u in G[i]:if u in V:for v in G[u]:cnt[v] += 1if cnt[v] == 1:T.add(v, 1)