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

1. 题目

给你一个数组 arr ，该数组表示一个从 1 到 n 的数字排列。有一个长度为 n 的二进制字符串，该字符串上的所有位最初都设置为 0 。

在从 1 到 n 的每个步骤 i 中（假设二进制字符串和 arr 都是从 1 开始索引的情况下），二进制字符串上位于位置 arr[i] 的位将会设为 1 。

给你一个整数 m ，请你找出二进制字符串上存在长度为 m 的一组 1 的最后步骤。一组 1 是一个连续的、由 1 组成的子串，且左右两边不再有可以延伸的 1 。

返回存在长度 恰好 为 m 的 一组 1 的最后步骤。如果不存在这样的步骤，请返回 -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 。

示例 2：
输入：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 。

示例 3：
输入：arr = [1], m = 1
输出：1

示例 4：
输入：arr = [2,1], m = 2
输出：2
 
提示：
n == arr.length
1 <= n <= 10^5
1 <= arr[i] <= n
arr 中的所有整数 互不相同
1 <= m <= arr.length

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-latest-group-of-size-m
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

class Solution {
public:
    int findLatestStep(vector<int>& arr, int m) {
    	int n = arr.size(), i, ans = -1;
    	vector<vector<int>> pos(n+2, vector<int>(2, -1));//存储该组的左右端点位置
    	vector<int> len(n+1, 0);//长度为x的有多少组
    	int l1,r1,l2,r2,n1,n2,nall,l,r;
    	for(i = 0; i < n; ++i)
    	{
    		l = r = arr[i];
    		nall = 1;
    		n1 = n2 = 0;
    		if(pos[l-1][0] != -1)//左边存在
    		{
    			l1 = pos[l-1][0];
    			r1 = pos[l-1][1];
    			n1 = r1-l1+1;
    		}
    		if(pos[l+1][0] != -1)//右边存在
    		{
    			l2 = pos[l+1][0];
    			r2 = pos[l+1][1];
    			n2 = r2-l2+1;
    		}
    		if(n1)
    		{
    			l = min(l, l1);
    			nall += n1;
    			len[n1]--;
    		}
    		if(n2)
    		{
    			r = max(r, r2);
    			nall += n2;
    			len[n2]--;
    		}
            len[nall]++;
    		pos[l][0] = l;
    		pos[l][1] = r;
    		pos[r][0] = l;
    		pos[r][1] = r;
    		if(len[m])
    			ans = i+1;
    	}
    	return ans;
    }
};

648 ms 127.7 MB

---

我的CSDN博客地址 https://michael.blog.csdn.net/

长按或扫码关注我的公众号（Michael阿明），一起加油、一起学习进步！