結果

問題 No.1300 Sum of Inversions
ユーザー navel_tosnavel_tos
提出日時 2023-05-16 23:31:45
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,752 ms / 2,000 ms
コード長 4,141 bytes
コンパイル時間 541 ms
コンパイル使用メモリ 82,436 KB
実行使用メモリ 193,052 KB
最終ジャッジ日時 2024-12-14 20:58:42
合計ジャッジ時間 46,034 ms
ジャッジサーバーID
(参考情報)
judge1 / judge2
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 43 ms
54,272 KB
testcase_01 AC 43 ms
54,016 KB
testcase_02 AC 44 ms
53,888 KB
testcase_03 AC 1,397 ms
161,100 KB
testcase_04 AC 1,321 ms
158,304 KB
testcase_05 AC 1,130 ms
142,440 KB
testcase_06 AC 1,563 ms
178,740 KB
testcase_07 AC 1,530 ms
176,100 KB
testcase_08 AC 1,638 ms
184,260 KB
testcase_09 AC 1,612 ms
182,972 KB
testcase_10 AC 936 ms
133,184 KB
testcase_11 AC 929 ms
133,028 KB
testcase_12 AC 1,394 ms
159,316 KB
testcase_13 AC 1,346 ms
157,328 KB
testcase_14 AC 1,752 ms
191,516 KB
testcase_15 AC 1,642 ms
183,336 KB
testcase_16 AC 1,429 ms
162,192 KB
testcase_17 AC 912 ms
130,860 KB
testcase_18 AC 1,046 ms
137,900 KB
testcase_19 AC 1,221 ms
151,568 KB
testcase_20 AC 1,275 ms
152,664 KB
testcase_21 AC 1,248 ms
152,884 KB
testcase_22 AC 1,165 ms
142,656 KB
testcase_23 AC 1,539 ms
178,724 KB
testcase_24 AC 1,165 ms
144,136 KB
testcase_25 AC 991 ms
135,776 KB
testcase_26 AC 990 ms
136,072 KB
testcase_27 AC 1,090 ms
140,412 KB
testcase_28 AC 1,713 ms
185,736 KB
testcase_29 AC 1,209 ms
151,748 KB
testcase_30 AC 1,622 ms
183,264 KB
testcase_31 AC 1,098 ms
142,740 KB
testcase_32 AC 1,157 ms
145,064 KB
testcase_33 AC 1,097 ms
157,492 KB
testcase_34 AC 1,113 ms
171,000 KB
testcase_35 AC 1,175 ms
183,056 KB
testcase_36 AC 1,204 ms
193,052 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#yukicoder276C Sum of Inversions_2

'''
3ヶ月ぶりに挑戦。意外と解けそう。
「右側に単調減少列はいくつある?」を管理しながらやればいい。セグ木で殴れそう。
'''
#Segment Tree: O(logN)
class SegmentTree:                                    # Segment Tree
    def __init__(self,n,identity_e,combine_f):        # 適応条件: 単位元eがある、互換可能
        self._n=n; self._size=1                       # モノイド(単位元)の例:
        while self._size<self._n:self._size<<=1       #  足し算 0, かけ算 1, 最小 INF,
        self._identity_e=identity_e                   #  最大 -INF, LCM(最小公倍数) 1
        self._combine_f=combine_f                     #
        self._node=[self._identity_e]*2*self._size    # combine_f には関数を指定する
                                                      # def文で関数を自作してもいいし、
    def build(self,array):                            #  from operator import xor
        assert len(array)==self._n,'array too large'  # のようにimportしてもよい
        for i,v in enumerate(array,start=self._size): #
            self._node[i]=v                           # build: セグ木を建てる
        for i in range(self._size-1,0,-1):            # 異常時はassert関数でエラーを報告
            self._node[i]=self._combine_f(self._node[i<<1|0],self._node[i<<1|1])
                                                      #
    def update(self,index,value):                     # update: 一点更新 O(logN)
        i=self._size+index; self._node[i]=value       # 地点i(0-indexed)を更新する
        while i-1:                                    # 同時に上位のセグメントも更新する
            i>>=1                                     #
            self._node[i]=self._combine_f(self._node[i<<1|0],self._node[i<<1|1])
                                                      #
    def fold(self,L,R):                               # fold: 区間取得 O(logN)
        L+=self._size; R+=self._size                  # 区間 [L,R) の特定値を取得する
        vL,vR=[self._identity_e]*2                    #
        while L<R:                                    # nodeの遷移の考え方
            if L&1:                                   #  ---1---  L: 自身より右の最小
                vL=self._combine_f(vL,self._node[L])  #  -2- -3-  R: 自身-1より左の最小
                L+=1                                  #  4 5 6 7  Rは計算より先に-1の
            if R&1:                                   #           処理をする点に注意
                R-=1                                  # R---1---L
                vR=self._combine_f(self._node[R],vR)  # R-2- LLL. 例: L=6, R=5
            L>>=1; R>>=1                              # .R.5 L 7      Rの移動が変則的
        return self._combine_f(vL,vR)                 #  ←R L→


'''
と思ったが、意外と面倒くさそうだぞ。
jを固定して、iとkの取り得る組合わせ数を加算してゆく方針に変えるか。
っていうか、これ総和もいるんか。面倒くさすぎる。

ちゃんとTLEするなぁ。仕方ない、圧縮術を使おう。やりたくないんだけど・・・
'''
def add(x,y,base=3*10**9):
    x1,x2=x//base,x%base; y1,y2=y//base,y%base
    return (x1+y1)*base + (x2+y2)%MOD    
from collections import defaultdict as dd
N=int(input()); A=list(map(int,input().split())); MOD=998244353; base=3*10**9

D=dd(list)
for i in range(N): D[A[i]].append(i)

Key=sorted(D.keys()); ST=SegmentTree(N,0,add); K=[[0]*2 for _ in range(N)]
for num in Key:
    for pos in D[num]:
        X=ST.fold(pos+1,N); K[pos]=[X//base,X%base]
        ST.update(pos,base+A[pos])
    
del ST; ST=SegmentTree(N,0,add); I=[[0]*2 for _ in range(N)]
for num in Key[::-1]:
    for pos in D[num][::-1]:
        X=ST.fold(0,pos); I[pos]=[X//base,X%base]
        ST.update(pos,base+A[pos])

del ST; ans=0
for j in range(N):
    if I[j][0]*K[j][0]: ans+=I[j][1]*K[j][0] + I[j][0]*K[j][1] + A[j]*I[j][0]*K[j][0]; ans%=MOD
print(ans)
0