> 文章列表 > 蓝桥杯-数位排序

蓝桥杯-数位排序

蓝桥杯-数位排序

蓝桥杯-数位排序

  • 1、问题描述
  • 2、解题思路
  • 3、代码实现

1、问题描述

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

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

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

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

输入格式

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

  第二行包含一个正整数 m

输出格式

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

样例输入

13
5

样例说明

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

评测用例规模与约定

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

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

  • 对于所有评测用例, 1≤m≤n≤1061≤mn10610^6106

运行限制

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

2、解题思路

  我们可以用Map的key存真实的数字,用value存每个数字的位数之和,最后再根据Map的value进行排序返回一个新的Map或者List就行。

  注意:如果最后用List收集结果,那我们收集的是Map中排好序之后的key,此时的value对我们已经没用了,所以不要也行,直接用List比较方便,这里用LinkedHashMap去收集也是可以的。

3、代码实现

  写一个计算位数之和的函数

//返回位数和public static Integer sum(int number){int [] nums=new int[String.valueOf(number).length()];int temp=number;for (int i = 0; i < nums.length; i++) {nums[i]=temp%10;temp=(temp-nums[i])/10;}return Arrays.stream(nums).sum();}

   就是计算一个数字的个位、十位、百位等等,取模之后减去取模结果再除10就行(十进制)。

public static void main(String[] args) {Scanner scan = new Scanner(System.in);int n=scan.nextInt();int m=scan.nextInt();HashMap<Integer, Integer> map = new HashMap<>();for (int i = 1; i <= n; i++) {//value存位数之和map.put(i,sum(i));}//将排序之后的key用List收集,此时value对我们已经没用了。List<Integer> collect = map.entrySet().stream().sorted((o1, o2) -> o1.getValue().compareTo(o2.getValue())).map(Map.Entry::getKey).collect(Collectors.toList());System.out.println("排序之后的key:"+collect);    System.out.println(collect.get(m - 1));}

  运行结果如下:

image-20230226184828447

  这里用Map去收集结果也是可以的,用LinkedHashMap收集结果的代码如下所示:

HashMap<Integer, Integer> collect = map.entrySet().stream().sorted((o1, o2) -> o1.getValue().compareTo(o2.getValue())).collect(Collectors.toMap(Map.Entry::getKey,Map.Entry::getValue,(oldValue, newValue) -> newValue,LinkedHashMap::new));

  遍历一下key和value

collect.entrySet().forEach(System.out::println);

  运行结果如下:

image-20230226190332502

   这样确实也可以实现,不过已经都可以用List的get(int index)方法直接取出结果了,我觉得没必要用Map了,还多一步遍历。