LeetCode 5868. 可互换矩形的组数

184 阅读1分钟

题目

用一个下标从 0 开始的二维整数数组 rectangles 来表示 n 个矩形,其中 rectangles[i] = [widthi, heighti] 表示第 i 个矩形的宽度和高度。

如果两个矩形 i 和 j(i < j)的宽高比相同,则认为这两个矩形 可互换 。更规范的说法是,两个矩形满足 widthi/heighti == widthj/heightj(使用实数除法而非整数除法),则认为这两个矩形 可互换 。

计算并返回 rectangles 中有多少对 可互换 矩形。

示例 1:

输入:rectangles = [[4,8],[3,6],[10,20],[15,30]]
输出:6
解释:下面按下标(从 0 开始)列出可互换矩形的配对情况:
- 矩形 0 和矩形 1 :4/8 == 3/6
- 矩形 0 和矩形 2 :4/8 == 10/20
- 矩形 0 和矩形 3 :4/8 == 15/30
- 矩形 1 和矩形 2 :3/6 == 10/20
- 矩形 1 和矩形 3 :3/6 == 15/30
- 矩形 2 和矩形 3 :10/20 == 15/30
示例 2:

输入:rectangles = [[4,5],[7,8]]
输出:0
解释:不存在成对的可互换矩形。

提示:

n == rectangles.length 1 <= n <= 105 rectangles[i].length == 2 1 <= widthi, heighti <= 105

解题思路

class Solution:
    def interchangeableRectangles(self, rectangles: List[List[int]]) -> int:
        from collections import defaultdict
        resDict = defaultdict(list)
        from fractions import Fraction
        for arr in rectangles:
            x,y = arr[0],arr[1]
            fa = Fraction(x,y)
            resDict[(fa.numerator,fa.denominator)].append(arr)
        res = 0
        for key,val in resDict.items():
            length = len(val)
            if length >=2:
                res+= (length*(length-1))//2
        return res


if __name__ == '__main__':
    rectangles = [[4,8],[3,6],[10,20],[15,30]]
    result = Solution().interchangeableRectangles(rectangles)
    print(result)