結果

問題 No.2332 Make a Sequence
ユーザー navel_tosnavel_tos
提出日時 2024-02-25 22:48:54
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 4,852 bytes
コンパイル時間 471 ms
コンパイル使用メモリ 82,032 KB
実行使用メモリ 204,720 KB
最終ジャッジ日時 2024-09-29 11:26:30
合計ジャッジ時間 31,971 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 44 ms
54,784 KB
testcase_01 AC 41 ms
53,440 KB
testcase_02 AC 42 ms
53,412 KB
testcase_03 AC 41 ms
54,772 KB
testcase_04 AC 41 ms
53,136 KB
testcase_05 AC 41 ms
53,280 KB
testcase_06 AC 40 ms
53,228 KB
testcase_07 AC 274 ms
122,696 KB
testcase_08 AC 115 ms
103,704 KB
testcase_09 AC 203 ms
125,944 KB
testcase_10 AC 247 ms
135,184 KB
testcase_11 AC 175 ms
99,620 KB
testcase_12 AC 266 ms
135,836 KB
testcase_13 AC 264 ms
135,804 KB
testcase_14 AC 261 ms
135,476 KB
testcase_15 AC 315 ms
136,172 KB
testcase_16 AC 309 ms
135,400 KB
testcase_17 AC 279 ms
136,092 KB
testcase_18 AC 265 ms
135,728 KB
testcase_19 AC 282 ms
135,968 KB
testcase_20 AC 272 ms
135,588 KB
testcase_21 AC 270 ms
135,976 KB
testcase_22 AC 317 ms
136,428 KB
testcase_23 AC 320 ms
135,392 KB
testcase_24 AC 315 ms
136,492 KB
testcase_25 AC 335 ms
137,012 KB
testcase_26 AC 318 ms
134,936 KB
testcase_27 AC 335 ms
135,380 KB
testcase_28 AC 335 ms
135,972 KB
testcase_29 AC 364 ms
136,092 KB
testcase_30 AC 334 ms
136,228 KB
testcase_31 AC 342 ms
135,112 KB
testcase_32 AC 392 ms
136,372 KB
testcase_33 AC 405 ms
136,508 KB
testcase_34 AC 405 ms
136,624 KB
testcase_35 AC 409 ms
136,120 KB
testcase_36 AC 397 ms
136,360 KB
testcase_37 AC 464 ms
136,600 KB
testcase_38 AC 471 ms
136,360 KB
testcase_39 AC 462 ms
136,292 KB
testcase_40 AC 447 ms
136,392 KB
testcase_41 AC 412 ms
136,832 KB
testcase_42 AC 658 ms
180,520 KB
testcase_43 AC 621 ms
179,084 KB
testcase_44 AC 646 ms
178,976 KB
testcase_45 WA -
testcase_46 WA -
testcase_47 AC 1,297 ms
204,032 KB
testcase_48 AC 1,471 ms
204,720 KB
testcase_49 AC 1,202 ms
203,756 KB
testcase_50 AC 1,259 ms
204,508 KB
testcase_51 AC 1,396 ms
204,460 KB
testcase_52 AC 405 ms
135,992 KB
testcase_53 AC 367 ms
136,420 KB
testcase_54 AC 366 ms
136,496 KB
testcase_55 AC 393 ms
136,624 KB
testcase_56 AC 409 ms
136,492 KB
testcase_57 AC 360 ms
136,384 KB
testcase_58 AC 434 ms
135,848 KB
testcase_59 AC 383 ms
134,748 KB
testcase_60 AC 422 ms
137,140 KB
testcase_61 AC 383 ms
134,912 KB
testcase_62 AC 1,093 ms
176,484 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#MMA Contest 015 - K Make a Sequence
#static Li Chao Treeで通ったのでdynamic Li Chao Treeで通す

