結果

問題 No.1733 Sum of Sorted Subarrays
ユーザー aaaaaaaaaa2230aaaaaaaaaa2230
提出日時 2021-11-06 22:46:28
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 2,734 ms / 3,000 ms
コード長 4,055 bytes
コンパイル時間 990 ms
コンパイル使用メモリ 86,800 KB
実行使用メモリ 174,512 KB
最終ジャッジ日時 2023-08-07 17:37:02
合計ジャッジ時間 42,370 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 78 ms
71,452 KB
testcase_01 AC 79 ms
71,408 KB
testcase_02 AC 71 ms
71,204 KB
testcase_03 AC 76 ms
75,496 KB
testcase_04 AC 71 ms
71,000 KB
testcase_05 AC 78 ms
75,692 KB
testcase_06 AC 76 ms
75,692 KB
testcase_07 AC 72 ms
71,184 KB
testcase_08 AC 1,495 ms
116,944 KB
testcase_09 AC 2,153 ms
158,680 KB
testcase_10 AC 1,336 ms
119,592 KB
testcase_11 AC 1,640 ms
129,340 KB
testcase_12 AC 1,897 ms
139,880 KB
testcase_13 AC 1,752 ms
139,360 KB
testcase_14 AC 2,485 ms
173,168 KB
testcase_15 AC 2,347 ms
148,944 KB
testcase_16 AC 1,375 ms
121,460 KB
testcase_17 AC 2,559 ms
173,736 KB
testcase_18 AC 2,053 ms
146,544 KB
testcase_19 AC 2,045 ms
147,900 KB
testcase_20 AC 1,912 ms
135,392 KB
testcase_21 AC 2,270 ms
156,092 KB
testcase_22 AC 1,566 ms
129,264 KB
testcase_23 AC 2,643 ms
174,512 KB
testcase_24 AC 2,734 ms
174,436 KB
testcase_25 AC 2,529 ms
173,980 KB
testcase_26 AC 1,642 ms
172,864 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline
class LazySegTree:

    def __init__(self, n, op, e, mapping, composition, id):
        self.n = n
        self.op = op
        self.e = e
        self.mapping = mapping
        self.composition = composition
        self.id = id
        self.log = (n-1).bit_length()
        self.size = 1 << self.log
        self.data = [e]*(2*self.size)
        self.lazy = [id]*(self.size)

    def update(self, k):
        self.data[k] = self.op(self.data[2*k],self.data[2*k+1])

    def build(self, lis):
        for i, l in enumerate(lis,self.size):
            self.data[i] = l
        for i in range(self.size-1, 0, -1):
            self.update(i)

    def point_set(self, p, x):
        p += self.size
        for i in range(self.log, 0, -1):
            self.push(p >> i)
        self.data[p] = x
        for i in range(1,self.log+1):
            self.update(p>>i)

    def point_get(self, p):
        p += self.size
        for i in range(self.log, 0,-1):
            self.push(p >> i)
        return self.data[p]

    def apply(self, p, f):
        p += self.size
        for i in range(self.log, 0, -1):
            self.push(p >> i)
        self.data[p] = self.mapping(f, self.data[p])
        for i in range(1, self.log+1):
            self.update(p >> i)

    def range_apply(self, l, r, f):
        ''' change the value you define in "mapping" and "composition" of [l,r) '''
        if l == r:
            return 
        l += self.size
        r += self.size
        for i in range(self.log, 0, -1):
            if ((l >> i) << i) != l:
                self.push(l >> i)
            if ((r >> i) << i) != r:
                self.push((r-1) >> i)
        l2 = l
        r2 = r
        while l2 < r2:
            if l2 & 1:
                self.all_apply(l2, f)
                l2 += 1
            if r2 & 1:
                r2 -= 1
                self.all_apply(r2, f)
            l2 >>= 1
            r2 >>= 1
        for i in range(1, self.log+1):
            if ((l >> i) << i) != l:
                self.update(l >> i)
            if ((r >> i) << i) != r:
                self.update((r-1) >> i)

    def all_apply(self, k, f):
        self.data[k] = self.mapping(f, self.data[k])
        if k < self.size:
            self.lazy[k] = self.composition(f, self.lazy[k])

    def push(self, k):
        self.all_apply(2*k, self.lazy[k])
        self.all_apply(2*k+1, self.lazy[k])
        self.lazy[k] = self.id

    def prod(self, l, r):
        ''' get the value you define in "op" of [l,r) '''
        if l == r:
            return self.e
        l += self.size
        r += self.size
        for i in range(self.log, 0, -1):
            if ((l>>i) <<i) != l:
                self.push(l>>i)
            if ((r>>i) <<i) != r:
                self.push(r>>i)
        sml = smr = self.e
        while l < r:
            if l & 1:
                sml = self.op(sml, self.data[l])
                l += 1
            if r & 1:
                r -= 1
                smr = self.op(self.data[r], smr)
            l >>= 1
            r >>= 1
        return self.op(sml, smr)

    def all_prod(self):
        return self.data[1]



n = int(input())
A = list(map(int,input().split()))
sA = sorted([[a,i] for i,a in enumerate(A)])
mod = 998244353

def op(x, y):
    return (x+y)%mod

def mapping(p, x):
    return p*x%mod

def composition(p, q):
    return p*q%mod

e = 0
id = 1
lseg = LazySegTree(n+1, op, e, mapping, composition, id)
lseg.build([1]*(n+1))
rseg = LazySegTree(n+1, op, e, mapping, composition, id)
rseg.build([1]*(n+1))
inv2 = [1]
for i in range(n+1):
    inv2.append(inv2[-1]*2%mod)
dic = {}
dic[inv2[-1]]= pow(inv2[-1],mod-2,mod)
inv2[-1] = pow(inv2[-1],mod-2,mod)
for i in range(n+1)[::-1]:
    dic[inv2[i]] = inv2[i+1]*2%mod
    inv2[i] = inv2[i+1]*2%mod
ans = 0
for a,i in sA:
    l = lseg.prod(0,i+1)*dic[lseg.point_get(i+1)]%mod
    r = rseg.prod(i+1,n+1)*dic[rseg.point_get(i)]%mod
    ans += l*r%mod*a%mod
    ans %= mod
    lseg.range_apply(0,i+1,2)
    rseg.range_apply(i+1,n+1,2)
print(ans)
0