LeetCode 768. 最多能完成排序的块 II(swift)

class Solution768 {
    func maxChunksToSorted(_ arr: [Int]) -> Int {
        // 定义一个变量 ans，用于记录最终的答案
        var ans = 0
        var countA = [Int: Int]()
        var countB = [Int: Int]()
        
        // 遍历 arr 和 arr.sorted()，将它们的每个元素合并成一个元组 (a, b)
        for (a, b) in zip(arr, arr.sorted()) {
            // 在 countA 中增加 a 出现的次数
            countA[a, default: 0] += 1
            // 在 countB 中增加 b 出现的次数
            countB[b, default: 0] += 1
            // 如果 countA 和 countB 相等，那么说明 arr 的前 i 个元素可以分成一个 chunk，并将 ans 加 1
            print(countA,countB)
            if countA == countB {
                ans += 1
            }
        }

        // 返回最终的答案
        return ans
    }
    func test() {
//        var arr = [5,4,3,2,1]
        var arr = [2,1,3,4,4]
        let ret = maxChunksToSorted(arr);
        print(ret);
    }
    
}