LeetCode: 769.最多能完成排序的块

数组arr是[0, 1, …, arr.length - 1]的一种排列，我们将这个数组分割成几个“块”，并将这些块分别进行排序。之后再连接起来，使得连接的结果和按升序排序后的原数组相同。

我们最多能将数组分成多少块？

示例 1:

输入: arr = [4,3,2,1,0]
输出: 1
解释:
将数组分成2块或者更多块，都无法得到所需的结果。
例如，分成 [4, 3], [2, 1, 0] 的结果是 [3, 4, 0, 1, 2]，这不是有序的数组。
示例 2:

输入: arr = [1,0,2,3,4]
输出: 4
解释:
我们可以把它分成两块，例如 [1, 0], [2, 3, 4]。
然而，分成 [1, 0], [2], [3], [4] 可以得到最多的块数。
注意:

arr 的长度在 [1, 10] 之间。
arr[i]是 [0, 1, …, arr.length - 1]的一种排列。

链接：https://leetcode.com/problems/max-chunks-to-make-sorted/
解题思路：
如果前 k +1个元素为 [0, 1, …, k]，只需要检查其中最大的数是不是 k 就可以了。

我的代码：

class Solution(object):
    def maxChunksToSorted(self, arr):
        """
        :type arr: List[int]
        :rtype: int
        """
        # res = 0
        # i = 0
        # l = len(arr) 
        # cur = 0
        # while i < l:
        #     if arr[i] != i:
        #         cur = arr[i]
        #         while i <= cur:
        #             if arr[i] > cur:
        #                 cur = arr[i]
        #             i += 1
        #     else:
        #         i += 1
        #     res += 1
        # return res
        
        # 优化
        cur_max = res = 0
        for i, a in enumerate(arr):
            cur_max = max(cur_max, a)
            if cur_max == i:
                res += 1
        return res

Time： O(N)
Space: O(1)
难度： 中等