結果

問題 No.2888 Mamehinata
ユーザー norioc
提出日時 2024-11-24 07:09:04
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,459 bytes
コンパイル時間 388 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 142,096 KB
最終ジャッジ日時 2024-11-24 07:09:30
合計ジャッジ時間 21,639 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 45 WA * 7
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

from collections import defaultdict, deque
class FenwickTree:
def __init__(self, n):
self.data = [0] * (n+10)
self.n = (n+10)
def add(self, p, x):
assert 0 <= p < self.n
p += 1
while p < len(self.data):
self.data[p] += x
p += p & -p
def sum(self, p):
""" [0, p] """
assert 0 <= p < self.n
p += 1
s = 0
while p > 0:
s += self.data[p]
p -= p & -p
return s
def rangesum(self, l, r):
""" [l, r] """
assert 0 <= l <= r < self.n
s = self.sum(r)
if l > 0:
s -= self.sum(l-1)
return s
N, M = map(int, input().split())
adj = defaultdict(list)
for _ in range(M):
u, v = map(lambda x: int(x)-1, input().split())
adj[u].append(v)
adj[v].append(u)
def bfs():
evens = FenwickTree(N+10)
odds = FenwickTree(N+10)
q = deque([(0, 0)]) # (node, dist)
used = set()
while q:
v, d = q.popleft()
if v in used: continue
used.add(v)
if d % 2 == 0:
evens.add(d, 1)
else:
odds.add(d, 1)
for to in adj[v]:
if to in used: continue
q.append((to, d+1))
return evens, odds
evens, odds = bfs()
for i in range(1, N+1):
if i % 2 == 0:
print(evens.rangesum(0, i))
else:
print(odds.rangesum(0, i))
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0