《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;
}