天天看點

STL容器之bitset

bitset用來友善地管理一系列的bit位,它并不是一個标準的容器。

bitset定義于<bitset>中:

template <size_t N> class bitset;
           

bitset的接口

構造函數

// 預設構造函數, 初始化為全0
bitset();
// 以val初始化各比特位
bitset(unsigned long val);
// 用str從pos開始的n個字元初始化bitset的前n個比特, 這n個字元必須是0或1,
// 如果含有其他字元, 抛出invalid_argument異常,
// pos如果超出str範圍抛出out_of_range異常
template<class charT, class traits, class Alloc>
  explicit bitset (const basic_string<charT,traits,Alloc>& str,
    typename basic_string<charT,traits,Alloc>::size_type pos = ,
    typename basic_string<charT,traits,Alloc>::size_type n =
      basic_string<charT,traits,Alloc>::npos);
//沒有顯式的複制構造函數和指派運算符,因為預設的就足夠了
           

運算符

// 成員函數
bitset& operator&=(const bitset& rhs);
bitset& operator|=(const bitset& rhs);
bitset& operator^=(const bitset& rhs);
bitset& operator<<=(size_t pos);
bitset& operator>>=(size_t pos);
bitset operator~() const;
bitset operator<<(size_t pos) const;
bitset operator>>(size_t pos) const;
bool operator==(const bitset& rhs) const;
bool operator!=(const bitset& rhs) const;
// 非成員函數
template <size_t N>
  bitset<N> operator&(const bitset<N>& lhs, const bitset<N>& rhs);
template <size_t N>
  bitset<N> operator|(const bitset<N>& lhs, const bitset<N>& rhs);
template <size_t N>
  bitset<N> operator^(const bitset<N>& lhs, const bitset<N>& rhs);
// 輸入輸出
template<class charT, class traits, size_t N>
  basic_istream<charT, traits>&
    operator>> (basic_istream<charT,traits>& is, bitset<N>& rhs);
template<class charT, class traits, size_t N>
  basic_ostream<charT, traits>&
    operator<< (basic_ostream<charT,traits>& os, const bitset<N>& rhs);
           

成員通路

// 除了不進行越界檢查, 與test()行為相同
bool operator[](size_t pos) const;
// 傳回一個引用, 可進行指派操作
reference operator[](size_t pos);
// 傳回bitset中1的個數
size_t count() const;
// 傳回bitset的大小
size_t size() const;
// 檢測是否為1, 可能會抛出out_of_range異常
bool test(size_t pos) const;
// 如存在任意一個為1的位則傳回true
bool any() const;
// 全零傳回true, 等價于!any();
bool none() const;
           

位操作

// 将所有位置為1
bitset& set();
// 将相應位置為val, 可能會抛出out_of_range異常
bitset& set(size_t pos, bool val = true);
// 将所有位清零
bitset& reset();
// 将相應位清零, 可能會抛出out_of_range異常
bitset& reset(size_t pos);
// 翻轉所有位
bitset& flip();
// 翻轉相應位, 可能會抛出out_of_range異常
bitset& flip(size_t pos);
           

轉換

// 傳回一個01字元串, 與<<運算符産生的輸出相同
template <class charT, class traits, class Alloc>
  basic_string<charT,traits,Alloc> to_string() const;
// 傳回一個unsigned long, 如果bitset太大以至于無法轉換, 抛出overflow_error異常
unsigned long to_ulong() const;
           

解析

内部資料結構

bitset内部以unsigned long數組來存儲各比特位,是以如果unsigned long占4位元組的話,sizeof(bitset對象)的值總是4的整數倍。哦,sizeof(bitset<0>)除外,此時unsigned long數組長度為0,bitset<0>就是一個沒有資料成員的類,其大小是1。

在sgi的實作裡,定義了_Base_bitset的模闆類,bitset私有繼承此類,bitset的大部分操作都在_Base_bitset裡完成,unsigned long數組也定義在_Base_bitset裡:

