#include <algorithm>
#include <bitset>
#include <cmath>
#include <cstdlib>
#include <ctime>
#include <deque>
#include <iostream>
#include <list>
#include <map>
#include <queue>
#include <set>
#include <stack>
#include <string>
#include <vector>
using namespace std;
#define JOIN_(X, Y) X##Y
#define JOIN(X, Y) JOIN_(X, Y)
#define TMP JOIN(tmp, __LINE__)
#define PB push_back
#define SZ(x) int((x).size())
#define REP(i, n) for (int i = 0, TMP = (n); i < TMP; ++i)
#define FOR(i, a, b) for (int i = (a), TMP = (b); i <= TMP; ++i)
#define FORD(i, a, b) for (int i = (a), TMP = (b); i >= TMP; --i)
#ifdef DEBUG
#define DEB(x) (cerr << x)
#else
#define DEB(x)
#endif
typedef long long ll;
typedef vector<int> vi;
typedef pair<int, int> pii;
typedef unsigned int uint;
const int INF = 1e9 + 9;
struct Result {
int color;
ll result;
ll result_with_color;
};
void apply_pending(ll &best, vector<ll> &best_by_color, const vector<Result> &pending) {
for (const auto &result : pending) {
best_by_color[result.color] = max(best_by_color[result.color], result.result_with_color);
best = max({best, result.result, best_by_color[result.color]});
}
}
void inline one() {
int n, penalty;
cin >> n >> penalty;
ll best = 0;
vector<ll> best_by_color(500001);
int last_size = 0;
vector<Result> pending;
REP(i, n) {
int size, color;
cin >> size >> color;
DEB("size=" << size << " color=" << color << "\n");
if (size > last_size) {
DEB("apply pending\n");
apply_pending(best, best_by_color, pending);
last_size = size;
pending.clear();
}
ll result_with_color =
max({best_by_color[color], best_by_color[color] + size, best + size - penalty});
ll result = max({best, best + size - penalty, result_with_color});
DEB("result=" << result << " result_with_color=" << result_with_color << "\n");
pending.PB({.color = color, .result = result, .result_with_color = result_with_color});
}
apply_pending(best, best_by_color, pending);
ll bestest = best;
for (ll x : best_by_color) {
bestest = max(bestest, x);
}
cout << bestest << "\n";
}
int main() {
ios::sync_with_stdio(false);
cin.tie(0);
// int z; cin >> z; while(z--)
one();
}
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 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 | #include <algorithm> #include <bitset> #include <cmath> #include <cstdlib> #include <ctime> #include <deque> #include <iostream> #include <list> #include <map> #include <queue> #include <set> #include <stack> #include <string> #include <vector> using namespace std; #define JOIN_(X, Y) X##Y #define JOIN(X, Y) JOIN_(X, Y) #define TMP JOIN(tmp, __LINE__) #define PB push_back #define SZ(x) int((x).size()) #define REP(i, n) for (int i = 0, TMP = (n); i < TMP; ++i) #define FOR(i, a, b) for (int i = (a), TMP = (b); i <= TMP; ++i) #define FORD(i, a, b) for (int i = (a), TMP = (b); i >= TMP; --i) #ifdef DEBUG #define DEB(x) (cerr << x) #else #define DEB(x) #endif typedef long long ll; typedef vector<int> vi; typedef pair<int, int> pii; typedef unsigned int uint; const int INF = 1e9 + 9; struct Result { int color; ll result; ll result_with_color; }; void apply_pending(ll &best, vector<ll> &best_by_color, const vector<Result> &pending) { for (const auto &result : pending) { best_by_color[result.color] = max(best_by_color[result.color], result.result_with_color); best = max({best, result.result, best_by_color[result.color]}); } } void inline one() { int n, penalty; cin >> n >> penalty; ll best = 0; vector<ll> best_by_color(500001); int last_size = 0; vector<Result> pending; REP(i, n) { int size, color; cin >> size >> color; DEB("size=" << size << " color=" << color << "\n"); if (size > last_size) { DEB("apply pending\n"); apply_pending(best, best_by_color, pending); last_size = size; pending.clear(); } ll result_with_color = max({best_by_color[color], best_by_color[color] + size, best + size - penalty}); ll result = max({best, best + size - penalty, result_with_color}); DEB("result=" << result << " result_with_color=" << result_with_color << "\n"); pending.PB({.color = color, .result = result, .result_with_color = result_with_color}); } apply_pending(best, best_by_color, pending); ll bestest = best; for (ll x : best_by_color) { bestest = max(bestest, x); } cout << bestest << "\n"; } int main() { ios::sync_with_stdio(false); cin.tie(0); // int z; cin >> z; while(z--) one(); } |
English