結果
| 問題 | No.899 γatheree |
| ユーザー |
 mkawa2
|
| 提出日時 | 2022-04-05 13:28:49 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 1,393 ms / 2,000 ms |
| コード長 | 7,932 bytes |
| コンパイル時間 | 173 ms |
| コンパイル使用メモリ | 82,560 KB |
| 実行使用メモリ | 120,960 KB |
| 最終ジャッジ日時 | 2024-05-04 23:13:14 |
| 合計ジャッジ時間 | 24,601 ms |
|
ジャッジサーバーID （参考情報） |
judge5 / judge2 |
（要ログイン）
テストケース
テストケース表示| 入力 | 結果 | 実行時間 実行使用メモリ |
|---|---|---|
| testcase_00 | AC | 47 ms
55,808 KB |
| testcase_01 | AC | 107 ms
76,288 KB |
| testcase_02 | AC | 107 ms
76,160 KB |
| testcase_03 | AC | 96 ms
75,776 KB |
| testcase_04 | AC | 108 ms
77,056 KB |
| testcase_05 | AC | 107 ms
76,672 KB |
| testcase_06 | AC | 1,173 ms
118,160 KB |
| testcase_07 | AC | 1,159 ms
118,236 KB |
| testcase_08 | AC | 1,123 ms
118,080 KB |
| testcase_09 | AC | 1,145 ms
117,936 KB |
| testcase_10 | AC | 1,169 ms
118,076 KB |
| testcase_11 | AC | 1,207 ms
118,204 KB |
| testcase_12 | AC | 1,137 ms
117,556 KB |
| testcase_13 | AC | 1,117 ms
117,176 KB |
| testcase_14 | AC | 1,115 ms
117,652 KB |
| testcase_15 | AC | 1,154 ms
114,088 KB |
| testcase_16 | AC | 1,171 ms
117,948 KB |
| testcase_17 | AC | 1,161 ms
117,964 KB |
| testcase_18 | AC | 1,137 ms
118,008 KB |
| testcase_19 | AC | 1,112 ms
117,808 KB |
| testcase_20 | AC | 1,164 ms
117,548 KB |
| testcase_21 | AC | 1,106 ms
120,832 KB |
| testcase_22 | AC | 1,393 ms
108,568 KB |
| testcase_23 | AC | 1,085 ms
120,960 KB |
ソースコード
import sys
# sys.setrecursionlimit(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 << 63)-1
# inf = (1 << 31)-1
# md = 10**9+7
md = 998244353
class LazySegTree:
def __init__(self, op, e, mapping, composition, _id, v):
self._op = op
self._e = e
self._mapping = mapping
self._composition = composition
self._id = _id
if isinstance(v, int):
v = [e]*v
self._n = len(v)
self._log = (self._n-1).bit_length()
self._size = 1 << self._log
self._d = [self._e]*(2*self._size)
self._lz = [self._id]*self._size
for i in range(self._n):
self._d[self._size+i] = v[i]
for i in range(self._size-1, 0, -1):
self._update(i)
def set(self, p, x):
p += self._size
for i in range(self._log, 0, -1):
self._push(p >> i)
self._d[p] = x
for i in range(1, self._log+1):
self._update(p >> i)
def get(self, p):
p += self._size
for i in range(self._log, 0, -1):
self._push(p >> i)
return self._d[p]
def prod(self, left, right):
if left == right:
return self._e
left += self._size
right += self._size
for i in range(self._log, 0, -1):
if ((left >> i) << i) != left:
self._push(left >> i)
if ((right >> i) << i) != right:
self._push(right >> i)
sml = self._e
smr = self._e
while left < right:
if left & 1:
sml = self._op(sml, self._d[left])
left += 1
if right & 1:
right -= 1
smr = self._op(self._d[right], smr)
left >>= 1
right >>= 1
return self._op(sml, smr)
def all_prod(self):
return self._d[1]
def apply(self, left, right, f):
if right is None:
p = left
p += self._size
for i in range(self._log, 0, -1):
self._push(p >> i)
self._d[p] = self._mapping(f, self._d[p])
for i in range(1, self._log+1):
self._update(p >> i)
else:
if left == right:
return
left += self._size
right += self._size
for i in range(self._log, 0, -1):
if ((left >> i) << i) != left:
self._push(left >> i)
if ((right >> i) << i) != right:
self._push((right-1) >> i)
l2 = left
r2 = right
while left < right:
if left & 1:
self._all_apply(left, f)
left += 1
if right & 1:
right -= 1
self._all_apply(right, f)
left >>= 1
right >>= 1
left = l2
right = r2
for i in range(1, self._log+1):
if ((left >> i) << i) != left:
self._update(left >> i)
if ((right >> i) << i) != right:
self._update((right-1) >> i)
def max_right(self, left, g):
if left == self._n:
return self._n
left += self._size
for i in range(self._log, 0, -1):
self._push(left >> i)
sm = self._e
first = True
while first or (left & -left) != left:
first = False
while left%2 == 0:
left >>= 1
if not g(self._op(sm, self._d[left])):
while left < self._size:
self._push(left)
left *= 2
if g(self._op(sm, self._d[left])):
sm = self._op(sm, self._d[left])
left += 1
return left-self._size
sm = self._op(sm, self._d[left])
left += 1
return self._n
def min_left(self, right, g):
if right == 0:
return 0
right += self._size
for i in range(self._log, 0, -1):
self._push((right-1) >> i)
sm = self._e
first = True
while first or (right & -right) != right:
first = False
right -= 1
while right > 1 and right%2:
right >>= 1
if not g(self._op(self._d[right], sm)):
while right < self._size:
self._push(right)
right = 2*right+1
if g(self._op(self._d[right], sm)):
sm = self._op(self._d[right], sm)
right -= 1
return right+1-self._size
sm = self._op(self._d[right], sm)
return 0
def _update(self, k):
self._d[k] = self._op(self._d[2*k], self._d[2*k+1])
def _all_apply(self, k, f):
self._d[k] = self._mapping(f, self._d[k])
if k < self._size:
self._lz[k] = self._composition(f, self._lz[k])
def _push(self, k):
self._all_apply(2*k, self._lz[k])
self._all_apply(2*k+1, self._lz[k])
self._lz[k] = self._id
# treeのマージ
def op(x, y):
return x+y
# treeの単位元
e = 0
# lazy(f)からtree(x)への操作
def mapping(f, x):
if f == -1: return x
ln = x%(1 << 24)
return f*ln << 24 | ln
# lazyの下への分解
def composition(f, g):
if f == -1: return g
return f
# lazyの単位元
_id = -1
# 配列の初期値(サイズか配列そのもの)が最後
# seg = LazySegTree(op, e, mapping, composition, _id, n)
# seg = LazySegTree(op, e, mapping, composition, _id, aa)
from collections import deque
n = II()
to = [[] for _ in range(n)]
for _ in range(n-1):
u, v = LI()
to[u].append(v)
to[v].append(u)
aa = LI()
par = [-1]*n
q = deque()
q.append(0)
uu = []
ll = [[n]*n for _ in range(3)]
rr = [[-1]*n for _ in range(3)]
while q:
u = q.popleft()
ll[0][u] = len(uu)
rr[0][u] = len(uu)+1
uu.append(u)
for v in to[u]:
if v == par[u]: continue
par[v] = u
q.append(v)
# pDB(len(uu),uu)
for u in uu[::-1]:
p = par[u]
if p == -1: continue
ll[1][p] = min(ll[1][p], ll[0][u])
rr[1][p] = max(rr[1][p], rr[0][u])
ll[2][p] = min(ll[2][p], ll[1][u])
rr[2][p] = max(rr[2][p], rr[1][u])
# pDB(ll)
# pDB(rr)
st = LazySegTree(op, e, mapping, composition, _id, [aa[u] << 24 | 1 for u in uu])
for _ in range(II()):
u = II()
s = 0
l, r = ll[2][u], rr[2][u]
if r != -1:
s += st.prod(l, r) >> 24
st.apply(l, r, 0)
# pDB(_, l, r)
l, r = ll[1][u], rr[1][u]
if r != -1:
s += st.prod(l, r) >> 24
st.apply(l, r, 0)
# pDB(_, l, r)
p = par[u]
if p != -1:
s += st.get(ll[0][p]) >> 24
# st.set(ll[0][p], 1)
st.apply(ll[0][p], None, 0)
l, r = ll[1][p], rr[1][p]
if r != -1:
s += st.prod(l, r) >> 24
st.apply(l, r, 0)
p = par[p]
if p != -1:
s += st.get(ll[0][p]) >> 24
# st.set(ll[0][p], 1)
st.apply(ll[0][p], None, 0)
else:
s += st.get(ll[0][u]) >> 24
# st.set(ll[0][u], s << 24 | 1)
st.apply(ll[0][u], None, s)
print(s)