unsigned long M_w[Nw];
           

其中Nw是所需unsigned long的數目,可由(n+31)/32算得。

M_w數組的下标越大,所表示比特序列的位數越高。

_Base_bitset還針對隻有一個unsigned long的情況進行了特殊化,提高了效率。

查表法

count()内部使用了查表法。先定義了一個256長度的unsigned char數組,用來存儲所有unsigned char對應的1的個數。将每個unsigned long分成4個unsigned char,然後以unsigned char為數組下标直接可獲得1的個數,依次累加即可獲得最終結果。

未使用部分

由于unsigned long數組不一定恰好用完,未使用部分由一個内部函數進行清零操作。在構造對象,或進行有可能更改未使用部分的操作時,都調用了此函數。是以,未使用的部分始終是0。

reference類型

bitset内部定義了一個reference類,它是operator[]其中一個版本的傳回值,operator[]有兩個版本:

bool operator[] (size_t pos) const;
reference operator[] (size_t pos);
           

第一個版本隻讀,第二個版本可以寫。reference就是為了實作operator[]寫的功能。

reference operator[](size_t pos) { return reference(*this, pos); }
           

reference有一個unsigned long *M_wp的資料成員,通過

reference( bitset& b, size_t pos );
           

進行構造時,M_wp會指向pos對應的那個unsigned long,這樣就可以通過M_wp來對b進行操作了。

除了可以對b[i]進行指派操作,還可以對它進行

~b[i]和b[i].flip()

完整源碼(摘自std_bitset.h,有修改)

// 内部以unsigned long來存儲比特位, 是以占用位元組數是sizeof(unsigned long)的整數倍
// 大小為n的bitset占用的unsigned long的數量為__BITSET_WORDS(n)
// 在unsigned long占4位元組的機器, bitset<0>占1位元組, bitset<1>占4位元組, bitset<32>占4位元組, bitset<33>占8位元組
// 未使用比特始終是0
#define _GLIBCPP_BITSET_BITS_PER_WORD (CHAR_BIT*sizeof(unsigned long))
#define __BITSET_WORDS(__n) \
 ((__n) <  ?  : ((__n) + _GLIBCPP_BITSET_BITS_PER_WORD - )/_GLIBCPP_BITSET_BITS_PER_WORD)
/**
// gcc 4.7.1裡是這樣的
#define _GLIBCXX_BITSET_WORDS(__n) \
  ((__n) / _GLIBCXX_BITSET_BITS_PER_WORD + \
   ((__n) % _GLIBCXX_BITSET_BITS_PER_WORD == 0 ? 0 : 1))

   差別在于當n為0的時候, 前者是1, 後者是0
   實際上, 當n為0的時, _Base_bitset的_M_w數組大小為0即可, 此時對象占用空間為1
   不做n<1的判斷即等價:
   #define __BITSET_WORDS(__n) \
 (((__n) + _GLIBCPP_BITSET_BITS_PER_WORD - 1)/_GLIBCPP_BITSET_BITS_PER_WORD)
*/

// 查表法獲得unsigned char中1的個數, 可在哪初始化的, 我咋沒找着
template<bool __dummy>
struct _Bit_count {
  static unsigned char _S_bit_count[];
};

// Mapping from 8 bit unsigned integers to the index of the first one bit
template<bool __dummy>
struct _First_one {
  static unsigned char _S_first_one[];
};


// _Nw是bitset占用Word的個數
template<size_t _Nw>
struct _Base_bitset {
  typedef unsigned long _WordT;

  _WordT _M_w[_Nw]; // 實際的存儲位置, 從低位到高位

  _Base_bitset( void ) { _M_do_reset(); }
  _Base_bitset(unsigned long __val) {
    _M_do_reset();
    _M_w[] = __val;
  }
  void _M_do_reset() { memset(_M_w, , _Nw * sizeof(_WordT)); }


