华为OD机试- 最佳植树距离

题目描述

小明在直线的公路上种树，现在给定可以种树的坑位的数量和位置，以及需要种多少棵树苗，问树苗之间的最小间距是多少时，可以保证种的最均匀（两棵树苗之间的最小间距最大）

输入描述

输入三行：
第一行一个整数：坑位的数量
第二行以空格分隔的数组：坑位的位置
第三行一个整数：需要种植树苗的数量

输出描述

树苗之间的最小间距

示例1：

输入：
7
1 3 6 7 8 11 13
3

输出：
6

说明：
三颗树苗分别种在1、7、13的位置，可以保证种的最均匀，树苗之间的最小间距为6。

解题思路

代码实现

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[] positions = new int[n];
    for (int i = 0; i < n; i++) {
      positions[i] = sc.nextInt();
    }
 
    int m = sc.nextInt();
 
    System.out.println(getResult(n, positions, m));
  }
 
  public static int getResult(int n, int[] positions, int m) {
    Arrays.sort(positions);
 
    int min = 1, max = positions[n - 1] - positions[0];
    int ans = 0;
 
    while (min <= max) {
      int mid = (min + max) >> 1;
 
      if (check(positions, m, mid)) {
        ans = mid;
        min = mid + 1;
      } else {
        max = mid - 1;
      }
    }
 
    return ans;
  }
 
  public static boolean check(int[] positions, int m, int minDis) {
    int count = 1;
 
    int curPos = positions[0];
    for (int i = 1; i < positions.length; i++) {
      if (positions[i] - curPos >= minDis) {
        count++;
        curPos = positions[i];
      }
    }
 
    return count >= m;
  }
}