程序员面试金典 - 面试题 05.08. 绘制直线（位运算）

1. 题目

绘制直线。有个单色屏幕存储在一个一维数组中，使得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]

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

2. 解题

• 一行有几个int，n = w/32 个
• 从 y 行开始，那么起始下标 idx = y * n

class Solution {
public:
    vector<int> drawLine(int length, int w, int x1, int x2, int y) {
        int n = w>>5, idx = y*n;
        vector<int> ans(length,0);
        while(x1 > 31)//不在当前int内
        {
        	idx++;//下一个int
        	x1 -= 32;//区间挪动
        	x2 -= 32;
        }
        int l = x1, r;//找到了x1所在int
        for( ; x2>=0; l=0,x2-=32)
        {
        	r = min(x2,31);//右端点
        	for(int i = l; i <= r; i++)
        	{
        		ans[idx] |= 1<<(31-i);//把1挪到各个位置上
        	}
        	idx++;
        }
        return ans;
    }
};

4 ms 7.4 MB