#include <bits/stdc++.h>
using namespace std;
#ifdef LOCAL
template<class T, class U>
ostream& operator<<(ostream& os, pair<T, U> p) {
return os << "(" << p.first << ", " << p.second << ")";
}
template<class C, class = typename C::value_type>
typename enable_if<!is_same<C, string>::value, ostream&>::type
operator<<(ostream& os, C c) {
for (auto i = c.begin(); i != c.end(); i++) {
os << " {"[i == c.begin()] << *i << ",}"[next(i) == c.end()];
}
return os;
}
#define debug(x) { cerr << #x << " = " << x << endl; }
#else
#define debug(...) {}
#endif
const int N = 1000 * 1000;
long long dp[N];
int dod[N];
int counter[N];
long long tab[N];
int cyf(long long x) {
int res = 0;
while (x > 0) {
res ++;
x /= 10;
}
return res;
}
vector<int> split(long long x) {
vector<int> res;
while (res.empty() || x > 0) {
res.push_back(x % 10);
x /= 10;
}
reverse(res.begin(), res.end());
return res;
}
bool check(int pos, long long t, int td, long long prev, int d, int cnt) {
if (cyf(t) + td < cyf(prev) + d) return false;
if (cyf(t) + td > cyf(prev) + d) {
// debug("Longer var\n");
dp[pos] = t;
dod[pos] = td;
counter[pos] = 0;
return true;
}
int x = cyf(t);
int y = cyf(prev);
vector<int> cnt_split = split(cnt + 1);
long long prev_cnt = prev;
if (cnt_split.size() > d) prev_cnt ++;
vector<int> t_split = split(t);
vector<int> prev_split = split(prev_cnt);
if (prev_split.size() + d > x + td) return false;
while (t_split.size() > prev_split.size()) {
assert(d >= 1);
if (d > cnt_split.size()) {
prev_split.push_back(0);
} else {
prev_split.push_back(cnt_split[cnt_split.size() - d]);
}
d --;
}
// debug(prev_split);
// debug(t_split);
for (int i = 0; i < t_split.size(); i ++) {
if (t_split[i] < prev_split[i]) return false;
if (t_split[i] > prev_split[i]) {
// debug("Bigger in t range\n");
dp[pos] = t;
dod[pos] = td;
counter[pos] = 0;
return true;
}
}
// debug("Equal\n");
dp[pos] = prev_cnt;
dod[pos] = dod[pos - 1];
counter[pos] = (prev_cnt == prev ? cnt + 1 : 0);
return true;
}
int main() {
long long res = 0;
ios_base::sync_with_stdio(false);
int n;
cin >> n;
for (int i = 0; i < n; i ++) {
cin >> tab[i];
}
dp[0] = tab[0];
for (int i = 1; i < n; i ++) {
for (int j = max(dod[i-1] - cyf(tab[i]), 0); ; j ++) {
if (check(i, tab[i], j, dp[i - 1], dod[i - 1], counter[i - 1])) {
res += j;
// cerr << tab[i] << "\t" << dp[i] << "\t" << dod[i] << "\n";
break;
}
}
}
cout << res << "\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 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 110 111 112 113 114 115 116 117 | #include <bits/stdc++.h> using namespace std; #ifdef LOCAL template<class T, class U> ostream& operator<<(ostream& os, pair<T, U> p) { return os << "(" << p.first << ", " << p.second << ")"; } template<class C, class = typename C::value_type> typename enable_if<!is_same<C, string>::value, ostream&>::type operator<<(ostream& os, C c) { for (auto i = c.begin(); i != c.end(); i++) { os << " {"[i == c.begin()] << *i << ",}"[next(i) == c.end()]; } return os; } #define debug(x) { cerr << #x << " = " << x << endl; } #else #define debug(...) {} #endif const int N = 1000 * 1000; long long dp[N]; int dod[N]; int counter[N]; long long tab[N]; int cyf(long long x) { int res = 0; while (x > 0) { res ++; x /= 10; } return res; } vector<int> split(long long x) { vector<int> res; while (res.empty() || x > 0) { res.push_back(x % 10); x /= 10; } reverse(res.begin(), res.end()); return res; } bool check(int pos, long long t, int td, long long prev, int d, int cnt) { if (cyf(t) + td < cyf(prev) + d) return false; if (cyf(t) + td > cyf(prev) + d) { // debug("Longer var\n"); dp[pos] = t; dod[pos] = td; counter[pos] = 0; return true; } int x = cyf(t); int y = cyf(prev); vector<int> cnt_split = split(cnt + 1); long long prev_cnt = prev; if (cnt_split.size() > d) prev_cnt ++; vector<int> t_split = split(t); vector<int> prev_split = split(prev_cnt); if (prev_split.size() + d > x + td) return false; while (t_split.size() > prev_split.size()) { assert(d >= 1); if (d > cnt_split.size()) { prev_split.push_back(0); } else { prev_split.push_back(cnt_split[cnt_split.size() - d]); } d --; } // debug(prev_split); // debug(t_split); for (int i = 0; i < t_split.size(); i ++) { if (t_split[i] < prev_split[i]) return false; if (t_split[i] > prev_split[i]) { // debug("Bigger in t range\n"); dp[pos] = t; dod[pos] = td; counter[pos] = 0; return true; } } // debug("Equal\n"); dp[pos] = prev_cnt; dod[pos] = dod[pos - 1]; counter[pos] = (prev_cnt == prev ? cnt + 1 : 0); return true; } int main() { long long res = 0; ios_base::sync_with_stdio(false); int n; cin >> n; for (int i = 0; i < n; i ++) { cin >> tab[i]; } dp[0] = tab[0]; for (int i = 1; i < n; i ++) { for (int j = max(dod[i-1] - cyf(tab[i]), 0); ; j ++) { if (check(i, tab[i], j, dp[i - 1], dod[i - 1], counter[i - 1])) { res += j; // cerr << tab[i] << "\t" << dp[i] << "\t" << dod[i] << "\n"; break; } } } cout << res << "\n"; } |
English