結果

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

ソースコード

diff # 

# pragma GCC optimize("O3")
# pragma GCC optimize("unroll-loops")

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

using namespace std;

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[100005];
int h[100005], id[100005];

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;});
    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