leetcode1562. 查找大小为 M 的最新分组（Python3、c++）

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

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

---

方法：模拟维护区间

思路：

题意是按照arr的顺序不断将对应的0变成1，我们关注的是连续1的区间的长度。因此可以使用并查集来解决，相邻的两个点都是1，那么这两个点就可以合并，同时修改并查集定义，维护每个连通区间的长度，找到最后一次有长度为m的连通区间的步骤编号即可。

由于本题这些点是一位的，连通分量一定在一维上连续，因此我们可以简化，不使用并查集。我们维护两个数组left、right。

• left[i]表示以结点i为连通分量右端点，它对应的左端点
• right[i]表示以结点i为连通分量左端点，它对应的右端点

我们这种结构类似于双向链表，即对一个连通分量（即全是1的连通区间），左端点的right为右端点，右端点的left为左端点。因左端点的left无意义（因为左端点的左边肯定为0，因此以它为右端点的连通区间就是它自己，因此它的left设置为自己），同理右端点的right也设置为自己。

两个数组初始值都为0，长度应该为n+2（在下面介绍）。

这种情况下，我们如何计算连通区间的长度呢，很简单，我们输入连通区间的左端点坐标x，那么长度即为right[x]-x+1。我们用一个函数getlength来表示这个式子。

介绍完如何表示连通区间之后，我们介绍本题的总体思路，使用res来保存答案，初始为-1。使用变量count来保存当前长度为m的连通分量个数，刚开始肯定是0。

• 开始按照arr顺序进行遍历，根据arr[i]的不同，即将0变为1的位置不同，更新left和right变量（后面详细解释）。
• 对涉及到的，更新的连通区间进行求长度，如果原来是m，那么count–，如果新变成的区间是m，那么count++。
• 更新结束后，如果此时count>0，更新res为i+1（这是因为i从0开始，而答案返回从第1步开始，因此要+1），即该步结束后，存在长度为m的连通区间。
• 完成arr的遍历，返回res即可。如果前面遍历过程中，没有一次出现m，那么res为初始值-1；如果有，那么res为最后一次有长度为m的区间的步骤序号。

下面我们来详细讲解区间更新的步骤，可以知道，假设x=arr[i]，那么更新的时候，x的左右可能出现四种情况：

• 第一种是x的左右两边都是1，即左右两边都有连通区间了，那么最后要将这两个区间和x合并成一个区间。
• 第二种是左边为0，右边是1，那么只有右边有连通区间，那么更新之后需要将右边区间扩大一位
• 第三种是右边为0，左边是1，那么只有左边有连通区间，更新之后将左边区间扩大一位
• 最后一种是左右都为0，那么就只有自己，更新之后生成长度为1的新连通区间。

上面一共四种情况我们需要考虑，首先需要说明的是，我们不需要使用一个数组来维护当前1-n的排列如何，因为如果某个位置已经是1了，根据我们上面的介绍，它已经存在于一个连通区间中，因此它的left和right肯定是不为0的，因此我们可以直接用left和right来判断。

下面说明如何判断x左边和右边是不是1，首先我们知道，对于x=arr[i]，x点此时肯定是0，

• 那么如果x-1是1，那么它肯定是某个区间的右端点，因此我们判断left[x-1]是否不为0，如果不为0，那么即存在左区间。且将left[x-1]送入getlength函数即可得到左区间长度。
• 对于有区间，如果x+1是1，那么它肯定是某个区间的左端点，因此我们判断right[x+1]是否不为0，如果不为0，那么即存在右区间，且将x+1送入getlength即可得到右区间长度。
• 由上面可以看出，我们考虑left[x-1]和right[x+1]，x的取值为1-n，下标的取值为0-n+1，这就是为什么我们的两个数组长度应该为n+2。

下面我们考虑四种情况如何更新left和right。

第一种情况的图解如下图所示（其中有一些没用的左右连线就忽略了，因为我们只关心left[x-1]和right[x+1])：

**我们需要将左区间的左端点的right变为右区间的右端点，右区间的右端点的left变为左区间的左端点，中间点的这些不需要考虑，因为中间都是1了，继续遍历arr的过程中，也不会遍历到这些点了。**因此代码为：

				right[left[x-1]] = right[x+1]
                left[right[x+1]] = left[x-1]

然后判断新的区间长度是不是m，更新count。

同理，对于第二种和第三种情况，我们只需要考虑其中一边的更新，因为另一侧的端点为x，直接给出更新的代码：

			# 只有左边有区间
                # 将左边区间向右扩展一位，更新left和right
                right[