#include <iostream>
#include <cmath>
#include <unordered_map>
#include <map>
#include <set>
#include <queue>
#include <vector>
#include <string>
#include <iomanip>
#include <algorithm>
#define all(x) (x).begin(), (x).end()
#define rall(x) (x).rbegin(), (x).rend()
using namespace std;
typedef long long ll;
ll linf = 1e15+1;
inline void scoobydoobydoo(){
ios::sync_with_stdio(false);
ios_base::sync_with_stdio(false);
cin.tie(NULL); cout.tie(NULL);
}
set<int> bals[1000003];
int main(){
scoobydoobydoo();
int n; cin >> n;
set<pair<int, int>> v;
vector<int> val;
for (int i = 0; i < n; i++){
int h; cin >> h;
v.insert({i, h});
val.push_back(h);
bals[h].insert(i);
}
if (n <= 5000){
int sum = 0;
for (int i = 0; i < n; i++){
if (val[i] == -1)continue;
sum++;
int h = val[i]-1;
for (int j = i+1; j < n; j++){
if (h == 0)break;
if (val[j] == h){
val[j] = -1;
h--;
}
}
}
cout << sum << endl;
return 0;
}
int sum = 0;
while (!v.empty()){
sum++;
int h = (*v.begin()).second;
int i = (*v.begin()).first;
v.erase(v.begin());
if (v.empty())break;
if (bals[h-1].empty())break;
auto it = bals[h-1].upper_bound(i);
while (h >= 1 && it != bals[h-1].end()){
auto itr = v.find({*it, h-1});
v.erase(itr);
bals[h-1].erase(it);
i = *it;
h--;
if (bals[h-1].empty())break;
it = bals[h-1].upper_bound(i);
}
}
cout << sum << endl;
return 0;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
13 ms |
47452 KB |
Output is correct |
2 |
Correct |
11 ms |
47448 KB |
Output is correct |
3 |
Correct |
14 ms |
47708 KB |
Output is correct |
4 |
Correct |
16 ms |
47708 KB |
Output is correct |
5 |
Correct |
1565 ms |
125192 KB |
Output is correct |
6 |
Runtime error |
497 ms |
131072 KB |
Execution killed with signal 9 |
7 |
Correct |
1268 ms |
119140 KB |
Output is correct |
8 |
Correct |
1311 ms |
118312 KB |
Output is correct |
9 |
Correct |
1443 ms |
122808 KB |
Output is correct |
10 |
Correct |
1529 ms |
125312 KB |
Output is correct |