結果

問題 No.704 ゴミ拾い Medium
ユーザー tpynerivertpyneriver
提出日時 2019-07-05 19:45:27
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 754 ms / 1,500 ms
コード長 2,652 bytes
コンパイル時間 151 ms
コンパイル使用メモリ 82,580 KB
実行使用メモリ 184,688 KB
最終ジャッジ日時 2024-09-22 04:36:14
合計ジャッジ時間 16,596 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 44
権限があれば一括ダウンロードができます

ソースコード

diff # 

class Lichaotree:
    #区間は1-indexed
    #min
    def __init__(self, X):
        self.N = len(X)
        self.N0 = 2**(self.N-1).bit_length()
        self.comp = {v : k for k, v in enumerate(X)}
        self.inf = 10**12
        self.X = X + [self.inf]*(self.N0 - self.N)
        self.data = [None]*(2*self.N0)
        self.size = [0]*self.N0 + [1]*self.N0
        for i in range(self.N0-1, 0, -1):
            self.size[i] = 2*self.size[2*i]
        
    def f(self, line, x):
        return line[0]*x + line[1]

    def _update(self, k, line):
        
        if self.data[k] is None:
            self.data[k] = line
            return
        
        l = k - (1 << (k.bit_length() - 1))
        l = l*self.size[k]
        r = l + self.size[k]

        
        while True:
            if self.data[k] is None:
                self.data[k] = line
                return
            
            m = (l+r)//2
            lx = self.X[l]
            mx = self.X[m]
            rx = self.X[r-1]
            gl = (self.f(self.data[k], lx) > self.f(line, lx))
            gm = (self.f(self.data[k], mx) > self.f(line, mx))
            gr = (self.f(self.data[k], rx) > self.f(line, rx))
            if gl and gr:
                self.data[k] = line
                return
            
            if not gl and not gr:
                return
            
            if gm:
                self.data[k], line = line, self.data[k]
                gl = not gl
                
            if gl:
                r = m
                k = 2*k
            else:
                l = m
                k = 2*k+1
    
    def query(self, x):
        k = self.comp[x] + self.N0
        s = self.inf
        while k > 0:
            if self.data[k]:
                s = min(s, self.f(self.data[k], x))
            k = k >> 1
        return s
            
    def addline(self, line):
        self._update(1, line)
    
    def addlineseg(self, l, r, line):
        #区間[l, r)にlineを追加,l, rはXの元
        
        L = l+self.N0
        R = r+self.N0
        while L < R:
            if R & 1:
                R -= 1
                self._update(R, line)
            if L & 1:
                self._update(L, line)
                L += 1
            L >>= 1
            R >>= 1

N = int(input())
A = list(map(int, input().split()))
X = list(map(int, input().split()))
Y = list(map(int, input().split()))

mX = max(X+A)
T = Lichaotree(list(range(mX + 1)))
N0 = T.N0
dp = [0]*N
for i in range(N):
    T.addlineseg(0, X[i], (-1, X[i] + Y[i] + dp[i-1]))
    T.addlineseg(X[i], N0, (1, -X[i] + Y[i] + dp[i-1]))
    dp[i] = T.query(A[i])

print(dp[-1])
0