网络异常,图片无法展示
|
题目
用一个下标从 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)