結果

問題 No.1300 Sum of Inversions
ユーザー navel_tosnavel_tos
提出日時 2023-05-16 23:31:45
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,707 ms / 2,000 ms
コード長 4,141 bytes
コンパイル時間 206 ms
コンパイル使用メモリ 82,104 KB
実行使用メモリ 193,248 KB
最終ジャッジ日時 2024-05-08 15:59:25
合計ジャッジ時間 44,701 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 45 ms
54,688 KB
testcase_01 AC 45 ms
56,440 KB
testcase_02 AC 45 ms
55,120 KB
testcase_03 AC 1,388 ms
161,352 KB
testcase_04 AC 1,299 ms
158,240 KB
testcase_05 AC 1,101 ms
142,692 KB
testcase_06 AC 1,541 ms
178,868 KB
testcase_07 AC 1,456 ms
176,096 KB
testcase_08 AC 1,582 ms
183,972 KB
testcase_09 AC 1,537 ms
183,008 KB
testcase_10 AC 905 ms
133,056 KB
testcase_11 AC 915 ms
133,288 KB
testcase_12 AC 1,346 ms
159,560 KB
testcase_13 AC 1,309 ms
157,624 KB
testcase_14 AC 1,707 ms
191,260 KB
testcase_15 AC 1,594 ms
183,504 KB
testcase_16 AC 1,366 ms
162,580 KB
testcase_17 AC 889 ms
130,860 KB
testcase_18 AC 1,049 ms
138,076 KB
testcase_19 AC 1,184 ms
151,952 KB
testcase_20 AC 1,229 ms
152,464 KB
testcase_21 AC 1,230 ms
152,964 KB
testcase_22 AC 1,101 ms
143,040 KB
testcase_23 AC 1,487 ms
178,844 KB
testcase_24 AC 1,112 ms
144,004 KB
testcase_25 AC 952 ms
136,292 KB
testcase_26 AC 949 ms
136,080 KB
testcase_27 AC 1,056 ms
140,544 KB
testcase_28 AC 1,642 ms
185,868 KB
testcase_29 AC 1,180 ms
152,004 KB
testcase_30 AC 1,686 ms
183,204 KB
testcase_31 AC 1,072 ms
142,864 KB
testcase_32 AC 1,123 ms
145,572 KB
testcase_33 AC 1,091 ms
157,868 KB
testcase_34 AC 1,101 ms
171,064 KB
testcase_35 AC 1,165 ms
183,156 KB
testcase_36 AC 1,194 ms
193,248 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