結果
問題 | No.1975 Zigzag Sequence |
ユーザー | tktk_snsn |
提出日時 | 2022-06-10 22:37:08 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 515 ms / 2,000 ms |
コード長 | 1,802 bytes |
コンパイル時間 | 170 ms |
コンパイル使用メモリ | 81,480 KB |
実行使用メモリ | 108,952 KB |
最終ジャッジ日時 | 2023-10-21 05:28:38 |
合計ジャッジ時間 | 11,296 ms |
ジャッジサーバーID (参考情報) |
judge10 / judge9 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 43 ms
55,344 KB |
testcase_01 | AC | 44 ms
55,344 KB |
testcase_02 | AC | 45 ms
55,344 KB |
testcase_03 | AC | 156 ms
78,032 KB |
testcase_04 | AC | 142 ms
78,316 KB |
testcase_05 | AC | 386 ms
93,144 KB |
testcase_06 | AC | 322 ms
96,636 KB |
testcase_07 | AC | 150 ms
78,620 KB |
testcase_08 | AC | 280 ms
88,668 KB |
testcase_09 | AC | 442 ms
99,724 KB |
testcase_10 | AC | 309 ms
93,072 KB |
testcase_11 | AC | 346 ms
93,444 KB |
testcase_12 | AC | 159 ms
78,824 KB |
testcase_13 | AC | 44 ms
55,348 KB |
testcase_14 | AC | 45 ms
55,348 KB |
testcase_15 | AC | 44 ms
55,348 KB |
testcase_16 | AC | 44 ms
55,348 KB |
testcase_17 | AC | 44 ms
55,348 KB |
testcase_18 | AC | 277 ms
100,516 KB |
testcase_19 | AC | 275 ms
100,516 KB |
testcase_20 | AC | 322 ms
92,608 KB |
testcase_21 | AC | 318 ms
92,608 KB |
testcase_22 | AC | 360 ms
108,952 KB |
testcase_23 | AC | 362 ms
108,952 KB |
testcase_24 | AC | 414 ms
108,688 KB |
testcase_25 | AC | 418 ms
108,688 KB |
testcase_26 | AC | 402 ms
91,528 KB |
testcase_27 | AC | 515 ms
99,976 KB |
testcase_28 | AC | 402 ms
108,424 KB |
testcase_29 | AC | 405 ms
108,240 KB |
testcase_30 | AC | 422 ms
108,160 KB |
testcase_31 | AC | 415 ms
108,160 KB |
testcase_32 | AC | 285 ms
99,196 KB |
testcase_33 | AC | 284 ms
99,196 KB |
testcase_34 | AC | 370 ms
94,764 KB |
testcase_35 | AC | 387 ms
94,760 KB |
ソースコード
from collections import defaultdict mod = 10 ** 9 + 7 class FenwickTree(object): def __init__(self, n): self.n = n self.log = n.bit_length() self.data = [0] * n def __sum(self, r): s = 0 while r > 0: s += self.data[r - 1] s %= mod r -= r & -r return s def add(self, p, x): """ a[p] += xを行う""" p += 1 while p <= self.n: self.data[p - 1] += x self.data[p - 1] %= mod p += p & -p def sum(self, l, r): """a[l] + a[l+1] + .. + a[r-1]を返す""" return self.__sum(r) - self.__sum(l) def lower_bound(self, x): """a[0] + a[1] + .. a[i] >= x となる最小のiを返す""" if x <= 0: return -1 i = 0 k = 1 << self.log while k: if i + k <= self.n and self.data[i + k - 1] < x: x -= self.data[i + k - 1] i += k k >>= 1 return i def __repr__(self): res = [self.sum(i, i+1) for i in range(self.n)] return " ".join(map(str, res)) N = int(input()) A = list(map(int, input().split())) B = sorted(set(A)) d = defaultdict(list) for i, a in enumerate(A): d[a].append(i) two = [1] * N for i in range(1, N): two[i] = two[i-1] * 2 % mod rev = two[:][::-1] ans = 0 L = FenwickTree(N) R = FenwickTree(N) for a in B: for i in d[a]: ans += L.sum(0, i) * R.sum(i, N) ans %= mod for i in d[a]: L.add(i, two[i]) R.add(i, rev[i]) L = FenwickTree(N) R = FenwickTree(N) for a in B[::-1]: for i in d[a]: ans += L.sum(0, i) * R.sum(i, N) ans %= mod for i in d[a]: L.add(i, two[i]) R.add(i, rev[i]) print(ans)