結果
| 問題 | No.3303 Heal Slimes 2 | 
| コンテスト | |
| ユーザー |  timi | 
| 提出日時 | 2025-10-07 08:39:10 | 
| 言語 | PyPy3 (7.3.15) | 
| 結果 | 
                                AC
                                 
                             | 
| 実行時間 | 2,842 ms / 4,000 ms | 
| コード長 | 2,975 bytes | 
| コンパイル時間 | 386 ms | 
| コンパイル使用メモリ | 82,836 KB | 
| 実行使用メモリ | 181,376 KB | 
| 最終ジャッジ日時 | 2025-10-21 10:26:44 | 
| 合計ジャッジ時間 | 49,758 ms | 
| ジャッジサーバーID (参考情報) | judge4 / judge1 | 
(要ログイン)
| ファイルパターン | 結果 | 
|---|---|
| other | AC * 32 | 
ソースコード
N,K,D=map(int, input().split())
A=list(map(int, input().split()))
E={}
B=[]
for a in A:
  B.append(a)
  B.append(a+D)
  B.append(a-D)
  
B=sorted(list(set(B)))
for i in range(len(B)):
  E[B[i]]=i
  
##### segfunc#####
def segfunc(x, y):
    return x+y
#################
##### ide_ele#####
ide_ele = 0
#################
class SegTree:
    """
    init(init_val, ide_ele): 配列init_valで初期化 O(N)
    update(k, x): k番目の値をxに更新 O(N)
    query(l, r): 区間[l, r)をsegfuncしたものを返す O(logN)
    """
    def __init__(self, init_val, segfunc, ide_ele):
        """
        init_val: 配列の初期値
        segfunc: 区間にしたい操作
        ide_ele: 単位元
        n: 要素数
        num: n以上の最小の2のべき乗
        tree: セグメント木(1-index)
        """
        n = len(init_val)
        self.segfunc = segfunc
        self.ide_ele = ide_ele
        self.num = 1 << (n - 1).bit_length()
        self.tree = [ide_ele] * 2 * self.num
        # 配列の値を葉にセット
        for i in range(n):
            self.tree[self.num + i] = init_val[i]
        # 構築していく
        for i in range(self.num - 1, 0, -1):
            self.tree[i] = self.segfunc(self.tree[2 * i], self.tree[2 * i + 1])
    
    def set(self, k, x):
        """
        k番目の値をxに更新
        k: index(0-index)
        x: update value
        """
        k += self.num
        self.tree[k]+= x
        while k > 1:
            self.tree[k >> 1] = self.segfunc(self.tree[k], self.tree[k ^ 1])
            k >>= 1
    def prod(self, l, r):
        """
        [l, r)のsegfuncしたものを得る
        l: index(0-index)
        r: index(0-index)
        """
        res = self.ide_ele
        l += self.num
        r += self.num
        while l < r:
            if l & 1:
                res = self.segfunc(res, self.tree[l])
                l += 1
            if r & 1:
                res = self.segfunc(res, self.tree[r - 1])
            l >>= 1
            r >>= 1
        return res
import bisect
M=3*10**5+10
AA=[0]*(M+10)
st=SegTree(AA, segfunc, ide_ele)
st2=SegTree(AA, segfunc, ide_ele)
ans=10**20
for i in range(N):
  a=E[A[i]]
  st.set(a,A[i])
  st2.set(a,1)
  
  if i+1>=K:
    l,r=-1,len(B)
    while r-l>1:
      mid=(l+r)//2
      p=st2.prod(0,mid) 
      z=bisect.bisect_right(B,B[mid]+D)
      q=st2.prod(z,M+5)
      if p-q>0:
        r=mid 
      else:
        l=mid
        
    cnt=0
    mid=(l+r)//2
    p=st.prod(0,mid) 
    q=st2.prod(0,mid) 
    cnt+=B[mid]*q-p
    z=bisect.bisect_right(B,B[mid]+D)
    p=st.prod(z,M+5)
    q=st2.prod(z,M+5)
    cnt+=p-(B[mid]+D)*q
    ans=min(ans,cnt)
    
    cnt=0;mid+=1
    p=st.prod(0,mid) 
    q=st2.prod(0,mid) 
    cnt+=B[mid]*q-p
    z=bisect.bisect_right(B,B[mid]+D)
    p=st.prod(z,M)
    q=st2.prod(z,M)
    cnt+=p-(B[mid]+D)*q
    ans=min(ans,cnt)
    
    j=i-K+1
    a=E[A[j]]
    st.set(a,-A[j])
    st2.set(a,-1)
print(ans)
            
            
            
        