#include <iostream>
#include <iomanip>
#include <string>
#include <algorithm>
using namespace std;
//#define DEBUG
#ifdef DEBUG
#undef DEBUG
#define DEBUG(x) x
#else
#define DEBUG(x)
#endif
#define REP(x,n) for(int x=0;x<(n);++x)
#define VAR(x,n) __typeof(n) x = (n)
#define FOREACH(x,c) for(VAR(x, (c).begin()); x != (c).end(); ++x)
#define CONTAINS(x,elem) ((x).find(elem) != (x).end())
const int MAX_N = 500002;
const int MAX_M = 500002;
//highest score for tower ending with color [i]
struct BestTower {
long long score;
int size;
} bestForColors[MAX_M];
long long solve() {
int n,fine;
cin>>n>>fine;
REP(x,n) {
}
/* 2 scenarios:
1) do not add this brick - score unchanged
2) add this brick:
a) to the last brick (with fine)
b) to the highest tower containing same-color brick (without fine)
*/
long long highestScore = 0;
long long highestScoreForPreviousSize = 0;
int currentSize = 0;
int length, color;
REP(x,n) {
cin >> length >> color;
if (currentSize != length) {
currentSize = length;
highestScoreForPreviousSize = highestScore;
}
BestTower& bestForColor = bestForColors[color];
if (bestForColor.size == length) {
DEBUG(cerr<<"skipping step "<<x<<" due to repeated brick"<<endl;)
continue;
}
bestForColor.score = max(bestForColor.score + length, highestScoreForPreviousSize - fine + length);
bestForColor.size = length;
highestScore = max(highestScore, bestForColor.score);
DEBUG(cerr<<"after step "<<x<<" highest score is "<<highestScoreForPreviousSize << " / "<< highestScore << endl;)
}
return highestScore;
}
int main() {
ios_base::sync_with_stdio(0);
cout << solve() << endl;
}
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 | #include <iostream> #include <iomanip> #include <string> #include <algorithm> using namespace std; //#define DEBUG #ifdef DEBUG #undef DEBUG #define DEBUG(x) x #else #define DEBUG(x) #endif #define REP(x,n) for(int x=0;x<(n);++x) #define VAR(x,n) __typeof(n) x = (n) #define FOREACH(x,c) for(VAR(x, (c).begin()); x != (c).end(); ++x) #define CONTAINS(x,elem) ((x).find(elem) != (x).end()) const int MAX_N = 500002; const int MAX_M = 500002; //highest score for tower ending with color [i] struct BestTower { long long score; int size; } bestForColors[MAX_M]; long long solve() { int n,fine; cin>>n>>fine; REP(x,n) { } /* 2 scenarios: 1) do not add this brick - score unchanged 2) add this brick: a) to the last brick (with fine) b) to the highest tower containing same-color brick (without fine) */ long long highestScore = 0; long long highestScoreForPreviousSize = 0; int currentSize = 0; int length, color; REP(x,n) { cin >> length >> color; if (currentSize != length) { currentSize = length; highestScoreForPreviousSize = highestScore; } BestTower& bestForColor = bestForColors[color]; if (bestForColor.size == length) { DEBUG(cerr<<"skipping step "<<x<<" due to repeated brick"<<endl;) continue; } bestForColor.score = max(bestForColor.score + length, highestScoreForPreviousSize - fine + length); bestForColor.size = length; highestScore = max(highestScore, bestForColor.score); DEBUG(cerr<<"after step "<<x<<" highest score is "<<highestScoreForPreviousSize << " / "<< highestScore << endl;) } return highestScore; } int main() { ios_base::sync_with_stdio(0); cout << solve() << endl; } |
English