PAT-乙级B1016-7个6的两种写法

#include <stdio.h>
#include <math.h>
int main()
{
    long long a, da, pa=0, b, db, pb=0;
    int count=0;
    scanf("%lld %lld %lld %lld", &a, &da, &b, &db);
    while(a!=0) {
        if (a%10==da){               //取末位数
            pa+=da*pow(10,count++);  //666的一种写法
        }
        a=a/10;                      //去末位数
    }
    count=0;
    while(b!=0) {
        if (b%10==db){
            pb=pb*10+db;             //666的另一种写法
        }
        b=b/10;
    }
    printf("%lld", pa+pb);

}

//while重复了两次，可以考虑写成函数

#include <stdio.h>
#include <math.h>
long long sum(long long n,long long m) {
    long long p=0;
    int count=0;
    while(n!=0) {
        if (n%10==m){
            p+=m*pow(10,count++);       //666的一种写法,另一种写法是p=p*10+m
        }
        n=n/10;
    }
    return p;
}                                //写成函数
int main()
{
    long long a, da, pa, b, db, pb;
    int count=0;
    scanf("%lld %lld %lld %lld", &a, &da, &b, &db);
    pa=sum(a,da);
    pb=sum(b,db);
    printf("%lld", pa+pb);

}