#include <algorithm>
#include <iostream>
#include <vector>
#include <list>
#include <cassert>
#include <cstring>
using namespace std;
#define FOR(i, n) for(int i = 0, __n = (n); i < __n; i++)
const int MAXN = 1000010;
char A[MAXN];
char B[MAXN];
char S[MAXN];
int P[MAXN][2];
int main() {
scanf("%s%s%s", A+1, B+1, S+1);
int n = strlen(S+1);
long long res = 0;
int p0 = 0, p1 = 0;
P[0][0] = P[0][1] = 0;
for (int i = 1; i <= n; i++) {
int s = S[i] - '0';
int a = A[i] - '0';
int b = B[i] - '0';
if (a + b == s) {
// suma się zgadza, nie ma overflow
P[i][0] = P[i-1][0] + 1;
P[i][1] = 0; // overflow nam wszystko popsuje
} else if (a + b + 1 == s) {
// S jest o jeden większe, czyli czekamy na overflow
P[i][0] = 0;
P[i][1] = P[i-1][0] + 1;
} else if (a + b == s + 10) {
// S jest o 10 mniejsze, czyli przekazujemy overflow
P[i][0] = P[i-1][1] + 0;
P[i][1] = 0;
} else if (a + b + 1 == s + 10) {
P[i][0] = 0;
P[i][1] = P[i-1][1];
} else {
P[i][0] = P[i][1] = 0;
}
res += P[i][0];
}
printf("%lld\n", res);
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 | #include <algorithm> #include <iostream> #include <vector> #include <list> #include <cassert> #include <cstring> using namespace std; #define FOR(i, n) for(int i = 0, __n = (n); i < __n; i++) const int MAXN = 1000010; char A[MAXN]; char B[MAXN]; char S[MAXN]; int P[MAXN][2]; int main() { scanf("%s%s%s", A+1, B+1, S+1); int n = strlen(S+1); long long res = 0; int p0 = 0, p1 = 0; P[0][0] = P[0][1] = 0; for (int i = 1; i <= n; i++) { int s = S[i] - '0'; int a = A[i] - '0'; int b = B[i] - '0'; if (a + b == s) { // suma się zgadza, nie ma overflow P[i][0] = P[i-1][0] + 1; P[i][1] = 0; // overflow nam wszystko popsuje } else if (a + b + 1 == s) { // S jest o jeden większe, czyli czekamy na overflow P[i][0] = 0; P[i][1] = P[i-1][0] + 1; } else if (a + b == s + 10) { // S jest o 10 mniejsze, czyli przekazujemy overflow P[i][0] = P[i-1][1] + 0; P[i][1] = 0; } else if (a + b + 1 == s + 10) { P[i][0] = 0; P[i][1] = P[i-1][1]; } else { P[i][0] = P[i][1] = 0; } res += P[i][0]; } printf("%lld\n", res); return 0; } |
English