結果

問題 No.3303 Heal Slimes 2
ユーザー テナガザル
提出日時 2025-10-05 15:17:33
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 4,511 bytes
コンパイル時間 1,185 ms
コンパイル使用メモリ 110,504 KB
実行使用メモリ 8,064 KB
最終ジャッジ日時 2025-10-05 15:18:44
合計ジャッジ時間 66,564 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 28 TLE * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

template <typename T> class BIT
{
private:
  T e;
  int num;
  std::vector<T> dat;
  T (*eval)(T &, T &) {};
public:
  BIT() : e(0), num(0), dat(std::vector<int>(0)), eval([](int &a, int &b) -> int {return a + b;}){}
  BIT(const BIT<T> &bit) : e(bit.e), num(bit.num), dat(bit.dat), eval(bit.eval){}
  BIT(std::vector<T> &v, T E, T (*func)(T &, T &)) : e(E), eval(func)
  {
    num = static_cast<T>(v.size());
    dat = std::vector<T> (num + 1, e);
    for (int i = 0; i < num; ++i) dat[i + 1] = v[i];
    for (int i = 1; i <= num; ++i)
    {
      int j = i + (i & -i);
      if (j <= num) dat[j] = eval(dat[i], dat[j]);
    }
  }
  BIT(int n, T E, T (*func)(T &, T &)) : e(E), eval(func)
  {
    num = n;
    dat = std::vector<T> (num + 1, e);
  }
  //1-index
  void update(int i, T val) {for (; i <= num; i += i & -i) dat[i] = eval(dat[i], val);}
  //[0,right]
  T getval(int right)
  {
    T ans = e;
    for (; right > 0; right &= right - 1) ans = eval(ans, dat[right]);
    return ans;
  }
  int minright(T val)
  {
    if (val <= 0) return 0;
    int ret = 0, tmp = 1;
    for (int i = num; i > 0; i -= i & -i) tmp = i;
    for (; tmp > 0; tmp >>= 1)
    {
      if (ret + tmp <= num && dat[ret + tmp] < val)
      {
        ret += tmp;
        val -= dat[ret];
      }
    }
    return ret + 1;
  }
}; 

template <typename T> class segTree
{
private:
  const T e;
  int num;
  std::vector<T> dat;
  T (*const eval)(T, T) {};
public:
  segTree(std::vector<T> &v, T E, T (*func)(T, T)) : e(E), eval(func)
  {
    int siz = static_cast<int>(v.size());
    for (num = 1; num < siz; num <<= 1);
    dat = std::vector<T> (2 * num - 1, e);
    for (int i = 0; i < siz; ++i) dat[i + num - 1] = v[i];
    for (int i = num - 2; i >= 0; --i) dat[i] = eval(dat[i * 2 + 1], dat[i * 2 + 2]);
  }
  segTree(int n, T E, T (*func)(T, T)) : e(E), eval(func)
  {
    for (num = 1; num < n; num <<= 1);
    dat = std::vector<T> (2 * num - 1, e);
  }
  void update_a(int i, T val)
  {
    for (i += num - 1, dat[i] = val; i != 0;)
    {
      i = (i - 1) / 2;
      dat[i] = eval(dat[i * 2 + 1], dat[i * 2 + 2]);
    }
  }
  void update_r(int i, T val)
  {
    for (i += num - 1, dat[i] = eval(dat[i], val); i != 0;)
    {
      i = (i - 1) / 2;
      dat[i] = eval(dat[i * 2 + 1], dat[i * 2 + 2]);
    }
  }
  T getval(int left, int right)
  {
    left = std::max(0, left), right = std::min(num, right);
    T ansl = e, ansr = e;
    for (left += num - 1, right += num - 1; left < right; left >>= 1, right >>= 1)
    {
      if (!(left & 1)) ansl = eval(ansl, dat[left]);
      if (--right & 1) ansr = eval(dat[right], ansr);
    }
    return eval(ansl, ansr);
  }
  T getval(int id) {return dat[num - 1 + id];}
};

pair<int, int> a[1 << 17];
int h[1 << 17], id[1 << 17];

int main()
{
    ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    int n, k, d;
    cin >> n >> k >> d;
    for (int i = 0; i < n; ++i)
    {
        cin >> h[i];
        a[i] = {h[i], i};
    }
    sort(a, a + n);
    for (int i = 0; i < n; ++i) id[a[i].second] = i;
    segTree<long long> sg(n, 0, [](long long a, long long b) -> long long {return a + b;});
    // BIT<int> bit(n, 0, [](int &a, int &b) -> int {return a + b;});
    segTree<int> sgc(n, 0, [](int a, int b) -> int {return a + b;});
    for (int i = 0; i < k - 1; ++i) sg.update_a(id[i], h[i]);
    for (int i = 0; i < k - 1; ++i) sgc.update_a(id[i], 1);
    auto cal = [&](long long x) -> long long
    {
        auto idl = lower_bound(a, a + n, make_pair((int)x, 0)) - a;
        auto idr = upper_bound(a, a + n, make_pair((int)x + d, n + 1)) - a;
        long long ret = sg.getval(idr, n) - sgc.getval(idr, n) * (x + d);
        ret -= sg.getval(0, idl) - sgc.getval(0, idl) * x;
        return ret;
    };
    long long ans = (1LL << 60);
    for (int i = 0; i + k <= n; ++i)
    {
        sg.update_a(id[i + k - 1], h[i + k - 1]);
        sgc.update_a(id[i + k - 1], 1);
        long long l = 0, r = 1e9 - d;
        while (r - l > 2)
        {
            long long midl = ((long long)l + l + r) / 3;
            long long midr = ((long long)l + r + r) / 3;
            long long resl = cal(midl), resr = cal(midr);
            if (resl > resr) l = midl;
            else r = midr;
        }
        ans = min(ans, cal(l));
        ans = min(ans, cal(l + 1));
        ans = min(ans, cal(r));
        sg.update_a(id[i], 0);
        sgc.update_a(id[i], 0);
    }
    cout << ans << endl;
}
0