結果

問題 No.704 ゴミ拾い Medium
コンテスト
ユーザー tpyneriver
提出日時 2019-07-05 19:57:10
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 743 ms / 1,500 ms
コード長 2,842 bytes
コンパイル時間 309 ms
コンパイル使用メモリ 82,424 KB
実行使用メモリ 187,164 KB
最終ジャッジ日時 2024-09-22 04:49:27
合計ジャッジ時間 16,463 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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]
        self.start = [0]*(2*self.N0)
        for i in range(2, 2*self.N0):
            if i & 1:
                self.start[i] = self.start[i>>1] + self.size[i]
            else:
                self.start[i] = self.start[i>>1]
        
    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 = self.start[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