結果

問題 No.2250 Split Permutation
ユーザー titiatitia
提出日時 2023-03-18 00:02:52
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 478 ms / 3,000 ms
コード長 1,178 bytes
コンパイル時間 232 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 108,032 KB
最終ジャッジ日時 2024-09-18 12:41:16
合計ジャッジ時間 7,562 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline

N=int(input())
P=list(map(int,input().split()))

mod=998244353

BIT=[0]*(N+2) # 出現回数をbit indexed treeの形でもっておく.
LEN=N+1
def update(v,w): # index vにwを加える
    while v<=LEN:
        BIT[v]+=w
        v+=(v&(-v)) # 自分を含む大きなノードへ. たとえばv=3→v=4

def getvalue(v): # MIN～vの区間の和を求める
    ANS=0
    while v!=0:
        ANS+=BIT[v]
        v-=(v&(-v)) # たとえばv=3→v=2へ
    return ANS

BIT2=[0]*(N+2) # 出現回数をbit indexed treeの形でもっておく.

def update2(v,w): # index vにwを加える
    while v<=LEN:
        BIT2[v]+=w
        v+=(v&(-v)) # 自分を含む大きなノードへ. たとえばv=3→v=4

def getvalue2(v): # MIN～vの区間の和を求める
    ANS=0
    while v!=0:
        ANS+=BIT2[v]
        v-=(v&(-v)) # たとえばv=3→v=2へ
    return ANS

MAX=pow(2,N-1,mod)
ANS=0
for i in range(N):
    p=P[i]
    ko=i-getvalue2(p)
    kieru=getvalue(LEN)-getvalue(p)
    
    update(p,pow(2,i,mod))
    update2(p,1)

    ANS+=MAX*ko
    ANS-=kieru*pow(2,N-1-i,mod)
    #print(ko,ANS,kieru)
    ANS%=mod

print(ANS%mod)
0