結果
問題 | No.1975 Zigzag Sequence |
ユーザー | lilictaka |
提出日時 | 2022-06-12 03:09:14 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 498 ms / 2,000 ms |
コード長 | 3,033 bytes |
コンパイル時間 | 260 ms |
コンパイル使用メモリ | 82,296 KB |
実行使用メモリ | 112,680 KB |
最終ジャッジ日時 | 2024-09-22 13:25:12 |
合計ジャッジ時間 | 13,189 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 43 ms
55,960 KB |
testcase_01 | AC | 41 ms
54,504 KB |
testcase_02 | AC | 41 ms
55,188 KB |
testcase_03 | AC | 173 ms
79,180 KB |
testcase_04 | AC | 145 ms
78,496 KB |
testcase_05 | AC | 418 ms
101,028 KB |
testcase_06 | AC | 328 ms
96,716 KB |
testcase_07 | AC | 165 ms
79,036 KB |
testcase_08 | AC | 300 ms
91,596 KB |
testcase_09 | AC | 455 ms
106,860 KB |
testcase_10 | AC | 328 ms
93,560 KB |
testcase_11 | AC | 356 ms
96,556 KB |
testcase_12 | AC | 177 ms
79,368 KB |
testcase_13 | AC | 41 ms
54,536 KB |
testcase_14 | AC | 40 ms
54,676 KB |
testcase_15 | AC | 41 ms
55,432 KB |
testcase_16 | AC | 41 ms
55,192 KB |
testcase_17 | AC | 43 ms
55,760 KB |
testcase_18 | AC | 211 ms
97,316 KB |
testcase_19 | AC | 209 ms
97,148 KB |
testcase_20 | AC | 386 ms
100,868 KB |
testcase_21 | AC | 392 ms
99,856 KB |
testcase_22 | AC | 456 ms
112,680 KB |
testcase_23 | AC | 453 ms
111,744 KB |
testcase_24 | AC | 482 ms
111,648 KB |
testcase_25 | AC | 483 ms
111,404 KB |
testcase_26 | AC | 430 ms
101,624 KB |
testcase_27 | AC | 498 ms
106,540 KB |
testcase_28 | AC | 471 ms
111,412 KB |
testcase_29 | AC | 484 ms
111,748 KB |
testcase_30 | AC | 473 ms
110,852 KB |
testcase_31 | AC | 481 ms
112,048 KB |
testcase_32 | AC | 288 ms
97,220 KB |
testcase_33 | AC | 286 ms
97,148 KB |
testcase_34 | AC | 435 ms
102,712 KB |
testcase_35 | AC | 442 ms
103,088 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)