题目
1232. Check If It Is a Straight Line
解题方法
首先处理类似于x=0这种斜率不存在的情况,如果斜率存在,则根据第一个点和最后一个点计算出斜率k和截距b,再遍历数组中剩余的点,根据公式y=kx+b计算x对应的y,再拿这个y和coordinates[i][1]对比,不相等说明点不在线上。
时间复杂度:O(n)
空间复杂度:O(1)
代码
class Solution:
def fun(self, x, k, b):
return k * x + b
def checkStraightLine(self, coordinates: List[List[int]]) -> bool:
if not coordinates[-1][0] - coordinates[0][0]:
x = coordinates[0][0]
for i in coordinates:
if i[0] != x:
return False
return True
k = (coordinates[-1][1] - coordinates[0][1]) / (coordinates[-1][0] - coordinates[0][0])
b = coordinates[0][1] - coordinates[0][0] * k
for i in range(1, len(coordinates)-1):
if self.fun(coordinates[i][0], k, b) != coordinates[i][1]:
return False
return True