题目大意
给你一个数组 arr ,该数组表示一个从 1 到 n 的数字排列。有一个长度为 n 的二进制字符串,该字符串上的所有位最初都设置为 0 。
在从 1 到 n 的每个步骤 i 中(假设二进制字符串和 arr 都是从 1 开始索引的情况下),二进制字符串上位于位置 arr[i] 的位将会设为 1 。
给你一个整数 m ,请你找出二进制字符串上存在长度为 m 的一组 1 的最后步骤。一组 1 是一个连续的、由 1 组成的子串,且左右两边不再有可以延伸的 1 。
返回存在长度 恰好 为 m 的 一组 1 的最后步骤。如果不存在这样的步骤,请返回 -1 。
输入:arr = [3,5,1,2,4], m = 1
输出:4
解释:
步骤 1:"00100",由 1 构成的组:["1"]
步骤 2:"00101",由 1 构成的组:["1", "1"]
步骤 3:"10101",由 1 构成的组:["1", "1", "1"]
步骤 4:"11101",由 1 构成的组:["111", "1"]
步骤 5:"11111",由 1 构成的组:["11111"]
存在长度为 1 的一组 1 的最后步骤是步骤 4 。
输入:arr = [3,1,5,4,2], m = 2
输出:-1
解释:
步骤 1:"00100",由 1 构成的组:["1"]
步骤 2:"10100",由 1 构成的组:["1", "1"]
步骤 3:"10101",由 1 构成的组:["1", "1", "1"]
步骤 4:"10111",由 1 构成的组:["1", "111"]
步骤 5:"11111",由 1 构成的组:["11111"]
不管是哪一步骤都无法形成长度为 2 的一组 1 。
解题思路
这道题可以反向考虑,即:给你一个全为1的数组nums,根据arrs从后往前的顺序选择分割点,将数组nums进行分割。若某次分割完成后,被分割的nums数组中有一个子数组的长度为m,则返回结果;
具体看代码注释解释起来方便;
class Solution{
public:
int findLatestStep(vector<int> & arr, int m){
return helper(arr, 0, arr.size() - 1, m, arr.size() - 1);
}
// left和rigth是当前要被分割的子数组的两个端点位置
// nums是提供分割点的数组
// m是数组的分割终止条件
// index是当前nums中选择的分割点的索引
int helper(vector<int> & nums, int left, int right, int m, int index){
// 如果index==-1了,表示所有的分割点都被用完了,仍然没有达到条件,返回-1
// 好像这句永远不会出现,先写了再说
if (index == -1){
return -1;
}
// 如果当前要分割的子数组长度为负了,则表示当前子数组必然无法分割出长度为m的子数组
if (left > right){
return -1;
}
// 同理,当前子数组长度小于m,必然分割不出长度为m的数组
if (right - left + 1 < m){
return -1;
}
// 如果当前子数组长度刚好为m,则返回结果
// 因为我们是从后往前推的,题目的思路是从前往后推的,所以注意这里的返回值
if (right - left + 1 == m){
return index + 1;
}
// 上面条件都不满足,则表明当前子数组长度大于m,可以继续分割
// 本次分割点(题目是从1开始计数的,因此这里需要-1)
int pos = nums[index] - 1;
// 如果当前分割点不在子数组的范围内,则继续用下一个分割点尝试
if (pos > right || pos < left){
return helper(nums, left, right, m, index - 1);
}
// 否则将子数组进行分割,返回左右最大值即可
// 题目要求的最大值,所以用max返回结果,另外能够保证一边符合条件,另一边返回-1的时候,也能返回正确答案
// 另外,如果两边都是-1,返回的结果同样能满足结果
else{
return max(helper(nums, left, pos - 1, m, index - 1), helper(nums, pos + 1, right, m, index - 1));
}
}
};