  static size_t _S_whichword( size_t __pos )
    { return __pos / _GLIBCPP_BITSET_BITS_PER_WORD; }
  static size_t _S_whichbyte( size_t __pos )
    { return (__pos % _GLIBCPP_BITSET_BITS_PER_WORD) / CHAR_BIT; }
  static size_t _S_whichbit( size_t __pos )
    { return __pos % _GLIBCPP_BITSET_BITS_PER_WORD; }
    // 獲得掩碼, 以Word為機關操作
  static _WordT _S_maskbit( size_t __pos )
    { return (static_cast<_WordT>()) << _S_whichbit(__pos); }

  _WordT& _M_getword(size_t __pos)       { return _M_w[_S_whichword(__pos)]; }
  _WordT  _M_getword(size_t __pos) const { return _M_w[_S_whichword(__pos)]; }

  // 最高位Word
  _WordT& _M_hiword()       { return _M_w[_Nw - ]; }
  _WordT  _M_hiword() const { return _M_w[_Nw - ]; }

  void _M_do_and(const _Base_bitset<_Nw>& __x) {
    for ( size_t __i = ; __i < _Nw; __i++ ) {
      _M_w[__i] &= __x._M_w[__i];
    }
  }

  void _M_do_or(const _Base_bitset<_Nw>& __x) {
    for ( size_t __i = ; __i < _Nw; __i++ ) {
      _M_w[__i] |= __x._M_w[__i];
    }
  }

  void _M_do_xor(const _Base_bitset<_Nw>& __x) {
    for ( size_t __i = ; __i < _Nw; __i++ ) {
      _M_w[__i] ^= __x._M_w[__i];
    }
  }

  void _M_do_left_shift(size_t __shift);
  void _M_do_right_shift(size_t __shift);

  void _M_do_flip() {
    for ( size_t __i = ; __i < _Nw; __i++ ) {
      _M_w[__i] = ~_M_w[__i];
    }
  }

  void _M_do_set() {
    for ( size_t __i = ; __i < _Nw; __i++ ) {
      _M_w[__i] = ~static_cast<_WordT>();
    }
  }

  bool _M_is_equal(const _Base_bitset<_Nw>& __x) const {
    for (size_t __i = ; __i < _Nw; ++__i) {
      if (_M_w[__i] != __x._M_w[__i])
        return false;
    }
    return true;
  }

  bool _M_is_any() const {
    for ( size_t __i = ; __i < _Nw; __i++ ) {
      if ( _M_w[__i] != static_cast<_WordT>() )
        return true;
    }
    return false;
  }

  // 統計1的個數, 直接強轉成char*, 靈活極了
  size_t _M_do_count() const {
    size_t __result = ;
    const unsigned char* __byte_ptr = (const unsigned char*)_M_w;
    const unsigned char* __end_ptr = (const unsigned char*)(_M_w+_Nw);

    while ( __byte_ptr < __end_ptr ) {
      // 用查表法得到unsigned char中1的個數
      __result += _Bit_count<true>::_S_bit_count[*__byte_ptr];
      __byte_ptr++;
    }
    return __result;
  }

  unsigned long _M_do_to_ulong() const;

  // find first "on" bit
  size_t _M_do_find_first(size_t __not_found) const;

  // find the next "on" bit that follows "prev"
  size_t _M_do_find_next(size_t __prev, size_t __not_found) const;
};


// _Base_bitset的非内聯函數定義
template<size_t _Nw>
void _Base_bitset<_Nw>::_M_do_left_shift(size_t __shift)
{
  if (__shift != ) {
    const size_t __wshift = __shift / _GLIBCPP_BITSET_BITS_PER_WORD;
    const size_t __offset = __shift % _GLIBCPP_BITSET_BITS_PER_WORD;

    if (__offset == )
      for (size_t __n = _Nw - ; __n >= __wshift; --__n)
        _M_w[__n] = _M_w[__n - __wshift];

    else {
      const size_t __sub_offset = _GLIBCPP_BITSET_BITS_PER_WORD - __offset;
      for (size_t __n = _Nw - ; __n > __wshift; --__n)
        _M_w[__n] = (_M_w[__n - __wshift] << __offset) |
                    (_M_w[__n - __wshift - ] >> __sub_offset);
      _M_w[__wshift] = _M_w[] << __offset;
    }

    fill(_M_w + , _M_w + __wshift, static_cast<_WordT>());
  }
}

