結果
問題 | No.489 株に挑戦 |
ユーザー |
|
提出日時 | 2022-01-31 23:18:58 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 188 ms / 1,000 ms |
コード長 | 2,827 bytes |
コンパイル時間 | 157 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 79,744 KB |
最終ジャッジ日時 | 2024-06-11 09:05:17 |
合計ジャッジ時間 | 4,914 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 35 |
ソースコード
import mathclass SegTree:DEFAULT = {'min': 1 << 60,'max': -(1 << 60),'sum': 0,'prd': 1,'gcd': 0,'lmc': 1,'^': 0,'&': (1 << 60) - 1,'|': 0,}FUNC = {'min': min,'max': max,'sum': (lambda x, y: x + y),'prd': (lambda x, y: x * y),'gcd': math.gcd,'lmc': (lambda x, y: (x * y) // math.gcd(x, y)),'^': (lambda x, y: x ^ y),'&': (lambda x, y: x & y),'|': (lambda x, y: x | y),}def __init__(self, ls, mode='min', func=None, default=None):"""要素ls, 関数mode (min,max,sum,prd(product),gcd,lmc,^,&,|)func,defaultを指定すれば任意の関数、単位元での計算が可能"""N = len(ls)if default == None:self.default = self.DEFAULT[mode]else:self.default = defaultif func == None:self.func = self.FUNC[mode]else:self.func = funcself.N = Nself.K = (N - 1).bit_length()self.N2 = 1 << self.Kself.dat = [self.default] * (2**(self.K + 1))for i in range(self.N): # 葉の構築self.dat[self.N2 + i] = ls[i]self.build()def build(self):for j in range(self.N2 - 1, -1, -1):self.dat[j] = self.func(self.dat[j << 1], self.dat[j << 1 | 1]) # 親が持つ条件def leafvalue(self, x): # リストのx番目の値return self.dat[x + self.N2]def update(self, x, y): # index(x)をyに変更i = x + self.N2self.dat[i] = ywhile i > 0: # 親の値を変更i >>= 1self.dat[i] = self.func(self.dat[i << 1], self.dat[i << 1 | 1])returndef query(self, L, R): # [L,R)の区間取得L += self.N2R += self.N2vL = self.defaultvR = self.defaultwhile L < R:if L & 1:vL = self.func(vL, self.dat[L])L += 1if R & 1:R -= 1vR = self.func(self.dat[R], vR)L >>= 1R >>= 1return self.func(vL, vR)def __iter__(self):for i in range(self.N):yield self[i]def __getitem__(self, x): return self.leafvalue(x)def __setitem__(self, x, val): return self.update(x, val)N,D,K = map(int,input().split())lsx = [int(input()) for i in range(N)]SG = SegTree(lsx)m = 0ind = 0for i in range(1,N):mi = SG.query(max(0,i-D),i)c = lsx[i]-miif m < c:m = cind = iif m == 0:print(0)exit()st = 0for i in range(max(0,ind-D),ind,1):if lsx[i] == lsx[ind]-m:st = ibreakprint(m*K)print(st,ind)