PAT乙级 1064 朋友数 (20 分) Java 实现

注意，这道题即使只出现一次也算是好友数。

import java.io.*;

public class pat1064 {

	public static void main(String[] args) throws IOException{
		// TODO Auto-generated method stub
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int N = Integer.parseInt(br.readLine());
		int[] friend = new int[37];		//因为数字小于10^4，所以最大的朋友数为9*4=36；
		String[] arr = br.readLine().trim().split(" ");
		for(int i=0;i<N;i++) {
			int sum = 0;
			for(int j=0;j<arr[i].length();j++) {
				int n = arr[i].charAt(j) - '0';
				sum += n;
			}
			friend[sum]++;					//相应的位置++
		}
		int[] output = new int[N];		//创建输出数组
		int friends = 0;
		for(int i=0;i<37;i++) {
			if(friend[i]>0) {
				output[friends] = i;
				friends++;
			}
		}
		System.out.println(friends);
		for(int i=0;i<friends-1;i++) System.out.print(output[i] + " ");
		System.out.println(output[friends-1]);
	}

}