template<size_t _Nw>
void _Base_bitset<_Nw>::_M_do_right_shift(size_t __shift)
{
  if (__shift != ) {
    const size_t __wshift = __shift / _GLIBCPP_BITSET_BITS_PER_WORD;
    const size_t __offset = __shift % _GLIBCPP_BITSET_BITS_PER_WORD;
    const size_t __limit = _Nw - __wshift - ;

    if (__offset == )
      for (size_t __n = ; __n <= __limit; ++__n)
        _M_w[__n] = _M_w[__n + __wshift];

    else {
      const size_t __sub_offset = _GLIBCPP_BITSET_BITS_PER_WORD - __offset;
      for (size_t __n = ; __n < __limit; ++__n)
        _M_w[__n] = (_M_w[__n + __wshift] >> __offset) |
                    (_M_w[__n + __wshift + ] << __sub_offset);
      _M_w[__limit] = _M_w[_Nw-] >> __offset;
    }

    fill(_M_w + __limit + , _M_w + _Nw, static_cast<_WordT>());
  }
}

// 僅傳回最低字, 如高位有資料, 抛出overflow_error異常
template<size_t _Nw>
unsigned long _Base_bitset<_Nw>::_M_do_to_ulong() const
{
  for (size_t __i = ; __i < _Nw; ++__i)
    if (_M_w[__i])
      __STL_THROW(overflow_error("bitset"));

  return _M_w[];
}

// 同樣使用了查表法
template<size_t _Nw>
size_t _Base_bitset<_Nw>::_M_do_find_first(size_t __not_found) const
{
  for ( size_t __i = ; __i < _Nw; __i++ ) {
    _WordT __thisword = _M_w[__i];
    if ( __thisword != static_cast<_WordT>() ) {
      // find byte within word
      for ( size_t __j = ; __j < sizeof(_WordT); __j++ ) {
        unsigned char __this_byte
          = static_cast<unsigned char>(__thisword & (~(unsigned char)));
        if ( __this_byte )
          return __i*_GLIBCPP_BITSET_BITS_PER_WORD + __j*CHAR_BIT +
            _First_one<true>::_S_first_one[__this_byte];

        __thisword >>= CHAR_BIT;
      }
    }
  }
  // not found, so return an indication of failure.
  return __not_found;
}

template<size_t _Nw>
size_t
_Base_bitset<_Nw>::_M_do_find_next(size_t __prev, size_t __not_found) const
{
  // make bound inclusive
  ++__prev;

  // check out of bounds
  if ( __prev >= _Nw * _GLIBCPP_BITSET_BITS_PER_WORD )
    return __not_found;

    // search first word
  size_t __i = _S_whichword(__prev);
  _WordT __thisword = _M_w[__i];

    // mask off bits below bound
  __thisword &= (~static_cast<_WordT>()) << _S_whichbit(__prev);

  if ( __thisword != static_cast<_WordT>() ) {
    // find byte within word
    // get first byte into place
    __thisword >>= _S_whichbyte(__prev) * CHAR_BIT;
    for ( size_t __j = _S_whichbyte(__prev); __j < sizeof(_WordT); __j++ ) {
      unsigned char __this_byte
        = static_cast<unsigned char>(__thisword & (~(unsigned char)));
      if ( __this_byte )
        return __i*_GLIBCPP_BITSET_BITS_PER_WORD + __j*CHAR_BIT +
          _First_one<true>::_S_first_one[__this_byte];

      __thisword >>= CHAR_BIT;
    }
  }

  // check subsequent words
  __i++;
  for ( ; __i < _Nw; __i++ ) {
    __thisword = _M_w[__i];
    if ( __thisword != static_cast<_WordT>() ) {
      // find byte within word
      for ( size_t __j = ; __j < sizeof(_WordT); __j++ ) {
        unsigned char __this_byte
          = static_cast<unsigned char>(__thisword & (~(unsigned char)));
        if ( __this_byte )
          return __i*_GLIBCPP_BITSET_BITS_PER_WORD + __j*CHAR_BIT +
            _First_one<true>::_S_first_one[__this_byte];

        __thisword >>= CHAR_BIT;
      }
    }
  }

  // not found, so return an indication of failure.
  return __not_found;
}


