leetcode1007. 行相等的最少多米诺旋转

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/sinat_36811967/article/details/88555971

在一排多米诺骨牌中,A[i] 和 B[i] 分别代表第 i 个多米诺骨牌的上半部分和下半部分。(一个多米诺是两个从 1 到 6 的数字同列平铺形成的 —— 该平铺的每一半上都有一个数字。)
我们可以旋转第 i 张多米诺,使得 A[i] 和 B[i] 的值交换。
返回能使 A 中所有值或者 B 中所有值都相同的最小旋转次数。
如果无法做到,返回 -1.

示例 1:
输入:A = [2,1,2,4,2,2], B = [5,2,6,2,3,2]
输出:2
解释:
图一表示:在我们旋转之前, A 和 B 给出的多米诺牌。
如果我们旋转第二个和第四个多米诺骨牌,我们可以使上面一行中的每个值都等于 2,如图二所示。
示例 2:
输入:A = [3,5,1,2,3], B = [3,6,3,3,4]
输出:-1
解释:
在这种情况下,不可能旋转多米诺牌使一行的值相等。
提示:
1 <= A[i], B[i] <= 6
2 <= A.length == B.length <= 20000

先找candi,再判断次数:

class Solution:
    def minDominoRotations(self, A: List[int], B: List[int]) -> int:
        all, candi = A+B, []
        for i in range(1, 7):
            if all.count(i) >= len(A):
                candi.append(i)
        if not candi:
            return -1
        if len(candi) == 2:
            a, b = candi
            res = [0, 0]  # 上a下b,上b下a
            for i in range(len(A)):
                if A[i] == B[i]:
                    return -1
                if A[i] == a:
                    res[1] += 1
                else:
                    res[0] += 1
        else:
            candi, res = candi[0], [0, 0]
            for i in range(len(A)):
                if A[i] != candi and B[i] != candi:
                    return -1
                if A[i] == candi and B[i] != candi:
                    res[1] += 1
                elif A[i] != candi and B[i] == candi:
                    res[0] += 1
        return min(res)

猜你喜欢

转载自blog.csdn.net/sinat_36811967/article/details/88555971