結果

問題 No.3303 Heal Slimes 2
ユーザー timi
提出日時 2025-10-07 08:39:10
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 3,080 ms / 4,000 ms
コード長 2,975 bytes
コンパイル時間 432 ms
コンパイル使用メモリ 82,516 KB
実行使用メモリ 181,828 KB
最終ジャッジ日時 2025-10-07 08:40:09
合計ジャッジ時間 54,232 ms
ジャッジサーバーID
(参考情報)
judge5 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 31
権限があれば一括ダウンロードができます

ソースコード

diff #

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