// 隻有一個word時的特化
template<> struct _Base_bitset<> {
  typedef unsigned long _WordT;
  _WordT _M_w;

  _Base_bitset( void ) : _M_w() {}
  _Base_bitset(unsigned long __val) : _M_w(__val) {}

  static size_t _S_whichword( size_t __pos )
    { return __pos / _GLIBCPP_BITSET_BITS_PER_WORD; }
  static size_t _S_whichbyte( size_t __pos )
    { return (__pos % _GLIBCPP_BITSET_BITS_PER_WORD) / CHAR_BIT; }
  static size_t _S_whichbit( size_t __pos )
    {  return __pos % _GLIBCPP_BITSET_BITS_PER_WORD; }
  static _WordT _S_maskbit( size_t __pos )
    { return (static_cast<_WordT>()) << _S_whichbit(__pos); }

  _WordT& _M_getword(size_t)       { return _M_w; }
  _WordT  _M_getword(size_t) const { return _M_w; }

  _WordT& _M_hiword()       { return _M_w; }
  _WordT  _M_hiword() const { return _M_w; }

  void _M_do_and(const _Base_bitset<>& __x) { _M_w &= __x._M_w; }
  void _M_do_or(const _Base_bitset<>& __x)  { _M_w |= __x._M_w; }
  void _M_do_xor(const _Base_bitset<>& __x) { _M_w ^= __x._M_w; }
  void _M_do_left_shift(size_t __shift)     { _M_w <<= __shift; }
  void _M_do_right_shift(size_t __shift)    { _M_w >>= __shift; }
  void _M_do_flip()                       { _M_w = ~_M_w; }
  void _M_do_set()                        { _M_w = ~static_cast<_WordT>(); }
  void _M_do_reset()                      { _M_w = ; }

  bool _M_is_equal(const _Base_bitset<>& __x) const
    { return _M_w == __x._M_w; }
  bool _M_is_any() const
    { return _M_w != ; }

  size_t _M_do_count() const {
    size_t __result = ;
    const unsigned char* __byte_ptr = (const unsigned char*)&_M_w;
    const unsigned char* __end_ptr
      = ((const unsigned char*)&_M_w)+sizeof(_M_w);
    while ( __byte_ptr < __end_ptr ) {
      __result += _Bit_count<true>::_S_bit_count[*__byte_ptr];
      __byte_ptr++;
    }
    return __result;
  }

  unsigned long _M_do_to_ulong() const { return _M_w; }

  size_t _M_do_find_first(size_t __not_found) const;

  // find the next "on" bit that follows "prev"
  size_t _M_do_find_next(size_t __prev, size_t __not_found) const;

};


// 輔助類, 用來将最高word沒有用到的比特清零
template <size_t _Extrabits> struct _Sanitize {
  static void _M_do_sanitize(unsigned long& __val)
    { __val &= ~((~static_cast<unsigned long>()) << _Extrabits); }
};

template<> struct _Sanitize<> {
  static void _M_do_sanitize(unsigned long) {}
};


