結果
問題 | No.1975 Zigzag Sequence |
ユーザー | ygd. |
提出日時 | 2022-06-16 21:20:14 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 410 ms / 2,000 ms |
コード長 | 3,383 bytes |
コンパイル時間 | 348 ms |
コンパイル使用メモリ | 82,236 KB |
実行使用メモリ | 140,656 KB |
最終ジャッジ日時 | 2024-10-07 04:03:37 |
合計ジャッジ時間 | 10,576 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 48 ms
67,452 KB |
testcase_01 | AC | 48 ms
67,344 KB |
testcase_02 | AC | 48 ms
67,372 KB |
testcase_03 | AC | 138 ms
84,056 KB |
testcase_04 | AC | 123 ms
84,168 KB |
testcase_05 | AC | 320 ms
101,256 KB |
testcase_06 | AC | 276 ms
104,384 KB |
testcase_07 | AC | 141 ms
84,324 KB |
testcase_08 | AC | 242 ms
92,876 KB |
testcase_09 | AC | 358 ms
120,256 KB |
testcase_10 | AC | 268 ms
100,096 KB |
testcase_11 | AC | 297 ms
102,624 KB |
testcase_12 | AC | 144 ms
84,528 KB |
testcase_13 | AC | 49 ms
67,432 KB |
testcase_14 | AC | 48 ms
67,000 KB |
testcase_15 | AC | 48 ms
67,124 KB |
testcase_16 | AC | 48 ms
67,436 KB |
testcase_17 | AC | 48 ms
67,388 KB |
testcase_18 | AC | 92 ms
97,204 KB |
testcase_19 | AC | 93 ms
97,260 KB |
testcase_20 | AC | 272 ms
99,436 KB |
testcase_21 | AC | 290 ms
99,304 KB |
testcase_22 | AC | 373 ms
140,456 KB |
testcase_23 | AC | 338 ms
140,656 KB |
testcase_24 | AC | 381 ms
137,788 KB |
testcase_25 | AC | 373 ms
137,852 KB |
testcase_26 | AC | 313 ms
98,368 KB |
testcase_27 | AC | 410 ms
124,064 KB |
testcase_28 | AC | 363 ms
137,572 KB |
testcase_29 | AC | 370 ms
137,456 KB |
testcase_30 | AC | 378 ms
137,380 KB |
testcase_31 | AC | 398 ms
137,248 KB |
testcase_32 | AC | 173 ms
97,280 KB |
testcase_33 | AC | 177 ms
97,232 KB |
testcase_34 | AC | 350 ms
105,836 KB |
testcase_35 | AC | 327 ms
105,680 KB |
ソースコード
import sys #input = sys.stdin.readline input = sys.stdin.buffer.readline #文字列はダメ #sys.setrecursionlimit(1000000) #import bisect #import itertools #import random #from heapq import heapify, heappop, heappush #from collections import defaultdict #from collections import deque #import copy #import math #from functools import lru_cache #@lru_cache(maxsize=None) MOD = pow(10,9) + 7 #MOD = 998244353 #dx = [1,0,-1,0] #dy = [0,1,0,-1] #dx8 = [1,1,0,-1,-1,-1,0,1] #dy8 = [0,1,1,1,0,-1,-1,-1] #和のセグ木にするときは以下の関数を定義する必要あり。 def segfunc(x,y): return x+y class SegmentTree(object): def __init__(self, A, dot, unit): n = 1 << (len(A) - 1).bit_length() tree = [unit] * (2 * n) for i, v in enumerate(A): tree[i + n] = v for i in range(n - 1, 0, -1): tree[i] = dot(tree[i << 1], tree[i << 1 | 1]) self._n = n self._tree = tree self._dot = dot self._unit = unit def __getitem__(self, i): return self._tree[i + self._n] def update(self, i, v): i += self._n self._tree[i] = v while i != 1: i >>= 1 self._tree[i] = self._dot(self._tree[i << 1], self._tree[i << 1 | 1]) def add(self, i, v): self.update(i, self[i] + v) def sum(self, l, r): #これで[l,r)から取り出す。 l += self._n r += self._n l_val = r_val = self._unit while l < r: if l & 1: l_val = self._dot(l_val, self._tree[l]) l += 1 if r & 1: r -= 1 r_val = self._dot(self._tree[r], r_val) l >>= 1 r >>= 1 return self._dot(l_val, r_val) def compress(S): S = set(S); S = list(S) S.sort() dic_comp_ori = {} dic_ori_comp = {} for i,a in enumerate(S): dic_ori_comp[a] = i dic_comp_ori[i] = a return dic_comp_ori, dic_ori_comp MAX = 110000 two = [1] for i in range(MAX): two.append(two[-1]*2%MOD) def main(): N = int(input()) A = list(map(int,input().split())) dic_comp_ori, dic_ori_comp = compress(A) A = [dic_ori_comp[a] for a in A] # print(A) n = max(A) + 1 #Aの最大値 left_tree = SegmentTree([0 for _ in range(n)], segfunc, 0) right_tree = SegmentTree([0 for _ in range(n)], segfunc, 0) for i,a in enumerate(A): val = two[N-1-i] # print("i",i,"val",val) right_tree.add(a,val) ans = 0 for i,a in enumerate(A): # TL = []; TR = [] # for j in range(n): # TL.append(left_tree[j]) # TR.append(right_tree[j]) # print(TL) # print(TR) left_small = left_tree.sum(0,a) #[0,a) right_small = right_tree.sum(0,a) #[a+1,MAX) left_large = left_tree.sum(a+1,n) #[0,a) right_large = right_tree.sum(a+1,n) #[a+1,MAX) # print("small",left_small,right_small) # print("large",left_large,right_large) temp = (left_small*right_small + left_large*right_large)%MOD ans += temp ans %= MOD # print(left,right,temp) left_val = two[i] right_val = two[N-1-i] left_tree.add(a,left_val) right_tree.add(a,-right_val) print(ans) if __name__ == '__main__': main()