面试题 05.08. 绘制直线

绘制直线。有个单色屏幕存储在一个一维数组中，使得32个连续像素可以存放在一个 int 里。屏幕宽度为w，且w可被32整除（即一个 int 不会分布在两行上），屏幕高度可由数组长度及屏幕宽度推算得出。请实现一个函数，绘制从点(x1, y)到点(x2, y)的水平线。

给出数组的长度 length，宽度 w（以比特为单位）、直线开始位置 x1（比特为单位）、直线结束位置 x2（比特为单位）、直线所在行数 y。返回绘制过后的数组。

示例1:

输入：length = 1, w = 32, x1 = 30, x2 = 31, y = 0
输出：[3]
说明：在第0行的第30位到第31为画一条直线，屏幕表示为[0b000000000000000000000000000000011]

示例2:

输入：length = 3, w = 96, x1 = 0, x2 = 95, y = 0
输出：[-1, -1, -1]

Python 解答：

class Solution:
    def drawLine(self, length: int, w: int, x1: int, x2: int, y: int) -> List[int]:
        width = w // 32
        high = length // width
        bits = [0 for i in range(w)]
        for i in range(x1, x2+1):
            bits[i] = 1
        def retransform(arr):
            x = 0xffffffff
            res = 0
            for item in arr:
                res <<= 1
                res += item
            return ~(res^x) if arr[0] else res
        return [retransform(bits[(i-width*y)*32:(i-width*y+1)*32]) if (width*y <= i < width*(y+1)) else 0 for i in range(length) ]