#include <bits/stdc++.h>
using namespace std;
#define st first
#define nd second
#define pb push_back
#define all(x) (x).begin(), (x).end()
#define BOOST ios_base::sync_with_stdio(0), cin.tie(0)
template<typename T>
using vc = vector<T>;
using ll = long long;
using ld = long double;
using ii = pair<int, int>;
const int N = 1e6 + 5;
int dp[2][N];
int main(){
BOOST;
string _a, _b, _c;
cin >> _a >> _b >> _c;
int n = _a.size();
vc<int> a(n+2), b(n+2), c(n+2);
for(int i=1; i<=n; i++){
a[i] = _a[i-1] - '0';
b[i] = _b[i-1] - '0';
c[i] = _c[i-1] - '0';
}
ll ans = 0;
for(int i=n; i>=1; i--){
int sum = a[i] + b[i];
if(sum == c[i]){
dp[0][i] = dp[0][i+1] + 1;
}
if(sum-10 == c[i]){
dp[1][i] = dp[0][i+1] + 1;
}
if(sum+1 == c[i]){
dp[0][i] = dp[1][i+1];
}
if(sum+1-10 == c[i]){
dp[1][i] = dp[1][i+1];
}
ans += dp[0][i];
}
cout << ans << "\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 | #include <bits/stdc++.h> using namespace std; #define st first #define nd second #define pb push_back #define all(x) (x).begin(), (x).end() #define BOOST ios_base::sync_with_stdio(0), cin.tie(0) template<typename T> using vc = vector<T>; using ll = long long; using ld = long double; using ii = pair<int, int>; const int N = 1e6 + 5; int dp[2][N]; int main(){ BOOST; string _a, _b, _c; cin >> _a >> _b >> _c; int n = _a.size(); vc<int> a(n+2), b(n+2), c(n+2); for(int i=1; i<=n; i++){ a[i] = _a[i-1] - '0'; b[i] = _b[i-1] - '0'; c[i] = _c[i-1] - '0'; } ll ans = 0; for(int i=n; i>=1; i--){ int sum = a[i] + b[i]; if(sum == c[i]){ dp[0][i] = dp[0][i+1] + 1; } if(sum-10 == c[i]){ dp[1][i] = dp[0][i+1] + 1; } if(sum+1 == c[i]){ dp[0][i] = dp[1][i+1]; } if(sum+1-10 == c[i]){ dp[1][i] = dp[1][i+1]; } ans += dp[0][i]; } cout << ans << "\n"; } |
English