#動的Li Chao Tree
class dynamic_Li_Chao_Tree:
    def __init__(self, min_x, max_x):
        #定義域 [min_x, max_x]内でのみ正確に動作する
        #_C[i]: (左子のid, 右子のid)  親のidは保持しない
        #_S[i]: (Lt, mid, Rt)  区間は閉区間[Lt, Rt]で扱う
        Lt, Rt = -1, 0
        while min_x < Lt: Lt = Lt * 2
        while Rt < max_x: Rt = Rt * 2 + 1
        self._C = [(None, None)]
        self._S = [(Lt, -1, Rt)]
        self._L = [None]

    def _f(self, line, x):
        return line[0] * x + line[1]

    def _get_child(self, index, get_left = True):
        #子のidを返す。子が存在しなければ作る
        c_Lt, c_Rt = self._C[index]
        Lt, mid, Rt = self._S[index]
        if get_left:
            if c_Lt != None:
                return c_Lt
            self._C[index] = (len(self._S), c_Rt)
            self._S.append((Lt, (Lt + mid) // 2, mid))
        else:
            if c_Rt != None:
                return c_Rt
            self._C[index] = (c_Lt, len(self._S))
            self._S.append((mid + 1, (mid + 1 + Rt) // 2, Rt))
        self._C.append((None, None))
        self._L.append(None)
        return len(self._S) - 1

    def _add_line(self, line, i):  #node iとその子に線分を追加
        Q = [(i, line)]
        while Q:
            i, line = Q.pop()
            if self._L[i] == None:
                self._L[i] = line
                continue
            Lt, mid, Rt = self._S[i]
            hL = self._f(line, Lt ) < self._f(self._L[i], Lt )
            hM = self._f(line, mid) < self._f(self._L[i], mid)     
            hR = self._f(line, Rt ) < self._f(self._L[i], Rt )
            if hL == hR == True:
                self._L[i] = line
                continue
            if hL == hR == False:
                continue
            if hM == line:
                self._L[i], line = line, self._L[i]
            if hL != hM:  #左子を作り降りる
                Q.append((self._get_child(i, True), line))
            else:
                Q.append((self._get_child(i, False), line))

    def add_line(self, line, x_Lt = None, x_Rt = None):  #半開区間[x_Lt, x_Rt)に線分追加
        if x_Lt == x_Rt == None:
            self._add_line(line, 0)
            return
        x_Lt = self._S[0][0] if x_Lt == None else x_Lt
        x_Rt = self._S[0][2] if x_Rt == None else x_Rt - 1
        Q = [(0, line, x_Lt, x_Rt)]
        while Q:
            i, line, x_Lt, x_Rt = Q.pop()
            Lt, mid, Rt = self._S[i]
            if x_Lt > x_Rt or x_Rt < Lt or Rt < x_Lt:
                continue
            x_Lt = max(x_Lt, Lt)
            x_Rt = min(x_Rt, Rt)
            if x_Lt <= Lt <= Rt <= x_Rt:
                self._add_line(line, i)
            elif x_Rt <= mid:
                Q.append((self._get_child(i, True), line, x_Lt, x_Rt))
            elif mid < x_Lt:
                Q.append((self._get_child(i, False), line, x_Lt, x_Rt))
            else:
                Q.append((self._get_child(i, True), line, x_Lt, mid))
                Q.append((self._get_child(i, False), line, mid + 1, x_Rt))

    def fold(self, x):  #座標xの最小値を計算
        i = 0
        ans = 4 * 10 ** 18
        while i != None:
            Lt, mid, Rt = self._S[i]
            c_Lt, c_Rt = self._C[i]
            if self._L[i] != None:
                ans = min(ans, self._f(self._L[i], x))
            i = c_Lt if x <= mid else c_Rt
        return ans


#Z-algorithm  TとS[i:]のLCP判定は、T+'$'+S のようにSに出ない文字を挟んで実行するとよい
#Reference: https://tjkendev.github.io/procon-library/python/string/z-algorithm.html
def Z_algorithm(S):  #SとS[i:]の最長共通接頭辞を求める
    N=len(S); A=[0]*N; i,j=1,0; A[0]=L=N
    while i<L:
        while i+j<L and S[j]==S[i+j]: j+=1
        if not j: i+=1; continue
        A[i]=j; k=1
        while L-i>k<j-A[k]: A[i+k]=A[k]; k+=1
        i+=k; j-=k
    return A


#入力受取
N, M = map(int, input().split())
A = list(map(int, input().split()))
B = list(map(int, input().split()))
C = list(map(int, input().split()))

#Bの(Aとの)最長共通接頭辞を計算
same = Z_algorithm(A + [0] + B)[N + 1:]

#DP[i]: 列Sを[0: i)まで(つまり、i文字目の直前まで)一致させるための最小コスト とする
#これはLi Chao Treeの線分追加で計算が可能
#k = same[i]として、[i: i + k)に y = C[i] * x + (DP[i] - C[i] * i) の線分を追加する
LCT = dynamic_Li_Chao_Tree(0, M + 1)
LCT.add_line((0, 0), 0, 1)  #初期化
for i in range(M):
    b = LCT.fold(i)
    k = same[i]
    LCT.add_line((C[i], b - C[i] * i), i + 1, min(M, i + k) + 1)
    
#答えを出力
ans = LCT.fold(M)
print(ans if ans < 10 ** 18 else -1)
0