leetcode 1007. 行相等的最少多米诺旋转(Minimum Domino Rotations For Equal Row)

题目描述:

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

解法:

class Solution {
public:
    int minDominoRotations(vector<int>& A, vector<int>& B) {
        vector<int> vals(7, 0);
        vector<int> up(7, 0);
        vector<int> down(7, 0);
        int sz = A.size();
        for(int i = 0; i < sz; i++){
            vals[A[i]]++;
            up[A[i]]++;
            down[B[i]]++;
            if(A[i] != B[i]){
                vals[B[i]]++;
            }
        }
        for(int i = 1; i < 7; i++){
            if(vals[i] == sz){
                return min(sz - up[i], sz - down[i]);
            }
        }
        return -1;
    }
};

猜你喜欢

转载自www.cnblogs.com/zhanzq/p/10919443.html