PAT 1120. Friend Numbers (20) Java

Two integers are called “friend numbers” if they share the same sum of their digits, and the sum is their “friend ID”. For example, 123 and 51 are friend numbers since 1+2+3 = 5+1 = 6, and 6 is their friend ID. Given some numbers, you are supposed to count the number of different friend ID’s among them. Note: a number is considered a friend of itself.

Input Specification:

Each input file contains one test case. For each case, the first line gives a positive integer N. Then N positive integers are given in the next line, separated by spaces. All the numbers are less than 10⁴.

Output Specification:

For each case, print in the first line the number of different friend ID’s among the given integers. Then in the second line, output the friend ID’s in increasing order. The numbers must be separated by exactly one space and there must be no extra space at the end of the line.

Sample Input:

8
123 899 51 998 27 33 36 12

1 2	8 123 899 51 998 27 33 36 12

Sample Output:

4
3 6 9 26

1 2	4 3 6 9 26

分析：在接收输入数据的时候就把该数字的每一位相加，并把结果插入一个set集合中。因为set是有序的、不重复的，所以set的size值就是输出的个数，set中的每一个数字即所有答案的数字序列

此为Java版，C/C++版本请戳：http://www.liuchuo.net/archives/2901

import java.util.Scanner;
import java.util.Set;
import java.util.TreeSet;

public class Main {
	public static void main(String[] args) {
		Scanner scanner = new Scanner(System.in);
		int numOfInteger = scanner.nextInt();
		Set<Integer> set = new TreeSet<Integer>();
		for (int i = 0; i < numOfInteger; i++) {
			int temp = scanner.nextInt();
			set.add(getFriendId(temp));
		}
		scanner.close();
		System.out.println(set.size());
		boolean isFirst = true;
		for (int id : set) {
			if (!isFirst) {
				System.out.print(" ");
			}
			System.out.print(id);
			isFirst = false;
		}
	}
	
	private static int getFriendId(int num) {
		int sum = 0;
		while(num != 0) {
			sum += num % 10;
			num /= 10;
		}
		return sum;
	}
}

import java.util.Scanner;

import java.util.Set;

import java.util.TreeSet;

public class Main {

public static void main(String[] args) {

Scanner scanner = new Scanner(System.in);

int numOfInteger = scanner.nextInt();

Set<Integer> set = new TreeSet<Integer>();

for (int i = 0; i < numOfInteger; i++) {

int temp = scanner.nextInt();

set.add(getFriendId(temp));

}

scanner.close();

System.out.println(set.size());

boolean isFirst = true;

for (int id : set) {

if (!isFirst) {

System.out.print(" ");

}

System.out.print(id);

isFirst = false;

}

private static int getFriendId(int num) {

int sum = 0;

while(num != 0) {

sum += num % 10;

num /= 10;

}

return sum;

}

❤ 点击这里 -> 订阅《PAT | 蓝桥 | LeetCode学习路径 & 刷题经验》by 柳婼

❤ 点击这里 -> 订阅《从放弃C语言到使用C++刷算法的简明教程》by 柳婼

❤ 点击这里 -> 订阅PAT甲级乙级、蓝桥杯、GPLT天梯赛、LeetCode题解离线版