結果

問題 No.2859 Falling Balls
ユーザー square1001square1001
提出日時 2024-07-15 12:00:39
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,935 bytes
コンパイル時間 1,089 ms
コンパイル使用メモリ 90,664 KB
実行使用メモリ 18,656 KB
最終ジャッジ日時 2024-07-15 12:00:46
合計ジャッジ時間 7,271 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 1 ms
6,940 KB
testcase_03 AC 2 ms
6,944 KB
testcase_04 AC 2 ms
6,944 KB
testcase_05 AC 1 ms
6,940 KB
testcase_06 WA -
testcase_07 AC 1 ms
6,940 KB
testcase_08 AC 1 ms
6,940 KB
testcase_09 AC 1 ms
6,940 KB
testcase_10 AC 40 ms
7,064 KB
testcase_11 AC 137 ms
16,452 KB
testcase_12 AC 7 ms
6,940 KB
testcase_13 AC 127 ms
15,732 KB
testcase_14 AC 109 ms
14,648 KB
testcase_15 AC 40 ms
6,968 KB
testcase_16 AC 52 ms
8,876 KB
testcase_17 AC 53 ms
8,912 KB
testcase_18 AC 6 ms
6,940 KB
testcase_19 AC 81 ms
10,672 KB
testcase_20 AC 151 ms
17,508 KB
testcase_21 AC 151 ms
17,456 KB
testcase_22 AC 152 ms
17,416 KB
testcase_23 AC 152 ms
17,508 KB
testcase_24 AC 151 ms
18,656 KB
testcase_25 AC 150 ms
17,536 KB
testcase_26 AC 151 ms
17,448 KB
testcase_27 AC 163 ms
18,464 KB
testcase_28 AC 158 ms
18,084 KB
testcase_29 AC 150 ms
17,524 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <vector>
#include <iostream>
#include <algorithm>
using namespace std;

const long long INF = 3LL << 59;

class segment_tree {
private:
	int size;
	vector<long long> v;
public:
	segment_tree() : size(0), v() {}
	segment_tree(int n) : size(1 << (n >= 2 ? 32 - __builtin_clz(n - 1) : 0)), v(vector<long long>(size * 2, -INF)) {}
	void update(int pos, long long x) {
		pos += size;
		v[pos] = x;
		while (pos > 1) {
			pos /= 2;
			v[pos] = max(v[pos * 2 + 0], v[pos * 2 + 1]);
		}
	}
	long long range_max(int l, int r) {
		l += size;
		r += size;
		long long z = -INF;
		while (l != r) {
			if (l & 1) z = max(z, v[l++]);
			if (r & 1) z = max(z, v[--r]);
			l /= 2;
			r /= 2;
		}
		return z;
	}
};

int main() {
	// step #1. input
	cin.tie(nullptr);
	ios::sync_with_stdio(false);
	int N; long long K;
	cin >> N >> K;
	vector<long long> T(N), X(N), C(N);
	for (int i = 0; i < N; i++) {
		cin >> T[i];
	}
	for (int i = 0; i < N; i++) {
		cin >> X[i];
	}
	for (int i = 0; i < N; i++) {
		cin >> C[i];
	}

	// step #2. preparation
	vector<long long> PX(N), PY(N);
	for (int i = 0; i < N; i++) {
		PX[i] = T[i] * K - X[i];
		PY[i] = T[i] * K + X[i];
	}
	vector<long long> SY = PY;
	SY.push_back(0);
	sort(SY.begin(), SY.end());
	SY.erase(unique(SY.begin(), SY.end()), SY.end());
	int Z = SY.size();
	vector<int> perm(N);
	for (int i = 0; i < N; i++) {
		perm[i] = i;
	}
	sort(perm.begin(), perm.end(), [&](int a, int b) {
		return make_pair(PX[a], PY[a]) < make_pair(PX[b], PY[b]);
	});

	// step #3. dynamic programming
	segment_tree seg(Z);
	int base_pos = lower_bound(SY.begin(), SY.end(), 0) - SY.begin();
	seg.update(base_pos, 0);
	for (int i : perm) {
		int pos = lower_bound(SY.begin(), SY.end(), PY[i]) - SY.begin();
		long long x = seg.range_max(0, pos + 1);
		if (x != -INF) {
			seg.update(pos, x + C[i]);
		}
	}

	// step #4. get answer & output
	long long ans = seg.range_max(0, Z);
	cout << ans << endl;

	return 0;
}
0