結果

問題 No.489 株に挑戦
コンテスト
ユーザー Wataoka Koki
提出日時 2019-12-27 14:51:35
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 2,171 bytes
コンパイル時間 253 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 19,072 KB
最終ジャッジ日時 2024-10-08 05:02:07
合計ジャッジ時間 19,061 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 14 WA * 14 TLE * 7
権限があれば一括ダウンロードができます

ソースコード

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()))

stmin = SegmentTree(x, segfunc=min, ide_ele=float('inf'))
stmax = SegmentTree(x, segfunc=max, ide_ele=0)

profit = -float('inf')
for i in range(N-D):
    can = stmax.query(i, i+D+1) - stmin.query(i, i+D+1)
    if profit < can:
        profit = can
        maxx = stmax.query(i, i+D+1)
        minx = stmin.query(i, i+D+1)

mini, maxi = -1, -1
for i in range(N):
    if mini==-1 and x[i] == minx:
        mini = i
    if maxi==-1 and x[i] == maxx:
        maxi = i


if mini < maxi:
    print(profit*K)
    print(mini, maxi)
else:
    print(0)
0