【PAT B】1016 部分A+B

#include <stdio.h>

long long lsolve(long long tmp,long long d){
    long long res = 0;
    while(tmp){
        if(tmp%10==d){
            res = res*10+d;
        }
        tmp/=10;
    }
    return res;
}                                                                                                            

int main(){
    long long a,da,b,db;
    scanf("%lld %lld %lld %lld",&a,&da,&b,&db);
    long long pa=0,pb=0;
    pa = lsolve(a,da);
    pb = lsolve(b,db);
    printf("%lld\n",pa+pb);
    return 0;
}

#include <stdio.h>
                          
int main(){
    long long a,da,b,db;
    long long pa=0,pb=0;
    scanf("%lld %lld %lld %lld",&a,&da,&b,&db);
    while(a){
        if(a%10==da){
            pa = pa*10 + da;
        }
        a/=10;
    }
    while(b){
        if(b%10==db){
            pb = pb*10 + db;
        }
        b/=10;
    }
    
    printf("%lld",pa+pb);
    return 0;
}