LeetCode 2560. 打家劫舍 IV

2560. 打家劫舍 IV
      提示
      中等
      177
      相关企业
      沿街有一排连续的房屋。每间房屋内都藏有一定的现金。现在有一位小偷计划从这些房屋中窃取现金。

由于相邻的房屋装有相互连通的防盗系统，所以小偷 不会窃取相邻的房屋 。

小偷的 窃取能力 定义为他在窃取过程中能从单间房屋中窃取的 最大金额 。

给你一个整数数组 nums 表示每间房屋存放的现金金额。形式上，从左起第 i 间房屋中放有 nums[i] 美元。

另给你一个整数 k ，表示窃贼将会窃取的 最少 房屋数。小偷总能窃取至少 k 间房屋。

返回小偷的 最小 窃取能力。

示例 1：

输入：nums = [2,3,5,9], k = 2
输出：5
解释：
小偷窃取至少 2 间房屋，共有 3 种方式：

• 窃取下标 0 和 2 处的房屋，窃取能力为 max(nums[0], nums[2]) = 5 。
• 窃取下标 0 和 3 处的房屋，窃取能力为 max(nums[0], nums[3]) = 9 。
• 窃取下标 1 和 3 处的房屋，窃取能力为 max(nums[1], nums[3]) = 9 。
  因此，返回 min(5, 9, 9) = 5 。
  示例 2：

输入：nums = [2,7,9,3,1], k = 2
输出：2
解释：共有 7 种窃取方式。窃取能力最小的情况所对应的方式是窃取下标 0 和 4 处的房屋。返回 max(nums[0], nums[4]) = 2 。

提示：

1 <= nums.length <= 105
1 <= nums[i] <= 109
1 <= k <= (nums.length + 1)/2

解题思路：
准备工作：先copy一份数组，然后对copy得到的数据进行排序；声明一个len + 1长度的boolean类型的数组(len长度会报错，但可修改)；
解题思路：最理想情况为结果为数组从小到大排序后的第k - 1位置的数据，但是题中限制不能有相邻的，所以需要通过声明的boolean类型数组判断最理想的取值情况下可获取的数据量与目标k数据的差距，根据差距将结果向前移动数据量与k的差值个单位，每一次移动后都需要更新boolean数组的数据，然后重新判断两者差值，直到得出结果为止。

java代码：

class Solution {
    public int minCapability(int[] nums, int k) {
        if (k == 1) return Arrays.stream(nums).min().getAsInt();
        int ans = -1;
        int[] nums_flag = nums.clone();
        Arrays.sort(nums_flag);  //排序后
        int len = nums.length;   //数组长度
        boolean[] flag_compare = new boolean[len + 1];
        int index_k = k - 1;   //记录下标
        int flag_nums = nums_flag[index_k];  // 存储无条件下的最小返回值
        int index = -1;  // boolean数组下标
        for (int i : nums) flag_compare[++index] = i <= flag_nums ? true : false;
        int difference;   //记录连续出现多少个true
        int index_begin;  //记录每一个false,true组合的true位置
        int flag;  //记录小于指定值的可偷取房屋书数量
        while (true) {   // 此时偷取房间数不足，需要继续寻找
            flag = 0;
            index = -1;
            for (int i = 0; i < len; i++) {
                index_begin = i;
                while (flag_compare[i] && i < len) i++;
                difference = i - index_begin;
                flag += difference % 2 == 0 ? difference / 2 : (difference + 1) / 2;
            }
            if (flag < k) {
                index_k += k - flag;
                if (index_k < len) {
                    flag_nums = nums_flag[index_k];
                } else {
                    return ans;
                }
                for (int i : nums) flag_compare[++index] = i <= flag_nums ? true : false;
            } else {
                ans = flag_nums;
                break;
            }
        }
        return ans;
    }
}

此方法时间复杂度为：k * 2 * num.length,即while死循环最多循环k次，循环体内存在两个时间复杂度为num.length的for循环