#include <bits/stdc++.h> using namespace std; using i64 = long long; int main() { int n, cost; cin >> n >> cost; vector<int> hei(n), col(n); for (int i = 0; i < n; i++) { cin >> hei[i] >> col[i]; } set<pair<i64,int>> pairs; map<int,i64> dp; auto consider = [&](int x, i64 v) { if (dp[x] >= v) return; pairs.erase({dp[x], x}); dp[x] = v; pairs.insert({v, x}); }; consider(0, 0); for (int i = 0; i < n; i++) { if (i == 0 || hei[i] > hei[i-1]) { set<int> xs; for (int j = i; j < n && hei[j] == hei[i]; j++) { xs.insert(col[j]); } i64 best = pairs.empty() ? 0 : rbegin(pairs)->first; for (int x: xs) { consider(x, hei[i] + max(dp[x], best-cost)); } } } cout << rbegin(pairs)->first << '\n'; }
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 | #include <bits/stdc++.h> using namespace std; using i64 = long long; int main() { int n, cost; cin >> n >> cost; vector<int> hei(n), col(n); for (int i = 0; i < n; i++) { cin >> hei[i] >> col[i]; } set<pair<i64,int>> pairs; map<int,i64> dp; auto consider = [&](int x, i64 v) { if (dp[x] >= v) return; pairs.erase({dp[x], x}); dp[x] = v; pairs.insert({v, x}); }; consider(0, 0); for (int i = 0; i < n; i++) { if (i == 0 || hei[i] > hei[i-1]) { set<int> xs; for (int j = i; j < n && hei[j] == hei[i]; j++) { xs.insert(col[j]); } i64 best = pairs.empty() ? 0 : rbegin(pairs)->first; for (int x: xs) { consider(x, hei[i] + max(dp[x], best-cost)); } } } cout << rbegin(pairs)->first << '\n'; } |