《Cracking the Coding Interview》——第5章：位操作——题目8

2014-03-19 06:33

题目：用一个byte数组来模拟WxH的屏幕，每个二进制位表示一个像素。请设计一个画水平线的函数。

解法：一个点一个点地画就可以了。如果要优化的话，其实可以把中间整字节的部分一口气画了，只用1/8的力气。

代码：

// 5.8 Given a byte array, set a consecutive bit segment to '1'.
#include <cstdio>
#include <vector>
using namespace std;

typedef unsigned char byte;

inline void setBit(vector<byte> &v, int i, int j, int w)
{
v[i * w / 8 + j / 8] |= (1 << (j - (j >> 3 << 3)));
}

inline int getBit(vector<byte> &v, int i, int j, int w)
{
return !!(v[i * w / 8 + j / 8] & (1 << (j - (j >> 3 << 3))));
}

int main()
{
int n, w, h;
int x1, x2, y;
vector<byte> v;
int i, j;

while (scanf("%d%d", &n, &w) == 2) {
h = n * 8 / w;
v.resize(n);
for (i = 0; i < n; ++i) {
v[i] = 0;
}
scanf("%d%d%d", &x1, &x2, &y);

// boundary check is omitted.
for (i = x1; i <= x2; ++i) {
setBit(v, y, i, w);
}

for (i = 0; i < h; ++i) {
for (j = 0; j < w; ++j) {
putchar('0' + getBit(v, i, j, w));
}
putchar('\n');
}
}

return 0;
}