結果

問題 No.489 株に挑戦
コンテスト
ユーザー Wataoka Koki
提出日時 2019-12-27 15:33:03
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 2,039 bytes
コンパイル時間 179 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 17,024 KB
最終ジャッジ日時 2024-10-08 07:16:53
合計ジャッジ時間 16,831 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28 RE * 2 TLE * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

class SegmentTree:

    def __init__(self, arr, segfunc=min, ide_ele=float('inf')):
        self.arr = arr
        self.segfunc = segfunc
        self.ide_ele = ide_ele

        # make self.segment tree
        self.n = len(arr)
        self.num = 2**(self.n-1).bit_length()
        self.seg = [ide_ele]*2*self.num
        self._init_seg(self.arr)

    # セグメント木を構築
    def _init_seg(self, init_val):
        for i in range(self.n):
            self.seg[i+self.num-1]=init_val[i]    
        for i in range(self.num-2,-1,-1):
            self.seg[i]=self.segfunc(self.seg[2*i+1],self.seg[2*i+2]) 
        
    # k番目の値をxに変更
    def update(self, k, x):
        self.arr[k] = x
        k += self.num-1
        self.seg[k] = x
        while k:
            k = (k-1)//2
            self.seg[k] = self.segfunc(self.seg[k*2+1],self.seg[k*2+2])
        
    # 区間[p, q)に対するクエリ
    def query(self, p, q):
        if q<=p:
            return self.ide_ele
        p += self.num-1
        q += self.num-2
        res=self.ide_ele
        while q-p>1:
            if p&1 == 0:
                res = self.segfunc(res,self.seg[p])
            if q&1 == 1:
                res = self.segfunc(res,self.seg[q])
                q -= 1
            p = p//2
            q = (q-1)//2
        if p == q:
            res = self.segfunc(res,self.seg[p])
        else:
            res = self.segfunc(self.segfunc(res,self.seg[p]),self.seg[q])
        return res
    
    def get_arr(self):
        return self.arr

N, D, K = map(int, input().split())
x = []
for _ in range(N):
    x.append(int(input()))

stmax = SegmentTree(x, segfunc=max, ide_ele=0)

profit = -float('inf')
for i in range(N-2):
    can = stmax.query(i+1, min(i+D+1, N-1)) - x[i]
    if profit < can:
        profit = can
        r = (i+1, min(i+D+1, N-1))
        mini = i

minx = float('inf')
maxx = 0
maxi = -1
for i in range(*r):
    if maxx < x[i]:
        maxx = x[i]
        maxi = i

print(profit*K)
if profit*K != 0:
    print(mini, maxi)
0