結果
問題 | No.1687 What the Heck? |
ユーザー | wolgnik |
提出日時 | 2021-09-24 21:43:40 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,248 bytes |
コンパイル時間 | 1,837 ms |
コンパイル使用メモリ | 86,744 KB |
実行使用メモリ | 150,392 KB |
最終ジャッジ日時 | 2023-09-18 21:02:51 |
合計ジャッジ時間 | 18,618 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge12 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 78 ms
71,244 KB |
testcase_01 | AC | 77 ms
71,436 KB |
testcase_02 | AC | 78 ms
71,328 KB |
testcase_03 | AC | 80 ms
71,384 KB |
testcase_04 | AC | 82 ms
71,516 KB |
testcase_05 | AC | 82 ms
71,280 KB |
testcase_06 | AC | 79 ms
71,252 KB |
testcase_07 | AC | 723 ms
101,768 KB |
testcase_08 | AC | 1,000 ms
111,416 KB |
testcase_09 | AC | 721 ms
101,304 KB |
testcase_10 | AC | 504 ms
93,072 KB |
testcase_11 | AC | 545 ms
93,324 KB |
testcase_12 | AC | 1,957 ms
150,300 KB |
testcase_13 | AC | 1,953 ms
150,300 KB |
testcase_14 | AC | 1,994 ms
150,392 KB |
testcase_15 | AC | 1,964 ms
150,260 KB |
testcase_16 | AC | 1,927 ms
149,620 KB |
testcase_17 | AC | 250 ms
82,832 KB |
testcase_18 | WA | - |
testcase_19 | WA | - |
ソースコード
import sys input = sys.stdin.readline N = int(input()) a = list(map(int, input().split())) import heapq hpush = heapq.heappush hpop = heapq.heappop h = [] for i in range(N): hpush(h, (-(i + 1) * 2, a[i] + 1, ~i)) hpush(h, (-(i + 1), a[i], ~i)) class BIT: def __init__(self, n): self.n = n self.data = [0] * (n + 1) self.el = [0] * (n + 1) def sum(self, i): s = 0 while i > 0: s += self.data[i] i -= i & -i return s def add(self, i, x): self.el[i] += x while i <= self.n: self.data[i] += x i += i & -i def get(self, i, j = None): if j is None: return self.el[i] return self.sum(j) - self.sum(i) def lowerbound(self, s): x = 0 y = 0 for i in range(self.n.bit_length(), -1, -1): k = x + (1 << i) if k <= self.n and (y + self.data[k] < s): y += self.data[k] x += 1 << i return x + 1 fwk = BIT(N + 1) for x in range(1, N + 1): fwk.add(x, 1) #print(h) res = -N * (N + 1) // 2 vis = [0] * N while len(h): c, x, i = hpop(h) i = ~i #print(h) if vis[i]: continue c = -c t = fwk.sum(x - 1) y = fwk.lowerbound(t + 1) if y > N: continue vis[i] = 1 fwk.add(y, -1) res += c #print(i, c, res, t, y) print(res)