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