#include<bits/stdc++.h>
using namespace std;
#define rep(i, a, b) for (int i = (a); i < (b); i++)
#define all(x) begin(x), end(x)
#define sz(x) (int)(x).size()
#define ST first
#define ND second
#define PB push_back
using ll = long long;
using ld = long double;
using pi = pair<int,int>;
using vi = vector<int>;
using vvi = vector<vi>;
using vll = vector<ll>;
#ifdef LOCAL
#define DTP(x, y) \
auto operator<<(auto &o, auto a)->decltype(y, o) \
{ \
o << "("; \
x; \
return o << ")"; \
}
DTP(o << a.first << ", " << a.second, a.second);
DTP(for (auto i : a) o << i << ", ", all(a));
void dump(auto... x) { ((cerr << x << ", "), ...) << '\n'; }
#define debug(x...) cerr << setw(4) << __LINE__ << ":[" #x "]: ", dump(x)
#else
#define debug(...) 0
#endif
void Solve()
{
int n;
string s;
cin >> s;
n=s.length();
vi A(n+1);
vi B(n+1);
vi C(n+1);
rep(i, 1, n+1){
A[i]=s[i-1]-'0';
}
cin >> s;
rep(i, 1, n+1){
B[i]=s[i-1]-'0';
}
cin >> s;
rep(i, 1, n+1){
C[i]=s[i-1]-'0';
}
vi dp(n+1);
rep(i, 1, n+1){
if(A[i]+B[i]==C[i])
dp[i]=1+dp[i-1];
if(A[i]+B[i]==C[i]+10){
for(int j=i-1; j>=0; --j){
if(A[j]+B[j]+1==C[j]){
dp[i]=dp[j-1]+1;
break;
}else if(A[j]+B[j]+1!=C[j]+10){
break;
}
}
}
}
ll res=0;
rep(i, 1, n+1)
res+=(ll)(dp[i]);
cout << res << "\n";
return;
}
int main()
{
ios_base::sync_with_stdio(false);
cin.tie(0);
int t=1;
// cin >> t;
while(t--)
{
Solve();
}
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 88 89 | #include<bits/stdc++.h> using namespace std; #define rep(i, a, b) for (int i = (a); i < (b); i++) #define all(x) begin(x), end(x) #define sz(x) (int)(x).size() #define ST first #define ND second #define PB push_back using ll = long long; using ld = long double; using pi = pair<int,int>; using vi = vector<int>; using vvi = vector<vi>; using vll = vector<ll>; #ifdef LOCAL #define DTP(x, y) \ auto operator<<(auto &o, auto a)->decltype(y, o) \ { \ o << "("; \ x; \ return o << ")"; \ } DTP(o << a.first << ", " << a.second, a.second); DTP(for (auto i : a) o << i << ", ", all(a)); void dump(auto... x) { ((cerr << x << ", "), ...) << '\n'; } #define debug(x...) cerr << setw(4) << __LINE__ << ":[" #x "]: ", dump(x) #else #define debug(...) 0 #endif void Solve() { int n; string s; cin >> s; n=s.length(); vi A(n+1); vi B(n+1); vi C(n+1); rep(i, 1, n+1){ A[i]=s[i-1]-'0'; } cin >> s; rep(i, 1, n+1){ B[i]=s[i-1]-'0'; } cin >> s; rep(i, 1, n+1){ C[i]=s[i-1]-'0'; } vi dp(n+1); rep(i, 1, n+1){ if(A[i]+B[i]==C[i]) dp[i]=1+dp[i-1]; if(A[i]+B[i]==C[i]+10){ for(int j=i-1; j>=0; --j){ if(A[j]+B[j]+1==C[j]){ dp[i]=dp[j-1]+1; break; }else if(A[j]+B[j]+1!=C[j]+10){ break; } } } } ll res=0; rep(i, 1, n+1) res+=(ll)(dp[i]); cout << res << "\n"; return; } int main() { ios_base::sync_with_stdio(false); cin.tie(0); int t=1; // cin >> t; while(t--) { Solve(); } return 0; } |
English