PTA乙级1064

1064

如果两个整数各位数字的和是一样的，则被称为是“朋友数”，而那个公共的和就是它们的“朋友证号”。例如 123 和 51 就是朋友数，因为 1+2+3 = 5+1 = 6，而 6 就是它们的朋友证号。给定一些整数，要求你统计一下它们中有多少个不同的朋友证号。

输入格式：

输入第一行给出正整数 N。随后一行给出 N 个正整数，数字间以空格分隔。题目保证所有数字小于 10
​4
​​ 。

输出格式：

首先第一行输出给定数字中不同的朋友证号的个数；随后一行按递增顺序输出这些朋友证号，数字间隔一个空格，且行末不得有多余空格。

输入样例：

8
123 899 51 998 27 33 36 12

输出样例：

4
3 6 9 26

思路:这题一开始我的想法是将存好的数字依次再分割存入新数组中遍历求和然后再放入新数组,再对新数组用sort排序,然后找出相同数字并删除,删除的方法想了好多,都挺烦的,然后就找了大佬的写法,发现他们用了TreeSet集合,目前是第一次接触TreeSet,初步了解里面的元素是有序且不重复的,所以这个集合很适合这题,最后用foreach遍历之后还要获取最后一个元素单独输出.

代码:
这是一开始的写法

package PTA2;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;

public class PTA1064 {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());
        String[] m = br.readLine().split(" ");//存放输入的数字

        int[] p=new int[N];
        for (int i=0;i<N;i++){
            String[] t=m[i].split("");
            for (String s : t) {
                p[i]+= Integer.parseInt(s);
            }
        }
        Arrays.sort(p);
        Object[] r = remove(p);
        for(int i=0;i<r.length-1;i++){
            System.out.print(r[i]+" ");
        }
        System.out.println(r[r.length-1]);
    }
    private static Object[] remove(int[] arr){
        //创建一个集合
        List<Object> list = new ArrayList<>();
        //遍历数组往集合里存元素
        for (Object o : arr) {
            //如果集合里面没有相同的元素才往里存
            if (!list.contains(o)) {
                list.add(o);
            }
        }
        //toArray()方法会返回一个包含集合所有元素的Object类型数组
        return list.toArray();
    }

}

这是后来使用TreeSet的写法

package PTA2;

import test1.S;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;

public class PTA1064 {
    public static void main(String[] args) throws IOException {
        BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
        int N=Integer.parseInt(br.readLine());
        String [] m=br.readLine().split(" ");//存放输入的数字
        TreeSet<Integer> p= new TreeSet<>();//存放朋友证号
        for (int i=0;i<N;i++){
            int d=0;
            String[] t=m[i].split("");
            for (String s : t) {
                d+= Integer.parseInt(s);
            }
            p.add(d);
        }
        System.out.println(p.size());
        int a= p.pollLast();
        for(int i:p){
            System.out.print(i+" ");
        }
        System.out.print(a);

    }

}