#include <iostream> #include <string> using namespace std; int zeroes = 0; int cnt = 0; long long res = 0; void fill_with_zeroes(string a, string b) { if(a.size() + zeroes > b.size()) { zeroes = a.size() + zeroes - b.size(); } else { zeroes = 0; } cnt = 0; } void fill_with_zeroes_plus(string a, string b) { if(a.size() + zeroes >= b.size()) { zeroes = a.size() + zeroes - b.size() + 1; } else { zeroes = 0; } cnt = 0; } void increment(string aF, string aT, string b) { // cout << "Incrementation" << endl; // cout << "aF: " << aF << " + aT: " << aT << ", b:" << b << endl; int aTi = 1; if(aT != "") { aTi += stoi(aT+string(zeroes, '0')); } aTi += cnt; string aTn = to_string(aTi); // cout << "aTn: " << aTn << endl; zeroes += aT.size(); if(aTn.size() > zeroes) { zeroes += 1; cnt = 0; } else { cnt = aTi; } } int main() { int n; cin >> n; string a, b; for(int i = 0; i < n; ++i) { cin >> b; if(i > 0) { if(b.size() > a.size()) { int filled_a = stoi(a+string(b.size()-a.size(), '0')); if(stoi(b) <= filled_a) { fill_with_zeroes_plus(a, b); } else { fill_with_zeroes(a, b); } } else if(b.size() == a.size() && stoi(b) > stoi(a)) { fill_with_zeroes(a, b); } else if(b.size() == a.size() && stoi(b) == stoi(a)) { increment(a, "", b); } else if(b.size() == a.size() && stoi(b) < stoi(a)) { fill_with_zeroes_plus(a, b); } else if(b.size() < a.size()) { string aF = a.substr(0, b.size()); string aT = a.substr(b.size()); if(aF < b) { fill_with_zeroes(a, b); } else if(aF == b) { increment(aF, aT, b); } else if(aF > b) { fill_with_zeroes_plus(a, b); } } } res += zeroes; a = b; // cout << a << endl; // cout << zeroes << ", " << cnt << ", " << res << endl; // cout << "----------------------------" << endl << endl; } cout << res << endl; return 0; }
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 | #include <iostream> #include <string> using namespace std; int zeroes = 0; int cnt = 0; long long res = 0; void fill_with_zeroes(string a, string b) { if(a.size() + zeroes > b.size()) { zeroes = a.size() + zeroes - b.size(); } else { zeroes = 0; } cnt = 0; } void fill_with_zeroes_plus(string a, string b) { if(a.size() + zeroes >= b.size()) { zeroes = a.size() + zeroes - b.size() + 1; } else { zeroes = 0; } cnt = 0; } void increment(string aF, string aT, string b) { // cout << "Incrementation" << endl; // cout << "aF: " << aF << " + aT: " << aT << ", b:" << b << endl; int aTi = 1; if(aT != "") { aTi += stoi(aT+string(zeroes, '0')); } aTi += cnt; string aTn = to_string(aTi); // cout << "aTn: " << aTn << endl; zeroes += aT.size(); if(aTn.size() > zeroes) { zeroes += 1; cnt = 0; } else { cnt = aTi; } } int main() { int n; cin >> n; string a, b; for(int i = 0; i < n; ++i) { cin >> b; if(i > 0) { if(b.size() > a.size()) { int filled_a = stoi(a+string(b.size()-a.size(), '0')); if(stoi(b) <= filled_a) { fill_with_zeroes_plus(a, b); } else { fill_with_zeroes(a, b); } } else if(b.size() == a.size() && stoi(b) > stoi(a)) { fill_with_zeroes(a, b); } else if(b.size() == a.size() && stoi(b) == stoi(a)) { increment(a, "", b); } else if(b.size() == a.size() && stoi(b) < stoi(a)) { fill_with_zeroes_plus(a, b); } else if(b.size() < a.size()) { string aF = a.substr(0, b.size()); string aT = a.substr(b.size()); if(aF < b) { fill_with_zeroes(a, b); } else if(aF == b) { increment(aF, aT, b); } else if(aF > b) { fill_with_zeroes_plus(a, b); } } } res += zeroes; a = b; // cout << a << endl; // cout << zeroes << ", " << cnt << ", " << res << endl; // cout << "----------------------------" << endl << endl; } cout << res << endl; return 0; } |