leetcode1562.查找大小为M的最新分组

题目大意

给你一个数组 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));
        }

    }
};