結果
問題 | No.778 クリスマスツリー |
ユーザー | htkb |
提出日時 | 2019-01-06 15:32:14 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
TLE
(最新)
AC
(最初)
|
実行時間 | - |
コード長 | 1,579 bytes |
コンパイル時間 | 91 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 76,624 KB |
最終ジャッジ日時 | 2024-05-03 03:13:17 |
合計ジャッジ時間 | 17,501 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 31 ms
11,008 KB |
testcase_01 | AC | 31 ms
10,880 KB |
testcase_02 | AC | 31 ms
10,752 KB |
testcase_03 | AC | 30 ms
10,880 KB |
testcase_04 | AC | 30 ms
10,752 KB |
testcase_05 | AC | 32 ms
11,008 KB |
testcase_06 | AC | 1,550 ms
76,600 KB |
testcase_07 | AC | 1,429 ms
59,472 KB |
testcase_08 | TLE | - |
testcase_09 | TLE | - |
testcase_10 | TLE | - |
testcase_11 | TLE | - |
testcase_12 | TLE | - |
testcase_13 | AC | 1,616 ms
70,280 KB |
testcase_14 | AC | 1,513 ms
76,624 KB |
ソースコード
class BinaryIndexedTree(object): __slots__ = ["tree"] def __init__(self, size: int): self.tree = [0]*(size+1) def add(self, index: int, value: int): tree = self.tree while index < len(tree): tree[index] += value index += index & -index def sum(self, index: int): tree, result = self.tree, 0 while index: result += tree[index] index -= index & -index return result def euler_tour(tree): from collections import deque tree_size, euler_tour_size = len(tree), len(tree)*2-1 edges = [[] for _ in [0]*tree_size] for origin, destination in enumerate(tree): if destination >= 0: edges[destination].append(origin) subtree_range = [[0, 0] for _ in [0]*tree_size] dq, index = deque([0]), 0 pop, extend = dq.pop, dq.extend while index < euler_tour_size: index, v = index+1, pop() if not subtree_range[v][0]: subtree_range[v][0] = index else: subtree_range[v][1] = index continue dq.append(tree[v]) if edges[v]: dq.extend(edges[v]) else: subtree_range[v][1] = index return subtree_range if __name__ == "__main__": N = int(input()) tree = [-1] + list(map(int, input().split())) subtree_range = euler_tour(tree) bit = BinaryIndexedTree(2*N-1) ans = 0 for start, end in subtree_range[::-1]: ans += bit.sum(end) - bit.sum(start-1) bit.add(start, 1) print(ans)