蓝桥杯——数位排序（JAVA）

题目：

小蓝对一个数的数位之和很感兴趣, 今天他要按照数位之和给数排序。当 两个数各个数位之和不同时, 将数位和较小的排在前面, 当数位之和相等时, 将数值小的排在前面。

例如, 2022 排在 409 前面, 因为 2022 的数位之和是 6, 小于 409 的数位 之和 13 。

又如, 6 排在 2022 前面, 因为它们的数位之和相同, 而 6 小于 2022 。

给定正整数n,m, 请问对 1 到 n 采用这种方法排序时, 排在第 m 个的元 素是多少?

输入格式

输入第一行包含一个正整数 n 。

第二行包含一个正整数 m 。

输出格式

输出一行包含一个整数, 表示答案。

样例输入">样例输入

13
5

样例输出

3

样例说明

1 到 13 的排序为: 1,10,2,11,3,12,4,13,5,6,7,8,9 。第 5 个数为 3 。

评测用例规模与约定

对于30% 的评测用例,1≤m≤n≤300 。

对于50% 的评测用例, 1≤m≤n≤1000 。

对于所有评测用例, 1≤m≤n≤106 。

运行限制

• 最大运行时间：3s
• 最大运行内存: 512M

代码： 

import java.util.Arrays;
import java.util.Scanner;

public class Main{
    public static void main(String[] args) {
        Scanner sc=new Scanner(System.in);
        int n,m;                         
        n=sc.nextInt();
        m=sc.nextInt();
        Integer[] arr=new Integer[n];  //必须用int的包装类Integer
        for (int i=0;i<n;i++){
            arr[i]=i+1;
        }
        Arrays.sort(arr,((o1, o2) -> js(o1)!=js(o2)?js(o1)-js(o2):o1-o2)); //Lambda表达式用sort方法进行排序
        System.out.println(arr[m-1]);
        sc.close();

    }
    public static int js(Integer a){    //求数位和
        int sum=0;
        while(a>0){
            sum+=a%10;
            a=a/10;
        }
        return sum;
    }
}

思路：本题用sort方法中的

• • sort(T[] a, Comparator<? super T> c) 根据指定的比较器引发的顺序对指定的对象数组进行排序

 此方法可以根据设置的条件交换元素的顺序，

Arrays.sort(arr,((o1, o2) -> js(o1)!=js(o2)?js(o1)-js(o2):o1-o2));

这段代码采用了Lambda表达式意思是比较前后两个数的数位和如果不同，则将数位和较小的放到前面，如果数位和相同则将较小的数字放到前面。