#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;
#define rep(a,b) for(int a = 0; a < (b); ++a)
#define all(t) t.begin(), t.end()
#define pb push_back
#define fi first
#define se second
#define each(a,x) for(auto &a:x)
#define vi vector<int>
#define vll vector<ll>
#define vull vector<ull>
#define pi pair<int,int>
#define pll pair<ll,ll>
#define nl '\n'
#pragma GCC optimize ("O3")
#pragma GCC optimize ("unroll-loops")
#pragma GCC target("avx2")
#pragma GCC target("bmi")
#pragma GCC target("bmi2")
#pragma GCC target("lzcnt")
#pragma GCC target("popcnt")
const int N = 5e5+1234,INF = 2e9+1234,mod = 1e9+7;
const ll INF_L = (ll)2e18+1234;
void solve()
{
string A,B,C;cin>>A>>B>>C;int n=(int)A.size();ll ans=0;
vector<vector<int>> dp;dp.assign(n+1,{});rep(i,n+1) dp[i].assign(2,0);
for(int i=n-1;i>=0;--i)
{
int a=(int)A[i]-48,b=(int)B[i]-48,c=(int)C[i]-48,d=a+b;
if(d%10==c) dp[i][d>=10]=1;
if(d%10==c) dp[i][d>=10]+=dp[i+1][0];
if((d+1)%10==c) dp[i][(d+1)>=10]+=dp[i+1][1];
ans+=dp[i][0];
}
cout << ans << nl;
}
int main()
{
ios_base::sync_with_stdio(0); cin.tie(0); cout.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 | #include<bits/stdc++.h> using namespace std; typedef long long ll; typedef unsigned long long ull; typedef long double ld; #define rep(a,b) for(int a = 0; a < (b); ++a) #define all(t) t.begin(), t.end() #define pb push_back #define fi first #define se second #define each(a,x) for(auto &a:x) #define vi vector<int> #define vll vector<ll> #define vull vector<ull> #define pi pair<int,int> #define pll pair<ll,ll> #define nl '\n' #pragma GCC optimize ("O3") #pragma GCC optimize ("unroll-loops") #pragma GCC target("avx2") #pragma GCC target("bmi") #pragma GCC target("bmi2") #pragma GCC target("lzcnt") #pragma GCC target("popcnt") const int N = 5e5+1234,INF = 2e9+1234,mod = 1e9+7; const ll INF_L = (ll)2e18+1234; void solve() { string A,B,C;cin>>A>>B>>C;int n=(int)A.size();ll ans=0; vector<vector<int>> dp;dp.assign(n+1,{});rep(i,n+1) dp[i].assign(2,0); for(int i=n-1;i>=0;--i) { int a=(int)A[i]-48,b=(int)B[i]-48,c=(int)C[i]-48,d=a+b; if(d%10==c) dp[i][d>=10]=1; if(d%10==c) dp[i][d>=10]+=dp[i+1][0]; if((d+1)%10==c) dp[i][(d+1)>=10]+=dp[i+1][1]; ans+=dp[i][0]; } cout << ans << nl; } int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int T=1; //cin >> T; while(T--) solve(); return 0; } |
English