2024华为OD机试(C卷+D卷+E卷)最新题库【超值优惠】Java/Python/C++合集
题目描述
按照环保公司要求,小明需要在沙化严重的地区进行植树防沙工作,初步目标是种植一条直线的树带。由于有些区域目前不适合种植树木,所以只能在一些可以种植的点来种植树木。
在树苗有限的情况下,要达到最佳效果,就要尽量散开种植,不同树苗之间的最小间距要尽量大。给你一个适合种植树木的点坐标和一个树苗的数量,请帮小明选择一个最佳的最小种植间距。
例如,适合种植树木的位置分别为1,3,5,6,7,10,13 树苗数量是3,种植位置在1,7,13,树苗之间的间距都是6,均匀分开,就达到了散开种植的目的,最佳的最小种植间距是6
输入描述
第1行表示适合种树的坐标数量
第2行是适合种树的坐标位置
第3行是树苗的数量
例如:
7
1 3 5 6 7 10 13
3
输出描述
最佳的最小种植问距
补充说明
位置范围为1~ 10000000,种植树苗的数量范围2 ~ 10000000,用例确保种植的树苗数量不会超过有效种植坐标数量。
示例1
输入:
7
1 3 5 6 7 10 13
3
输出:
6
题解
这道题属于二分查找与贪心算法相结合的问题,目的是通过二分查找来确定适合种植树木的最大最小间距。
解题思路
- 数据结构选择:
- 采用一个数组
locs
存储可种植的点坐标,并对该数组进行排序,便于后续计算。- 二分查找:
- 间距的范围在1到
(最大坐标 - 最小坐标)
之间。通过二分查找确定能种下所有树苗的最大最小间距。- 在二分查找的过程中,每次计算一个中间值
mid
,然后检查以此间距mid
能否种下所有的树苗。- 贪心判断:
- 对于每个给定的间距
mid
,从最左侧的可种植位置开始,贪心地选择每次尽量离上一个种植点最远的位置进行种植。如果种下的树苗数量达到要求,则返回true
。- 输出:
- 最后输出通过二分查找找到的最大间距。
代码复杂度
- 时间复杂度: O(N * log(D)) ,其中N是坐标数量,D是坐标范围(
locs[n-1] - locs[0]
)。- 空间复杂度: O(N) ,主要用于存储可种植点的数组。
Java
import java.util.Arrays;
import java.util.Scanner;
/**
* @author code5bug
*/
public class Main {
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
// 适合种树的坐标数量
int n = scanner.nextInt();
// 适合种树的坐标位置
int[] locs = new int[n];
for (int i = 0; i < n; i++) {
locs[i] = scanner.nextInt();
}
// 对坐标位置进行排序
Arrays.sort(locs);
// 树苗的数量
int szt = scanner.nextInt();
// 二分查找的边界
int left = 1;
int right = locs[n - 1] - locs[0] + 1;
// 二分查找,找到最大的合适间距
while (left + 1 < right) {
int mid = (left + right) / 2;
if (ok(mid, locs, szt)) {
left = mid; // 可以种下 szt 棵树,尝试更大的间距
} else {
right = mid; // 间距太大,尝试更小的间距
}
}
// 输出最大的合适间距
System.out.println(left);
}
// 判断间距为 x 时能否种下 szt 棵树苗
public static boolean ok(int x, int[] locs, int szt) {
// prev 表示上一棵树的位置,cnt 表示已种的树数量
int prev = locs[0], cnt = 1;
for (int cur : locs) {
if (cur - prev >= x) { // 当前距离符合间距要求则在当前位置种下一棵树
cnt++;
prev = cur;
}
}
return cnt >= szt; // 如果能种下足够数量的树苗则返回 true
}
}
Python
# 适合种树的坐标数量
n = int(input())
# 适合种树的坐标位置
locs = list(map(int, input().split()))
locs.sort()
# 树苗的数量
szt = int(input())
def ok(x:int) -> bool:
''' 种植间距为x 时能否种下 szt 颗树苗'''
global locs, szt
# 上一颗种树的位置,已经种的树数量
prev, cnt = locs[0], 1
for cur in locs:
if cur - prev >= x: # 当前距离符合间距要求则在当前位置种下一棵树
cnt += 1
prev = cur
return cnt >= szt
left, right = 1, locs[-1] - locs[0] + 1
while left + 1 < right:
mid = (left + right) // 2
if ok(mid):
left = mid
else:
right = mid
print(left)
'''
7
1 5 3 6 10 7 13
3
'''
# AC
C++
#include <algorithm>
#include <iostream>
#include <vector>
using namespace std;
// 种植间距为x 时能否种下 szt 颗树苗
bool ok(int x, const vector<int>& locs, int szt)
{
// prev 表示上一棵树的位置,cnt 表示已种的树数量
int prev = locs[0], cnt = 1;
for (int cur : locs) {
if (cur - prev >= x) { // 当前距离符合间距要求则在当前位置种下一棵树
cnt++;
prev = cur;
}
}
return cnt >= szt; // 如果能种下足够数量的树苗则返回 true
}
int main()
{
// 适合种树的坐标数量
int n;
cin >> n;
// 适合种树的坐标位置
vector<int> locs(n);
for (int i = 0; i < n; i++) {
cin >> locs[i];
}
// 对坐标位置进行排序
sort(locs.begin(), locs.end());
// 树苗的数量
int szt;
cin >> szt;
// 二分查找的边界
int left = 1;
int right = locs[n - 1] - locs[0] + 1;
// 二分查找,找到最大的合适间距
while (left + 1 < right) {
int mid = (left + right) / 2;
if (ok(mid, locs, szt)) {
left = mid; // 可以种下 szt 棵树,尝试更大的间距
} else {
right = mid; // 间距太大,尝试更小的间距
}
}
// 输出最大的合适间距
cout << left << endl;
return 0;
}
整理题解不易, 如果有帮助到您,请给点个赞 ❤️ 和收藏 ⭐,让更多的人看到。🙏🙏🙏