直线上最多的点数 - 哈希表、数学

题目描述

链接: LeetCode149(困难) https://leetcode-cn.com/problems/max-points-on-a-line/comments/

给定一个二维平面，平面上有 n 个点，求最多有多少个点在同一条直线上。

示例 1:

输入: [[1,1],[2,2],[3,3]]
输出: 3
解释:
^
|
|        o
|     o
|  o  
+------------->
0  1  2  3  4
示例 2:

输入: [[1,1],[3,2],[5,3],[4,1],[2,3],[1,4]]
输出: 4
解释:
^
|
|  o
|     o        o
|        o
|  o        o
+------------------->
0  1  2  3  4  5  6

思路

这个题目之所以为困难级别, 是因为有两点:

• 怎么对点和点的状态做分类并推广到所有点上
• 选取什么数据结构

解决了这两个问题之后, 就变得很简单了.

首先, 第一个问题, 如何分类点之间的状态.

我们尝试对点A和B的状态做所有分类, 无非这几种情况:

• 是同一个点
• 是同一个x轴上的点
• 是同一个y轴上的点
• 与源点的斜率为K的点

前面三种很好判断, 但是最后一种涉及到斜率, 但如果用除法, 一些测试样例是过不去的(即使能过样例, 但不是完美的解法).

解题过程

第一个问题: 分类讨论

比如从点A出发, 到点B, 点C, 点D, 那么我们将拥有最多四种不同的斜率(分别为A与B, A与C, A与D), 这样题目就变成了: 找相同斜率的点、 相同x轴的点、相同y轴的点之间的最大值.(相同斜率的点, 肯定是能被连成一个直线的)

第二个问题: 数据结构的选取

这里我们可以使用HashMap的嵌套结构, 其结构是这样的:

HashMap<x, HashMap<y, cnt>>.

注意!! 其中x和y表示: 斜率的分子和分母, 两点的xy分别做差.

那么cnt自然就变成了 x y两个值对应的斜率的点数. 这样避免了除法, 也就解决了精度的问题.

由于不同的点的斜率不一定是一样的, 但, 一定是某一个斜率的N倍, 也就是找两个数的最大公约数, 做约分. 这样我们就能把所有的斜率都归类起来.

代码

class Solution {
    class Point {
        int x;
        int y;
        Point(int x, int y) {
            this.x = x;
            this.y = y;
        }
    }
    public int maxPoints(int[][] points) {
        int result = 0;
        Map<Integer, Map<Integer, Integer>> map = new HashMap<>();
        // 表示从i点开始到其他所有点
        for (int i = 0; i < points.length; i++) {
            // 这里需要每次都清空一下
            map.clear();
            // 四种关系的点数
            // same初始化为1, 表明自身要数1个点
            int same = 1;
            int samex = 0;
            int samey = 0;
            int sameLine = 0;
            Point curPoint = new Point(points[i][0], points[i][1]);
            // 这里注意: 需要从i+1开始, 不然会重复判断
            for (int j = i + 1; j < points.length; j++) {
                Point point = new Point(points[j][0], points[j][1]);
                if (point.x == curPoint.x && point.y == curPoint.y) same++;
                else if (point.x == curPoint.x) samex++;
                else if (point.y == curPoint.y) samey++;
                else {
                    // 求最大公约数(辗转相除法), 并约分
                    int gcd = gcd(point.x - curPoint.x, point.y - curPoint.y);
                    int x = (point.x - curPoint.x) / gcd;
                    int y = (point.y - curPoint.y) / gcd;
                    if (!map.containsKey(x)) {
                        map.put(x, new HashMap<Integer, Integer>());
                    }
                    if (!map.get(x).containsKey(y)) {
                        map.get(x).put(y, 0);
                    }
                    map.get(x).put(y, map.get(x).get(y) + 1);
                    // sameLine存储的是相同斜率的点数的最大值.
                    sameLine = Math.max(sameLine, map.get(x).get(y));
                }
            }
            // 最后再比一下大小即可, 注意最后要加same(加上自身和相同的点)
            result = Math.max(result, Math.max(Math.max(samex, samey), sameLine) + same);
        }
        return result;
    }

    public int gcd(int a, int b) {
        return b == 0 ? a : gcd(b, a % b);
    }
}