小明今年升学到了小学一年级,来到新班级后,发现其他小朋友身高参差不齐,然后就想基于各小朋友和自己的身高差,对他们进行排序,请帮他实现排序。
第一行为正整数H
和N
,其中0 < H < 200
为小明的身高,0 < N < 50
为新班级其他小朋友个数。
第二行为 N 个正整数H1 ~ Hn
分别是其他小朋友的身高,取值范围0 < Hi < 200
,且N
个正整数各不相同。
输出排序结果,各正整数以空格分割,和小明身高差绝对值最小的小朋友排在前面,和小明身高差绝对值最大的小朋友排在后面,如果两个小朋友和小明身高差一样,则个子较小的小朋友排在前面。
100 10
95 96 97 98 99 101 102 103 104 105
99 101 98 102 97 103 96 104 95 105
直接使用lambda
匿名函数进行排序,排序依据为:
h
和小明的身高x
的差的绝对值lst.sort(key = lambda h: (abs(x-h), h))
注意, 不可以将排序的第二个依据隐去,写成形如
lst.sort(key = lambda h: abs(x-h))
这是因为在第一个依据同时满足的情况下,如果没有指定第二个依据,与x
差值绝对值相等的身高h
会按照在原数组中的顺序进行排序,这样是不符合题目要求的。
# 题目:2023B-身高排序
# 分值:100
# 作者:闭着眼睛学数理化
# 算法:直接调用排序API
# 代码看不懂的地方,请直接在群上提问
x, n = map(int, input().split())
lst = list(map(int, input().split()))
lst.sort(key = lambda h: (abs(x-h), h))
print(" ".join(map(str, lst)))
import java.util.*;
public class Main {
public static void main(String[] args) {
Scanner input = new Scanner(System.in);
int x = input.nextInt();
int n = input.nextInt();
List<Integer> lst = new ArrayList<>();
for (int i = 0; i < n; i++) {
lst.add(input.nextInt());
}
Collections.sort(lst, new Comparator<Integer>() {
@Override
public int compare(Integer h1, Integer h2) {
int diff1 = Math.abs(x - h1);
int diff2 = Math.abs(x - h2);
if (diff1 == diff2) {
return h1.compareTo(h2);
}
return Integer.compare(diff1, diff2);
}
});
for (int i = 0; i < n; i++) {
System.out.print(lst.get(i) + " ");
}
}
}
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
bool compare(int h1, int h2, int x) {
int diff1 = abs(x - h1);
int diff2 = abs(x - h2);
if (diff1 == diff2) {
return h1 < h2;
}
return diff1 < diff2;
}
int main() {
int x, n;
cin >> x >> n;
vector<int> lst(n);
for (int i = 0; i < n; i++) {
cin >> lst[i];
}
sort(lst.begin(), lst.end(), [x](int h1, int h2) {
return compare(h1, h2, x);
});
for (int i = 0; i < n; i++) {
cout << lst[i] << " ";
}
return 0;
}
时间复杂度:O(NlogN)
。排序时间复杂度。
空间复杂度:O(1)
。
华为OD算法/大厂面试高频题算法冲刺训练目前开始常态化报名!目前已服务100+同学成功上岸!
课程讲师为全网50w+粉丝编程博主@吴师兄学算法 以及小红书头部编程博主@闭着眼睛学数理化
每期人数维持在20人内,保证能够最大限度地满足到每一个同学的需求,达到和1v1同样的学习效果!
60+天陪伴式学习,40+直播课时,300+动画图解视频,300+LeetCode经典题,200+华为OD真题/大厂真题,还有简历修改、模拟面试、专属HR对接将为你解锁
可上全网独家的欧弟OJ系统练习华子OD、大厂真题
可查看链接 大厂真题汇总 & OD真题汇总(持续更新)
绿色聊天软件戳 od1336
了解更多