#include <iostream>
#include <vector>
using namespace std;
int main() {
ios_base::sync_with_stdio(false);
cin.tie(nullptr);
string s1, s2, s3;
vector<int> n1, n2, n3;
cin >> s1 >> s2 >> s3;
for (char c : s1) {
n1.push_back(c - '0');
}
for (char c : s2) {
n2.push_back(c - '0');
}
for (char c : s3) {
n3.push_back(c - '0');
}
int state = -1;
long long count = 0;
long long res{};
for (int i = size(n1) - 1; i >= 0; --i) {
if (n1[i] + n2[i] == n3[i]) {
if (state == 0) {
++count;
} else if (state == 1) {
res += (count + 1) * count / 2;
count = 1;
state = 0;
} else if (state == -1) {
count = 1;
state = 0;
}
} else if (n1[i] + n2[i] == n3[i] + 10) {
if (state == 0) {
state = 1;
} else if (state == 1) {
res += (count + 1) * count / 2;
count = 0;
state = 1;
} else if (state == -1) {
count = 0;
state = 1;
}
} else if (n1[i] + n2[i] + 1 == n3[i]) {
if (state == 0) {
res += (count + 1) * count / 2;
count = 0;
state = -1;
} else if (state == 1) {
state = 0;
++count;
} else if (state == -1) {
// carry on
}
} else if (n1[i] + n2[i] + 1 == n3[i] + 10) {
if (state == 0) {
res += (count + 1) * count / 2;
count = 0;
state = -1;
} else if (state == 1) {
// carry on
} else if (state == -1) {
// carry on
}
} else {
if (state != -1) {
res += (count + 1) * count / 2;
count = 0;
}
state = -1;
}
}
if (state != -1) {
res += (count + 1) * count / 2;
}
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 | #include <iostream> #include <vector> using namespace std; int main() { ios_base::sync_with_stdio(false); cin.tie(nullptr); string s1, s2, s3; vector<int> n1, n2, n3; cin >> s1 >> s2 >> s3; for (char c : s1) { n1.push_back(c - '0'); } for (char c : s2) { n2.push_back(c - '0'); } for (char c : s3) { n3.push_back(c - '0'); } int state = -1; long long count = 0; long long res{}; for (int i = size(n1) - 1; i >= 0; --i) { if (n1[i] + n2[i] == n3[i]) { if (state == 0) { ++count; } else if (state == 1) { res += (count + 1) * count / 2; count = 1; state = 0; } else if (state == -1) { count = 1; state = 0; } } else if (n1[i] + n2[i] == n3[i] + 10) { if (state == 0) { state = 1; } else if (state == 1) { res += (count + 1) * count / 2; count = 0; state = 1; } else if (state == -1) { count = 0; state = 1; } } else if (n1[i] + n2[i] + 1 == n3[i]) { if (state == 0) { res += (count + 1) * count / 2; count = 0; state = -1; } else if (state == 1) { state = 0; ++count; } else if (state == -1) { // carry on } } else if (n1[i] + n2[i] + 1 == n3[i] + 10) { if (state == 0) { res += (count + 1) * count / 2; count = 0; state = -1; } else if (state == 1) { // carry on } else if (state == -1) { // carry on } } else { if (state != -1) { res += (count + 1) * count / 2; count = 0; } state = -1; } } if (state != -1) { res += (count + 1) * count / 2; } cout << res << "\n"; } |
English