結果

問題 No.2859 Falling Balls
ユーザー square1001square1001
提出日時 2024-07-15 12:25:08
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 173 ms / 3,000 ms
コード長 1,902 bytes
コンパイル時間 1,199 ms
コンパイル使用メモリ 89,044 KB
実行使用メモリ 19,696 KB
最終ジャッジ日時 2024-07-15 14:11:43
合計ジャッジ時間 5,800 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge6
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,944 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 1 ms
6,940 KB
testcase_04 AC 2 ms
6,940 KB
testcase_05 AC 2 ms
6,940 KB
testcase_06 AC 2 ms
6,944 KB
testcase_07 AC 2 ms
6,944 KB
testcase_08 AC 2 ms
6,944 KB
testcase_09 AC 2 ms
6,944 KB
testcase_10 AC 41 ms
7,504 KB
testcase_11 AC 130 ms
17,492 KB
testcase_12 AC 9 ms
6,940 KB
testcase_13 AC 125 ms
17,012 KB
testcase_14 AC 113 ms
16,104 KB
testcase_15 AC 40 ms
7,452 KB
testcase_16 AC 56 ms
10,056 KB
testcase_17 AC 55 ms
9,384 KB
testcase_18 AC 7 ms
6,944 KB
testcase_19 AC 76 ms
11,224 KB
testcase_20 AC 151 ms
18,568 KB
testcase_21 AC 147 ms
19,008 KB
testcase_22 AC 144 ms
18,584 KB
testcase_23 AC 146 ms
18,424 KB
testcase_24 AC 146 ms
18,792 KB
testcase_25 AC 151 ms
18,556 KB
testcase_26 AC 147 ms
18,584 KB
testcase_27 AC 147 ms
18,528 KB
testcase_28 AC 173 ms
19,696 KB
testcase_29 AC 147 ms
18,460 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;
	}
};

struct point {
	long long x, y, c;
	bool operator<(const point& p) const {
		return x != p.x ? x < p.x : y < p.y;
	}
};

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<point> P;
	vector<long long> SY = { 0 };
	for (int i = 0; i < N; i++) {
		long long x = T[i] * K - X[i];
		long long y = T[i] * K + X[i];
		if (x >= 0 && y >= 0) {
			P.push_back(point{x, y, C[i]});
			SY.push_back(y);
		}
	}
	sort(P.begin(), P.end());
	sort(SY.begin(), SY.end());
	SY.erase(unique(SY.begin(), SY.end()), SY.end());
	int M = P.size();
	int Z = SY.size();

	// step #3. dynamic programming
	segment_tree seg(Z);
	seg.update(0, 0);
	for (point p : P) {
		int pos = lower_bound(SY.begin(), SY.end(), p.y) - SY.begin();
		long long x = seg.range_max(0, pos + 1);
		if (x != INF) {
			seg.update(pos, x + p.c);
		}
	}

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

	return 0;
}
0