Leetcode 3267. Count Almost Equal Pairs II

• Leetcode 3267. Count Almost Equal Pairs II
  • 1. 解题思路
  • 2. 代码实现

• 题目链接：3267. Count Almost Equal Pairs II

1. 解题思路

这一题同样是题目3265. Count Almost Equal Pairs I的进阶版本。

它主要的区别在于说：

1. 最大的操作次数增加到两次；
2. 数组长度增加到最大5000

然后，首先对于操作次数的追加，我们要判断两个数是否是almost equal，就要满足如下几个要求：

1. 所有出现的数字必须相同
2. 位与位之间不同的位数最多不能超过4位
3. 如果不同的位数恰好为4位，那么其必定是两个pair才行。

此时，我们就能完成almost equal的判断，剩下的就是如何对计算复杂度进行优化，这里的话我主要是进行了一下分组，由于两个数要满足almost equal那么他出现的所有位上的数必然是完全相同的，因此我们据此进行一下分组，然后只要对每一个组内的数字进行一下考察即可。

其次，在每一个组内，可能会有同一个数多次出现的case，我们也可以对这部分进行一个二次分组，这样也可以优化一些计算复杂度。

此时，我们得到的算法就可以通过全部的的测试样例了，虽然还是蛮暴力的，不过能work就行……

2. 代码实现

给出python代码实现如下：

class Solution:
    def countPairs(self, nums: List[int]) -> int:
        nums = [str(x).rjust(7, "0") for x in nums]
        cnt = defaultdict(lambda : defaultdict(int))
        for x in nums:
            sx = "".join(sorted(x))
            cnt[sx][x] += 1
            
        def is_almost_equal(x, y):
            diff = []
            for ch1, ch2 in zip(x, y):
                if ch1 != ch2:
                    diff.append((ch1, ch2))
            if len(diff) <= 3:
                return True
            if len(diff) == 4:
                return (diff[0][1], diff[0][0]) in diff
            return False
        
        ans = 0
        for group in cnt.values():
            items = list(group.items())
            n = len(items)
            for i in range(n):
                ans += items[i][1] * (items[i][1]-1) // 2
                for j in range(i+1, n):
                    if is_almost_equal(items[i][0], items[j][0]):
                        ans += items[i][1] * items[j][1]
        return ans

提交代码评测得到：耗时7247ms，占用内存17.7MB。