// Class bitset
template<size_t _Nb>
class bitset : private _Base_bitset<__BITSET_WORDS(_Nb)>
{
private:
  typedef _Base_bitset<__BITSET_WORDS(_Nb)> _Base;
  typedef unsigned long _WordT;

private:
  // 将最高word沒有用到的比特清零
  void _M_do_sanitize() {
    _Sanitize<_Nb%_GLIBCPP_BITSET_BITS_PER_WORD>::_M_do_sanitize(this->_M_hiword());
  }

public:

  // bit reference:
  class reference;
  friend class reference;

  class reference {
    friend class bitset;

    _WordT *_M_wp;
    size_t _M_bpos;

    // left undefined
    reference();

  public:
    reference( bitset& __b, size_t __pos ) {
      _M_wp = &__b._M_getword(__pos);
      _M_bpos = _Base::_S_whichbit(__pos);
    }

    ~reference() {}

    // for b[i] = __x;
    reference& operator=(bool __x) {
      if ( __x )
        *_M_wp |= _Base::_S_maskbit(_M_bpos);
      else
        *_M_wp &= ~_Base::_S_maskbit(_M_bpos);

      return *this;
    }

    // for b[i] = b[__j];
    reference& operator=(const reference& __j) {
      if ( (*(__j._M_wp) & _Base::_S_maskbit(__j._M_bpos)) )
        *_M_wp |= _Base::_S_maskbit(_M_bpos);
      else
        *_M_wp &= ~_Base::_S_maskbit(_M_bpos);

      return *this;
    }

    // flips the bit
    bool operator~() const
      { return (*(_M_wp) & _Base::_S_maskbit(_M_bpos)) == ; }

    // for __x = b[i];
    operator bool() const
      { return (*(_M_wp) & _Base::_S_maskbit(_M_bpos)) != ; }

    // for b[i].flip();
    reference& flip() {
      *_M_wp ^= _Base::_S_maskbit(_M_bpos);
      return *this;
    }
  };

  // constructors:
  bitset() {}
  bitset(unsigned long __val) : _Base_bitset<__BITSET_WORDS(_Nb)>(__val)
    { _M_do_sanitize(); }

  template<class _CharT, class _Traits, class _Alloc>
  explicit bitset(const basic_string<_CharT, _Traits, _Alloc>& __s,
                  size_t __pos = )
    : _Base()
  {
    if (__pos > __s.size())
      __STL_THROW(out_of_range("bitset"));
    _M_copy_from_string(__s, __pos,
                        basic_string<_CharT, _Traits, _Alloc>::npos);
  }
  template<class _CharT, class _Traits, class _Alloc>
  bitset(const basic_string<_CharT, _Traits, _Alloc>& __s,
         size_t __pos,
         size_t __n)
    : _Base()
  {
    if (__pos > __s.size())
      __STL_THROW(out_of_range("bitset"));
    _M_copy_from_string(__s, __pos, __n);
  }

  // bitset operations:
  bitset<_Nb>& operator&=(const bitset<_Nb>& __rhs) {
    this->_M_do_and(__rhs);
    return *this;
  }

  bitset<_Nb>& operator|=(const bitset<_Nb>& __rhs) {
    this->_M_do_or(__rhs);
    return *this;
  }

  bitset<_Nb>& operator^=(const bitset<_Nb>& __rhs) {
    this->_M_do_xor(__rhs);
    return *this;
  }

  bitset<_Nb>& operator<<=(size_t __pos) {
    this->_M_do_left_shift(__pos);
    this->_M_do_sanitize();
    return *this;
  }

  bitset<_Nb>& operator>>=(size_t __pos) {
    this->_M_do_right_shift(__pos);
    this->_M_do_sanitize();
    return *this;
  }

  // 單比特操作
  bitset<_Nb>& _Unchecked_set(size_t __pos) {
    this->_M_getword(__pos) |= _Base::_S_maskbit(__pos);
    return *this;
  }

