leetcode weekly 159 周赛 5230. 缀点成线 - python

目录

1. 解答

在一个 XY 坐标系中有一些点，我们用数组 coordinates 来分别记录它们的坐标，其中 coordinates[i] = [x, y] 表示横坐标为 x、纵坐标为 y 的点。

请你来判断，这些点是否在该坐标系中属于同一条直线上，是则返回 true，否则请返回 false。

示例 1：

输入：coordinates = [[1,2],[2,3],[3,4],[4,5],[5,6],[6,7]]
输出：true

示例 2：

输入：coordinates = [[1,1],[2,2],[3,4],[4,5],[5,6],[7,7]]
输出：false

提示：

• 2 <= coordinates.length <= 1000
• coordinates[i].length == 2
• -10^4 <= coordinates[i][0], coordinates[i][1] <= 10^4
• coordinates 中不含重复的点

解答

这个题主要是浮点数相等的比较方法： abs(a-b) < 1e-6

还有我通过直线方程判断点是否共线，需要单独考虑斜率为 0 的情况

class Solution:
    def checkStraightLine(self, coordinates) -> bool:
        if coordinates[0][0] == coordinates[1][0]:
            for i in range(2, len(coordinates)):
                p = coordinates[i]
                if p[0] != coordinates[0][0]: return False
        else:
            k = (coordinates[0][1] - coordinates[1][1]) / (coordinates[0][0]-coordinates[1][0])
            b = coordinates[0][1] - coordinates[0][0] * k
            for i in range(2, len(coordinates)):
                p = coordinates[i]
                #print(p[1] - b - k * p[0])
                if abs(p[1] - b - k * p[0]) > 1e-6: return False
        return True