1007. Minimum Domino Rotations For Equal Row
Description
In a row of dominoes, tops[i]
and bottoms[i]
represent the top and bottom halves of the ith
domino. (A domino is a tile with two numbers from 1 to 6 - one on each half of the tile.)
We may rotate the ith
domino, so that tops[i]
and bottoms[i]
swap values.
Return the minimum number of rotations so that all the values in tops
are the same, or all the values in bottoms
are the same.
If it cannot be done, return -1
.
Example 1:
Input: tops = [2,1,2,4,2,2], bottoms = [5,2,6,2,3,2] Output: 2 Explanation: The first figure represents the dominoes as given by tops and bottoms: before we do any rotations. If we rotate the second and fourth dominoes, we can make every value in the top row equal to 2, as indicated by the second figure.
Example 2:
Input: tops = [3,5,1,2,3], bottoms = [3,6,3,3,4] Output: -1 Explanation: In this case, it is not possible to rotate the dominoes to make one row of values equal.
Constraints:
2 <= tops.length <= 2 * 104
bottoms.length == tops.length
1 <= tops[i], bottoms[i] <= 6
Solution
minimum-domino-rotations-for-equal-row.py
class Solution:
def minDominoRotations(self, tops: List[int], bottoms: List[int]) -> int:
n = len(tops)
tPos = collections.defaultdict(list)
bPos = collections.defaultdict(list)
res = float('inf')
for i, x in enumerate(tops):
tPos[x].append(i)
for i, x in enumerate(bottoms):
bPos[x].append(i)
for k, v in tPos.items():
A = set(range(0, n))
for index in v:
if index in A:
A.remove(index)
for index in bPos[k]:
if index in A:
A.remove(index)
if len(A) == 0:
res = min(res, n - max(len(v), len(bPos[k])))
return -1 if res == float('inf') else res