  bitset<_Nb>& _Unchecked_set(size_t __pos, int __val) {
    if (__val)
      this->_M_getword(__pos) |= _Base::_S_maskbit(__pos);
    else
      this->_M_getword(__pos) &= ~_Base::_S_maskbit(__pos);

    return *this;
  }

  bitset<_Nb>& _Unchecked_reset(size_t __pos) {
    this->_M_getword(__pos) &= ~_Base::_S_maskbit(__pos);
    return *this;
  }

  bitset<_Nb>& _Unchecked_flip(size_t __pos) {
    this->_M_getword(__pos) ^= _Base::_S_maskbit(__pos);
    return *this;
  }

  bool _Unchecked_test(size_t __pos) const {
    return (this->_M_getword(__pos) & _Base::_S_maskbit(__pos))
      != static_cast<_WordT>();
  }

  // Set, reset, and flip
  bitset<_Nb>& set() {
    this->_M_do_set();
    this->_M_do_sanitize();
    return *this;
  }

  bitset<_Nb>& set(size_t __pos, bool __val = true) {
    if (__pos >= _Nb)
      __STL_THROW(out_of_range("bitset"));

    return _Unchecked_set(__pos, __val);
  }

  bitset<_Nb>& reset() {
    this->_M_do_reset();
    return *this;
  }

  bitset<_Nb>& reset(size_t __pos) {
    if (__pos >= _Nb)
      __STL_THROW(out_of_range("bitset"));

    return _Unchecked_reset(__pos);
  }

  bitset<_Nb>& flip() {
    this->_M_do_flip();
    this->_M_do_sanitize();
    return *this;
  }

  bitset<_Nb>& flip(size_t __pos) {
    if (__pos >= _Nb)
      __STL_THROW(out_of_range("bitset"));

    return _Unchecked_flip(__pos);
  }

  bitset<_Nb> operator~() const {
    return bitset<_Nb>(*this).flip();
  }

  // element access
  reference operator[](size_t __pos) { return reference(*this,__pos); }
  bool operator[](size_t __pos) const { return _Unchecked_test(__pos); }

  unsigned long to_ulong() const { return this->_M_do_to_ulong(); }

  template <class _CharT, class _Traits, class _Alloc>
  basic_string<_CharT, _Traits, _Alloc> to_string() const {
    basic_string<_CharT, _Traits, _Alloc> __result;
    _M_copy_to_string(__result);
    return __result;
  }

  // Helper functions for string operations.
  template<class _CharT, class _Traits, class _Alloc>
  void _M_copy_from_string(const basic_string<_CharT,_Traits,_Alloc>& __s,
                          size_t, size_t);

  template<class _CharT, class _Traits, class _Alloc>
  void _M_copy_to_string(basic_string<_CharT,_Traits,_Alloc>&) const;

  size_t count() const { return this->_M_do_count(); }

  size_t size() const { return _Nb; }

  bool operator==(const bitset<_Nb>& __rhs) const {
    return this->_M_is_equal(__rhs);
  }
  bool operator!=(const bitset<_Nb>& __rhs) const {
    return !this->_M_is_equal(__rhs);
  }

  bool test(size_t __pos) const {
    if (__pos >= _Nb)
      __STL_THROW(out_of_range("bitset"));
    return _Unchecked_test(__pos);
  }

  bool any() const { return this->_M_is_any(); }
  bool none() const { return !this->_M_is_any(); }

  bitset<_Nb> operator<<(size_t __pos) const
    { return bitset<_Nb>(*this) <<= __pos; }
  bitset<_Nb> operator>>(size_t __pos) const
    { return bitset<_Nb>(*this) >>= __pos; }

  // 擴充部分, 非标準的一部分, 不要依賴這些非标準部分
  // find the index of the first "on" bit
  size_t _Find_first() const
    { return this->_M_do_find_first(_Nb); }

