蓝桥杯 数位排序

问题描述

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

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

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

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

输入格式

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

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

输出格式

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

样例输入

13
5

样例输出

3

样例说明

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 \leq m \leq n \leq 3001≤m≤n≤300 。

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

对于所有评测用例, 1 \leq m \leq n \leq 10^{6}1≤m≤n≤106 。

运行限制

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

总通过次数: 1627  |  总提交次数: 1802  |  通过率: 90.3%

难度: 简单   标签: 2022, 省赛

答案

import os
import sys

n=int(input())
m=int(input())
s=list(range(1,n+1))
s.sort(key=lambda x:sum(int(i) for i in str(x)))
print(s[m-1])

lambda

        lambada简化了函数定义的书写格式。

定义:

        <函数名> = lamba<参数>：<表达式>

冒号后面的表达式的计算结果即为该lambda函数的返回值

等价于：

       def <函数名>(<参数>)：

             <函数题>

             return <返回值>

可传入多个参数

例1：

g=lambda x:x+1
print(g(1))
#结果为2

例2：

g=lambda x,y,z:x+y+z*2
print(g(1,2,3))
#结果为9


#也可直接传递参数
(lambda x:x**2)(3)
#结果为9

Python sorted()函数

描述：

        sorted()函数时全局排序函数，对所有可迭代的对象进行排序操作，它不会修改源对象，而将排序后的结果作为函数的返回值。

sort是应用在list上的方法，sorted可以对所有可迭代的对象进行排序操作。

list的sort方法返回的是对已经存在的列表进行操作，而内建函数sorted方法返回的是一个新的list，而不是在原来的基础上进行的操作。

sorted语法:

sorted(iterable, key=None, reverse=False)

参数说明：

• iteravle - 可迭代对象
• key -主要是用来进行比较的元素，只有一个参数，具体的函数的参数就是取自于可迭代对象中，指定可迭代对象中的一个元素来进行排序。
• reverse -排序规则，reverse = True降序， reverse = False 升序（默认）

返回值：

        返回排序后的对象

例1：

# example1
nums = [2, 3, 0, 1, 5, 4]
sorted(nums)
print(nums)  # 没有改变原数组

输出：

[2, 3, 0, 1, 5, 4]

例2：

# example2
nums = [2, 3, 0, 1, 5, 4]
nums1 = sorted(nums)  # 升序
print(nums1)

nums2 = sorted(nums, reverse=True)  # 降序
print(nums2)

输出：

[0, 1, 2, 3, 4, 5]
[5, 4, 3, 2, 1, 0]

sorted()函数与lambda表达式结合

        lambda表达式是Python中一类特殊的定义函数的形式，使用它可以定义一个匿名函数。与其它语言不同，Python的lambda表达式的函数体只能有单独的一条语句，也就是返回值表达式语句。 

例1：

        使用lambda表达式对一位数组进行倒序排序：

nums = [2, 3, 0, 1, 5, 4]
nums1 = sorted(nums, key=lambda x:x*-1)
print(nums1)

输出：

[5, 4, 3, 2, 1, 0]

 例2：

        按照二维矩阵下标为1的列进行排序

matrix = [[7,0],[4,4],[7,1],[5,0],[6,1],[5,2]]
ret = sorted(matrix, key=lambda x: x[1])
print(ret)

输出：

[[7, 0], [5, 0], [7, 1], [6, 1], [5, 2], [4, 4]]

例3：

        对字典数组的某一关键字进行排序 

# 以 age 的降序进行排序
array = [{"age": 20, "name": "a"}, {"age": 25, "name": "b"}, {"age": 10, "name": "c"}]
array = sorted(array, key=lambda x:x["age"], reverse=True)
# 上一句等同于
# array = sorted(array, key=lambda x:-x["age"])
print(array)

输出：

[{‘age’: 25, ‘name’: ‘b’}, {‘age’: 20, ‘name’: ‘a’}, {‘age’: 10, ‘name’: ‘c’}]

例4：

        先按照成绩降序排序，相同成绩的按照名字升序排序 。

d1 = [{'name': 'alice', 'score': 38}, {'name': 'bob', 'score': 18}, {'name': 'darl', 'score': 28}, {'name': 'christ', 'score': 28}]
l = sorted(d1, key=lambda x:(-x['score'], x['name']))
print(l)

输出：

[{‘name’: ‘alice’, ‘score’: 38}, {‘name’: ‘christ’, ‘score’: 28}, {‘name’: ‘darl’, ‘score’: 28}, {‘name’: ‘bob’, ‘score’: 18}]