PAT 1016 部分A+B（15）（C++&JAVA&&Python）

1016 部分A+B（15 分）

正整数 A 的“D​A​​（为 1 位整数）部分”定义为由 A 中所有 D​A​​ 组成的新整数 P​A​​。例如：给定 A=3862767，D​A​​=6，则 A 的“6 部分”P​A​​ 是 66，因为 A 中有 2 个 6。

现给定 A、D​A​​、B、D​B​​，请编写程序计算 P​A​​+P​B​​。

输入格式：

输入在一行中依次给出 A、D​A​​、B、D​B​​，中间以空格分隔，其中 0<A,B<10​10​​。

输出格式：

在一行中输出 P​A​​+P​B​​ 的值。

输入样例 1：

3862767 6 13530293 3

输出样例 1：

399

输入样例 2：

3862767 1 13530293 8

输出样例 2：

0

C++ 

#include<iostream>
#include<string>
using namespace std;
int  trans(string A, string a) {
	long count = 0;
	string result;
	for (long i = 0; i < A.length(); i++) 
		if (a[0] == A[i])
			count++;
	while (count--)
		result += a;
	return atoi(result.c_str());
}
int main() {
	long long  result;    //注意范围
	string A, B;   
	string ac, bc;
	cin >> A >> ac >> B >> bc;
	cout << trans(A, ac) + trans(B, bc);
	return 0;
}

JAVA:

import java.util.Scanner;
public class Main{
	public static int trans(String A,String a){
		long count=0;
		String result=new String();
		for(int i=0;i<A.length();i++)
			if(A.charAt(i)==a.charAt(0))
				count++;
		if(count==0)
			result="0";
		while(count--!=0)
			result+=a;
		return Integer.parseInt(result.trim());
	}
	public static void main(String [] args){
		Scanner input=new Scanner(System.in);
		String A,B,a,b;
		A=input.next();
		a=input.next();
		B=input.next();
		b=input.next();
		System.out.println(trans(A,a)+trans(B,b));
	}
}

Python：

if __name__=="__main__":
    a,ac,b,bc=map(str,input().split())
    A=int(ac*a.count(ac) if(ac*a.count(ac)) else '0')          #注意不能由空字符串转为int
    B=int(bc*b.count(bc) if(bc*b.count(bc)) else '0')
    print(A+B)