結果

問題 No.489 株に挑戦
ユーザー Wataoka Koki
提出日時 2019-12-27 15:49:14
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 1,747 bytes
コンパイル時間 185 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 16,896 KB
最終ジャッジ日時 2024-10-08 09:03:39
合計ジャッジ時間 16,089 ms
ジャッジサーバーID
(参考情報)
judge4 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 30 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

        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]) 
    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()))
if N==2:
    if x[0] < x[1]:
        print((x[1]-x[0])*K)
        print(0, 1)
    else:
        print(0)
    exit(0)
st=SegmentTree(x,segfunc=max,ide_ele=0)
profit=-float('inf')
r=(0,1)
mini=0
for i in range(N-2):
    can = st.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