#include <iostream>
#include <string>
#include <vector>
using namespace std;
int main() {
ios_base::sync_with_stdio(false);
cin.tie(NULL);
string a, b, c;
if (!(cin >> a >> b >> c)) return 0;
int n = a.length();
vector<long long> dp0(n, 0);
vector<long long> dp1(n, 0);
long long total_count = 0;
for (int k = n - 1; k >= 0; --k) {
int d1 = a[k] - '0';
int d2 = b[k] - '0';
int d3 = c[k] - '0';
if (d1 + d2 == d3) {
dp0[k] = 1;
}
else if (d1 + d2 == 10 + d3) {
dp1[k] = 1;
}
if (k < n - 1) {
if (d1 + d2 == d3) {
dp0[k] += dp0[k + 1];
}
if (d1 + d2 + 1 == d3) {
dp0[k] += dp1[k + 1];
}
if (d1 + d2 == 10 + d3) {
dp1[k] += dp0[k + 1];
}
if (d1 + d2 + 1 == 10 + d3) {
dp1[k] += dp1[k + 1];
}
}
total_count += dp0[k];
}
cout << total_count;
}
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 | #include <iostream> #include <string> #include <vector> using namespace std; int main() { ios_base::sync_with_stdio(false); cin.tie(NULL); string a, b, c; if (!(cin >> a >> b >> c)) return 0; int n = a.length(); vector<long long> dp0(n, 0); vector<long long> dp1(n, 0); long long total_count = 0; for (int k = n - 1; k >= 0; --k) { int d1 = a[k] - '0'; int d2 = b[k] - '0'; int d3 = c[k] - '0'; if (d1 + d2 == d3) { dp0[k] = 1; } else if (d1 + d2 == 10 + d3) { dp1[k] = 1; } if (k < n - 1) { if (d1 + d2 == d3) { dp0[k] += dp0[k + 1]; } if (d1 + d2 + 1 == d3) { dp0[k] += dp1[k + 1]; } if (d1 + d2 == 10 + d3) { dp1[k] += dp0[k + 1]; } if (d1 + d2 + 1 == 10 + d3) { dp1[k] += dp1[k + 1]; } } total_count += dp0[k]; } cout << total_count; } |
English