面试题 05.08. 绘制直线（中等）- LeetCode

题目描述

自己解法

class Solution:
    def drawLine(self, length: int, w: int, x1: int, x2: int, y: int) -> List[int]:
        ans = [0 for i in range(length)]
        arr = [0 for i in range(32 * length)]
        ## 绘制直线
        if x1 > x2:
            x1,x2 = x2,x1
        for i in range(x1+w*y,x2+1+w*y):
            arr[i] = 1
        ## 补码转原码
        start = int(y * w / 32) 
        for i in range(w*y,w*(y+1),32): ## 该行每个int开头符号位    
            if arr[i] == 1:
                for j in range(i+1,i+32):
                    if arr[j] == 1:
                        arr[j] = 0
                    else:
                        arr[j] = 1
                for m in range(i+31,i,-1):
                    if arr[m] == 1:
                        arr[m] = 0
                    else:
                        arr[m] = 1
                        break
            num,factor = 0,1
            for l in range(i+31,i,-1):
                num += arr[l] * factor
                factor *= 2
            ans[start] = -num if arr[i] == 1 else num
            start += 1 
        return ans