[LeetCode]2001. 可互换矩形的组数

互换矩形

Posted by JinFei on September 15, 2021

题目描述

用一个下标从 0 开始的二维整数数组 rectangles 来表示 n 个矩形,其中 rectangles[i] = [widthi, heighti] 表示第 i 个矩形的宽度和高度。 如果两个矩形 i 和 j(i < j)的宽高比相同,则认为这两个矩形 可互换 。更规范的说法是,两个矩形满足 widthi/heighti == widthj/heightj(使用实数除法而非整数除法),则认为这两个矩形 可互换 。 计算并返回 rectangles 中有多少对 可互换 矩形。

Example 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

Example 2:

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

Constraints:

  • n == rectangles.length
  • 1 <= n <= 105
  • rectangles[i].length == 2
  • 1 <= widthi, heighti <= 10^5

解题思路

  • 求最大公约数
  • map<pair<int,int>, int>用法

C++代码

class Solution {
public:
    int getGcd(int a, int b){
        if(b == 0){
            return a;
        }
        return getGcd(b, a % b);
    }
    long long interchangeableRectangles(vector<vector<int>>& rectangles) {
        map<pair<int,int>, int> mmap;
        long long ans = 0;
        for(int i = 0; i < rectangles.size(); i++){
            int x = rectangles[i][0];
            int y = rectangles[i][1];
            int gcd = getGcd(x, y);
            x /= gcd;
            y /= gcd;
            ans += mmap[make_pair(x, y)];   // 这里最开始mmap映射的值为0
            mmap[make_pair(x, y)]++;
        }
        return ans;
    }
};