《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 /  + j / ] |= ( << (j - (j >>  << )));

 }

 inline int getBit(vector<byte> &v, int i, int j, int w)

 {

     return !!(v[i * w /  + j / ] & ( << (j - (j >>  << ))));

 }

 int main()

 {

     int n, w, h;

     int x1, x2, y;

     vector<byte> v;

     int i, j;

     while (scanf("%d%d", &n, &w) == ) {

         h = n *  / w;

         v.resize(n);

         for (i = ; i < n; ++i) {

             v[i] = ;

         }

         scanf("%d%d%d", &x1, &x2, &y);

         // boundary check is omitted.

         for (i = x1; i <= x2; ++i) {

             setBit(v, y, i, w);

         }

         for (i = ; i < h; ++i) {

             for (j = ; j < w; ++j) {

                 putchar('' + getBit(v, i, j, w));

             }

             putchar('\n');

         }

     }

     return ;

 }