結果

問題 No.3303 Heal Slimes 2
コンテスト
ユーザー 遭難者
提出日時 2025-10-05 15:47:45
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 4,366 bytes
コンパイル時間 5,213 ms
コンパイル使用メモリ 336,816 KB
実行使用メモリ 20,648 KB
最終ジャッジ日時 2025-10-05 15:48:19
合計ジャッジ時間 11,402 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 4 WA * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using std::cerr;
using std::cin;
using std::cout;
#if __has_include(<atcoder/all>)
#include <atcoder/all>
using mint = atcoder::modint998244353;
istream &operator>>(istream &is, mint &a) {
	long long t;
	is >> t;
	a = t;
	return is;
}
ostream &operator<<(ostream &os, mint a) { return os << a.val(); }
#endif
typedef long double ld;
#define long long long
#define uint unsigned int
#define ull unsigned long
#define overload3(a, b, c, name, ...) name
#define rep3(i, a, b) for (int i = (a); i < (b); i++)
#define rep2(i, n) rep3(i, 0, n)
#define rep1(n) rep2(__i, n)
#define rep(...) overload3(__VA_ARGS__, rep3, rep2, rep1)(__VA_ARGS__)
#define per3(i, a, b) for (int i = (b) - 1; i >= (a); i--)
#define per2(i, n) per3(i, 0, n)
#define per1(n) per2(__i, n)
#define per(...) overload3(__VA_ARGS__, per3, per2, per1)(__VA_ARGS__)
#define all(a) a.begin(), a.end()
#define UNIQUE(a)                                                              \
	sort(all(a));                                                              \
	a.erase(unique(all(a)), a.end())
#define sz(a) (int)a.size()
#define vec vector
#ifndef DEBUG
#define cerr                                                                   \
	if (0)                                                                     \
	cerr
#undef assert
#define assert(...) void(0)
#undef endl
#define endl '\n'
#endif
template <typename T> ostream &operator<<(ostream &os, vector<T> a) {
	const int n = a.size();
	rep(i, n) {
		os << a[i];
		if (i + 1 != n)
			os << " ";
	}
	return os;
}
template <typename T, size_t n>
ostream &operator<<(ostream &os, array<T, n> a) {
	rep(i, n) {
		os << a[i];
		if (i + 1 != n)
			os << " ";
	}
	return os;
}
template <typename T> istream &operator>>(istream &is, vector<T> &a) {
	for (T &i : a)
		is >> i;
	return is;
}
template <typename T, typename S> bool chmin(T &x, S y) {
	if ((T)y < x) {
		x = (T)y;
		return true;
	}
	return false;
}
template <typename T, typename S> bool chmax(T &x, S y) {
	if (x < (T)y) {
		x = (T)y;
		return true;
	}
	return false;
}
template <typename T> void operator++(vector<T> &a) {
	for (T &i : a)
		++i;
}
template <typename T> void operator--(vector<T> &a) {
	for (T &i : a)
		--i;
}
template <typename T> void operator++(vector<T> &a, int) {
	for (T &i : a)
		i++;
}
template <typename T> void operator--(vector<T> &a, int) {
	for (T &i : a)
		i--;
}
template <typename T> vec<int> z(vec<T> a) {
	int n = sz(a);
	auto b = a;
	sort(b.begin(), b.end());
	unordered_map<int, int> s;
	vector<int> res(n);
	rep(i, n) {
		int key = lower_bound(b.begin(), b.end(), a[i]) - b.begin();
		res[i] = key + s[key]++;
	}
	return res;
}
int op(int x, int y) { return x + y; }
int e() { return 0; }
int k;
bool judge(int x) { return x < k; }
void solve() {
	int n;
	long d;
	cin >> n >> k >> d;
	vec<long> h(n);
	cin >> h;
	vec<long> hh(2 * n);
	rep(i, n) hh[i] = h[i], hh[i + n] = h[i] + d;
	auto sorted_hh = hh;
	sort(all(sorted_hh));
	vec<int> a = z(hh);
	atcoder::segtree<int, op, e> fw(2 * n), segl(2 * n), segr(2 * n);
	atcoder::fenwick_tree<long> fwl(2 * n), fwr(2 * n);
	rep(i, k - 1) {
		fw.set(a[i], 1), fw.set(a[i + n], 1);
		segl.set(a[i], 1), fwl.add(a[i], hh[i]);
		segr.set(a[i + n], 1), fwr.add(a[i + n], hh[i + n]);
	}
	constexpr long INF = 1e18;
	long ans = INF;
	rep(i, k - 1, n) {
		fw.set(a[i], 1), fw.set(a[i + n], 1);
		segl.set(a[i], 1), fwl.add(a[i], hh[i]);
		segr.set(a[i + n], 1), fwr.add(a[i + n], hh[i + n]);

		int idx1 = fw.max_right<judge>(0);
		long right_v = sorted_hh[idx1] + d;
		// right_v 以上のやつで最小 idx
		int idx2 = lower_bound(all(sorted_hh), right_v) - sorted_hh.begin();
		long res = 0;
		{
			long ko1 = segl.prod(0, idx1);
			long su1 = fwl.sum(0, idx1);
			res += sorted_hh[idx1] * ko1 - su1;
		}
		{
			long ko2 = segl.prod(idx2, 2 * n);
			long su2 = fwl.sum(idx2, 2 * n);
			res += su2 - sorted_hh[idx2] * ko2;
		}
		chmin(ans, res);

		fw.set(a[i - k + 1], 0), fw.set(a[i - k + 1 + n], 0);
		segl.set(a[i - k + 1], 0), fwl.add(a[i - k + 1], -hh[i - k + 1]);
		segr.set(a[i - k + 1 + n], 0),
			fwr.add(a[i - k + 1 + n], -hh[i - k + 1 + n]);
	}
	cout << ans << endl;
}
int main() {
	// srand((unsigned)time(NULL));
	cin.tie(nullptr);
	ios::sync_with_stdio(false);
	cout << fixed << setprecision(20);
	int t = 1;
	// cin >> t;
	while (t--)
		solve();
}
0