LeetCode—— 1232 缀点成线

问题描述

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

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

   

提示：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/check-if-it-is-a-straight-line
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

执行结果

代码描述

思路：一元一次方程 y = kx + b，根据前两个点(x1, y1),(x2, y2)，求出 k = (y1-y2)/(x1-x2) 和b = (x1y2 - x2y1)/(x1-x2) , 其余点带入方程，有一个点不成立，则返回false。一定要注意，分母不能为0的情况。庆幸，测试用例中没有平行于y坐标的直线。

class Solution {
public:
    bool checkStraightLine(vector<vector<int>>& coordinates) {
        float k = 0, b = 0;
        if((coordinates[0][0]-coordinates[1][0]) == 0)  return false;
        k = (coordinates[0][1]-coordinates[1][1])/(coordinates[0][0]-coordinates[1][0]);
        b = (coordinates[0][0]*coordinates[1][1] - coordinates[0][1]*coordinates[1][0])/(coordinates[0][0]-coordinates[1][0]);

        for(int i = 2; i < coordinates.size() - 1; ++i)
        {
            if(coordinates[i][1] != (k*coordinates[i][0] + b))
                return false;
        }
        return true;
    }
};