  // find the index of the next "on" bit after prev
  size_t _Find_next( size_t __prev ) const
    { return this->_M_do_find_next(__prev, _Nb); }
};

// bitset和string的互相轉換函數
template <size_t _Nb>
template<class _CharT, class _Traits, class _Alloc>
void bitset<_Nb>
  ::_M_copy_from_string(const basic_string<_CharT,_Traits,_Alloc>& __s,
                        size_t __pos, size_t __n)
{
  reset();
  const size_t __nbits = min(_Nb, min(__n, __s.size() - __pos));
  for (size_t __i = ; __i < __nbits; ++__i) {
    switch(__s[__pos + __nbits - __i - ]) {
    case '0':
      break;
    case '1':
      set(__i);
      break;
    default:
      __STL_THROW(invalid_argument("bitset"));
    }
  }
}

template <size_t _Nb>
template <class _CharT, class _Traits, class _Alloc>
void bitset<_Nb>
  ::_M_copy_to_string(basic_string<_CharT, _Traits, _Alloc>& __s) const
{
  __s.assign(_Nb, '0');

  for (size_t __i = ; __i < _Nb; ++__i)
    if (_Unchecked_test(__i))
      __s[_Nb -  - __i] = '1';
}


// 運算符函數:與,或,異或,<<,>>
template <size_t _Nb>
inline bitset<_Nb> operator&(const bitset<_Nb>& __x, const bitset<_Nb>& __y) {
  bitset<_Nb> __result(__x);
  __result &= __y;
  return __result;
}

template <size_t _Nb>
inline bitset<_Nb> operator|(const bitset<_Nb>& __x, const bitset<_Nb>& __y) {
  bitset<_Nb> __result(__x);
  __result |= __y;
  return __result;
}

template <size_t _Nb>
inline bitset<_Nb> operator^(const bitset<_Nb>& __x, const bitset<_Nb>& __y) {
  bitset<_Nb> __result(__x);
  __result ^= __y;
  return __result;
}

// 先将輸入讀入一個臨時字元串, 再用字元串初始化bitset, 可能會抛出invalid_argument異常
template <class _CharT, class _Traits, size_t _Nb>
basic_istream<_CharT, _Traits>&
operator>>(basic_istream<_CharT, _Traits>& __is, bitset<_Nb>& __x)
{
  typedef typename _Traits::char_type char_type;
  basic_string<_CharT, _Traits> __tmp;
  __tmp.reserve(_Nb);

  // Skip whitespace
  typename basic_istream<_CharT, _Traits>::sentry __sentry(__is);
  if (__sentry) {
    basic_streambuf<_CharT, _Traits>* __buf = __is.rdbuf();
    for (size_t __i = ; __i < _Nb; ++__i) {
      static typename _Traits::int_type __eof = _Traits::eof();

      typename _Traits::int_type __c1 = __buf->sbumpc();
      if (_Traits::eq_int_type(__c1, __eof)) {
        __is.setstate(ios_base::eofbit);
        break;
      }
      else {
        char_type __c2 = _Traits::to_char_type(__c1);
        char_type __c  = __is.narrow(__c2, '*');

        if (__c == '0' || __c == '1')
          __tmp.push_back(__c);
        else if (_Traits::eq_int_type(__buf->sputbackc(__c2), __eof)) {
          __is.setstate(ios_base::failbit);
          break;
        }
      }
    }

    if (__tmp.empty())
      __is.setstate(ios_base::failbit);
    else
      __x._M_copy_from_string(__tmp, static_cast<size_t>(), _Nb);
  }

  return __is;
}

// 先得到對應的字元串, 再輸出
template <class _CharT, class _Traits, size_t _Nb>
basic_ostream<_CharT, _Traits>&
operator<<(basic_ostream<_CharT, _Traits>& __os, const bitset<_Nb>& __x)
{
  basic_string<_CharT, _Traits> __tmp;
  __x._M_copy_to_string(__tmp);
  return __os << __tmp;
}