蓝桥杯2022年第十三届省赛真题-数位排序

题目描述

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

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

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

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

输入格式

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

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

输出格式

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

样例输入

复制

13
5

样例输出

复制

3

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 = sc.nextInt();
        int m = sc.nextInt();
        Integer[] arr = new Integer[n];
        for (int i = 1; i <= n; i++) {
            arr[i - 1] = i;
        }
        Arrays.sort(arr, (o1, o2) -> {
            int num1 = 0, num2 = 0;
            int a1=o1,a2=o2;
            while (a1 > 0) {
                num1 += (a1 % 10);
                a1 /= 10;
            }
            while (a2 > 0) {
                num2 += (a2 % 10);
                a2 /= 10;
            }
            if (num1 == num2) {
                return o1 - o2;
            } else if (num1 > num2) {
                if (o1 > o2) return o1 - o2;
                else return o2 - o1;
            } else {
                if (o1 > o2) return o2 - o1;
                else return o1 - o2;
            }
        });
        System.out.println(arr[m - 1]);
    }
}