結果
問題 | No.1975 Zigzag Sequence |
ユーザー | lilictaka |
提出日時 | 2022-06-12 03:09:14 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 507 ms / 2,000 ms |
コード長 | 3,033 bytes |
コンパイル時間 | 1,261 ms |
コンパイル使用メモリ | 81,796 KB |
実行使用メモリ | 112,080 KB |
最終ジャッジ日時 | 2023-10-23 19:42:29 |
合計ジャッジ時間 | 14,958 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge15 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 41 ms
55,592 KB |
testcase_01 | AC | 40 ms
55,592 KB |
testcase_02 | AC | 44 ms
55,592 KB |
testcase_03 | AC | 167 ms
78,724 KB |
testcase_04 | AC | 148 ms
78,200 KB |
testcase_05 | AC | 417 ms
100,928 KB |
testcase_06 | AC | 347 ms
96,440 KB |
testcase_07 | AC | 172 ms
78,664 KB |
testcase_08 | AC | 300 ms
90,804 KB |
testcase_09 | AC | 458 ms
106,392 KB |
testcase_10 | AC | 333 ms
93,052 KB |
testcase_11 | AC | 358 ms
95,620 KB |
testcase_12 | AC | 180 ms
78,952 KB |
testcase_13 | AC | 42 ms
55,596 KB |
testcase_14 | AC | 41 ms
55,596 KB |
testcase_15 | AC | 41 ms
55,596 KB |
testcase_16 | AC | 42 ms
55,596 KB |
testcase_17 | AC | 43 ms
55,596 KB |
testcase_18 | AC | 208 ms
96,684 KB |
testcase_19 | AC | 209 ms
96,684 KB |
testcase_20 | AC | 396 ms
100,532 KB |
testcase_21 | AC | 390 ms
99,364 KB |
testcase_22 | AC | 453 ms
112,080 KB |
testcase_23 | AC | 455 ms
111,440 KB |
testcase_24 | AC | 475 ms
110,452 KB |
testcase_25 | AC | 476 ms
110,988 KB |
testcase_26 | AC | 426 ms
100,972 KB |
testcase_27 | AC | 507 ms
106,188 KB |
testcase_28 | AC | 470 ms
110,508 KB |
testcase_29 | AC | 484 ms
111,320 KB |
testcase_30 | AC | 469 ms
110,496 KB |
testcase_31 | AC | 473 ms
111,492 KB |
testcase_32 | AC | 279 ms
96,684 KB |
testcase_33 | AC | 283 ms
96,684 KB |
testcase_34 | AC | 435 ms
102,056 KB |
testcase_35 | AC | 435 ms
102,304 KB |
ソースコード
class SegTree: """ init(init_val, ide_ele): 配列init_valで初期化 O(N) update(k, x): k番目の値をxに更新 O(logN) query(l, r): 区間[l, r)をsegfuncしたものを返す O(logN) """ def __init__(self, init_val, segfunc, ide_ele): """ init_val: 配列の初期値 segfunc: 区間にしたい操作 ide_ele: 単位元 n: 要素数 num: n以上の最小の2のべき乗 tree: セグメント木(1-index) """ n = len(init_val) self.segfunc = segfunc self.ide_ele = ide_ele self.num = 1 << (n - 1).bit_length() self.tree = [ide_ele] * 2 * self.num # 配列の値を葉にセット for i in range(n): self.tree[self.num + i] = init_val[i] # 構築していく for i in range(self.num - 1, 0, -1): self.tree[i] = self.segfunc(self.tree[2 * i], self.tree[2 * i + 1]) def update(self, k, x): """ k番目の値をxに更新 k: index(0-index) x: update value """ k += self.num self.tree[k] = x while k > 1: self.tree[k >> 1] = self.segfunc(self.tree[k], self.tree[k ^ 1]) k >>= 1 def query(self, l, r): """ [l, r)のsegfuncしたものを得る l: index(0-index) r: index(0-index) """ res = self.ide_ele l += self.num r += self.num while l < r: if l & 1: res = self.segfunc(res, self.tree[l]) l += 1 if r & 1: res = self.segfunc(res, self.tree[r - 1]) l >>= 1 r >>= 1 return res def value(self,index): return self.query(index,index+1) from collections import defaultdict def compress(A): tmp = defaultdict(int) B = list(set(A)) B.sort() zipped = [] #i番目のAの要素は何番目に大きいか unzipped = [] # i番目に大きいAの要素は何か for index,b in enumerate(B): tmp[b] = index unzipped.append(b) for a in A: zipped.append(tmp[a]) return zipped,unzipped N = int(input()) A = list(map(int,input().split())) mod = 10 ** 9 + 7 zipped,unzipped = compress(A) m = len(unzipped) segl = SegTree([0] * m,lambda x,y:(x+y)%mod,0) ldown = [] lup = [] for i in range(N): suml = segl.query(0,zipped[i]) ldown.append(suml) suml = segl.query(zipped[i]+1,m) lup.append(suml) segl.update(zipped[i],(segl.value(zipped[i]) + pow(2,i,mod))%mod) segr = SegTree([0] *m,lambda x,y:(x+y)%mod,0) rdown = [] rup = [] for i in reversed(range(N)): sumr = segr.query(0,zipped[i]) rdown.append(sumr) sumr = segr.query(zipped[i]+1,m) rup.append(sumr) segr.update(zipped[i],(segr.value(zipped[i]) + pow(2,N-i-1,mod))%mod) rdown.reverse() rup.reverse() ans = 0 for i in range(N): ans += ldown[i] * rdown[i] ans %= mod ans += lup[i] * rup[i] ans %= mod print(ans)