#include <bits/stdc++.h>
using namespace std;
using ll = long long;
struct number {
string prefix;
int zeros;
string suffix;
int len() const {
return prefix.size() + zeros + suffix.size();
}
int operator[](int ind) const {
if (ind < (int)prefix.size()) {
return prefix[ind] - '0';
}
ind -= prefix.size();
if (ind < zeros) {
return 0;
}
ind -= zeros;
return suffix[ind] - '0';
}
bool operator<(number const& rhs) const {
if (len() != rhs.len()) {
return len() < rhs.len();
}
for (int i = 0; i < (int)max(prefix.size(), rhs.prefix.size()); i++) {
if ((*this)[i] != rhs[i]) {
return (*this)[i] < rhs[i];
}
}
for (int i = len() - max(suffix.size(), rhs.suffix.size()); i < len(); i++) {
if ((*this)[i] != rhs[i]) {
return (*this)[i] < rhs[i];
}
}
return false;
}
number add1() const {
number res = *this;
for (int i = (int)(res.suffix.size()) - 1; i >= 0; i--) {
if (res.suffix[i] != '9') {
res.suffix[i]++;
return res;
}
res.suffix[i] = '0';
}
if (res.zeros) {
res.zeros--;
res.suffix = string("1") + res.suffix;
return res;
}
res.zeros = res.suffix.size();
res.suffix.clear();
for (int i = (int)(res.prefix.size()) - 1; i >= 0; i--) {
if (res.prefix[i] != '9') {
res.prefix[i]++;
break;
}
res.zeros++;
res.prefix.pop_back();
}
if (res.prefix.empty()) {
res.prefix = "1";
}
return res;
}
};
int main() {
int n;
cin >> n;
number prv{};
ll res = 0;
while (n--) {
string s;
cin >> s;
number current{std::move(s), 0, ""};
if (prv < current) {
prv = std::move(current);
} else {
auto p1 = prv.add1();
bool pref = true;
for (int i = 0; i < (int)current.prefix.size(); i++) {
if (current[i] != p1[i]) {
pref = false;
break;
}
}
if (pref) {
res += p1.len() - current.len();
prv = std::move(p1);
} else {
current.zeros = prv.len() - current.len();
if (!(prv < current)) {
current.zeros++;
}
res += current.zeros;
prv = std::move(current);
}
}
}
cout << res << 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 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 | #include <bits/stdc++.h> using namespace std; using ll = long long; struct number { string prefix; int zeros; string suffix; int len() const { return prefix.size() + zeros + suffix.size(); } int operator[](int ind) const { if (ind < (int)prefix.size()) { return prefix[ind] - '0'; } ind -= prefix.size(); if (ind < zeros) { return 0; } ind -= zeros; return suffix[ind] - '0'; } bool operator<(number const& rhs) const { if (len() != rhs.len()) { return len() < rhs.len(); } for (int i = 0; i < (int)max(prefix.size(), rhs.prefix.size()); i++) { if ((*this)[i] != rhs[i]) { return (*this)[i] < rhs[i]; } } for (int i = len() - max(suffix.size(), rhs.suffix.size()); i < len(); i++) { if ((*this)[i] != rhs[i]) { return (*this)[i] < rhs[i]; } } return false; } number add1() const { number res = *this; for (int i = (int)(res.suffix.size()) - 1; i >= 0; i--) { if (res.suffix[i] != '9') { res.suffix[i]++; return res; } res.suffix[i] = '0'; } if (res.zeros) { res.zeros--; res.suffix = string("1") + res.suffix; return res; } res.zeros = res.suffix.size(); res.suffix.clear(); for (int i = (int)(res.prefix.size()) - 1; i >= 0; i--) { if (res.prefix[i] != '9') { res.prefix[i]++; break; } res.zeros++; res.prefix.pop_back(); } if (res.prefix.empty()) { res.prefix = "1"; } return res; } }; int main() { int n; cin >> n; number prv{}; ll res = 0; while (n--) { string s; cin >> s; number current{std::move(s), 0, ""}; if (prv < current) { prv = std::move(current); } else { auto p1 = prv.add1(); bool pref = true; for (int i = 0; i < (int)current.prefix.size(); i++) { if (current[i] != p1[i]) { pref = false; break; } } if (pref) { res += p1.len() - current.len(); prv = std::move(p1); } else { current.zeros = prv.len() - current.len(); if (!(prv < current)) { current.zeros++; } res += current.zeros; prv = std::move(current); } } } cout << res << endl; } |
English