leetcode 1007. 行相等的最少多米诺旋转（C++、python）

在一排多米诺骨牌中，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

C++

class Solution {
public:
    int minDominoRotations(vector<int>& A, vector<int>& B) 
    {
        int n=A.size();
        int m=B.size();
        if(m!=n)
        {
            return -1;
        }
        vector<int> tmp(7,0);
        for(int i=0;i<n;i++)
        {
            if(A[i]==B[i])
            {
                tmp[A[i]]++;
            }
            else
            {
                tmp[A[i]]++;
                tmp[B[i]]++;
            }
        }
        int flag=1;
        int val=0;
        for(int i=1;i<=6;i++)
        {
            if(n==tmp[i])
            {
                val=i;
                flag=0;
                break;
            }
        }
        if(flag)
        {
            return -1;
        }
        else
        {
            int up=0;
            int same=0;
            for(int i=0;i<n;i++)
            {
                if(A[i]==B[i])
                {
                    same++;
                }
                else
                {
                    if(val==A[i])
                    {
                        up++;
                    }
                }
            }
            return min(up,n-same-up);

        }
    }
};

python

class Solution:
    def minDominoRotations(self, A: List[int], B: List[int]) -> int:
        n=len(A)
        m=len(B)
        if m!=n:
            return -1
        tmp=[0 for i in range(7)]
        for i in range(n):
            if A[i]==B[i]:
                tmp[A[i]]+=1
            else:
                tmp[A[i]]+=1
                tmp[B[i]]+=1
        flag=1
        val=0
        for i in range(1,7):
            if n==tmp[i]:
                flag=0
                val=i
                break
        if flag:
            return -1
        else:
            up=0
            same=0
            for i in range(n):
                if A[i]==B[i]:
                    same+=1
                else:
                    if A[i]==val:
                        up+=1
            return min(up,n-same-up)