mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
258 lines
6.5 KiB
258 lines
6.5 KiB
// bitset standard header
|
|
#ifndef _BITSET_
|
|
#define _BITSET_
|
|
#include <istream>
|
|
|
|
#ifdef _MSC_VER
|
|
#pragma pack(push,8)
|
|
#endif /* _MSC_VER */
|
|
// TEMPLATE CLASS bitset
|
|
_STD_BEGIN
|
|
template<size_t _N> class bitset {
|
|
typedef unsigned long _Ty;
|
|
public:
|
|
typedef bool element_type;
|
|
class reference {
|
|
friend class bitset<_N>;
|
|
public:
|
|
reference& operator=(bool _X)
|
|
{_Pbs->set(_Off, _X);
|
|
return (*this); }
|
|
reference& operator=(const reference& _Bs)
|
|
{_Pbs->set(_Off, bool(_Bs));
|
|
return (*this); }
|
|
reference& flip()
|
|
{_Pbs->flip(_Off);
|
|
return (*this); }
|
|
bool operator~() const
|
|
{return (!_Pbs->test(_Off)); }
|
|
operator bool() const
|
|
{return (_Pbs->test(_Off)); }
|
|
private:
|
|
reference(bitset<_N>& _X, size_t _P)
|
|
: _Pbs(&_X), _Off(_P) {}
|
|
bitset<_N> *_Pbs;
|
|
size_t _Off;
|
|
};
|
|
_STCONS(size_t, bitset_size, _N);
|
|
bool at(size_t _P) const
|
|
{if (_N <= _P)
|
|
_Xran();
|
|
return (test(_P)); }
|
|
reference at(size_t _P)
|
|
{if (_N <= _P)
|
|
_Xran();
|
|
return (reference(*this, _P)); }
|
|
bool operator[](size_t _P) const
|
|
{return (test(_P)); }
|
|
reference operator[](size_t _P)
|
|
{return (reference(*this, _P)); }
|
|
bitset()
|
|
{_Tidy(); }
|
|
bitset(unsigned long _X)
|
|
{_Tidy();
|
|
for (size_t _P = 0; _X != 0 && _P < _N; _X >>= 1, ++_P)
|
|
if (_X & 1)
|
|
set(_P); }
|
|
explicit bitset(const string& _S, size_t _P = 0,
|
|
size_t _L = (size_t)(-1))
|
|
{size_t _I;
|
|
if (_S.size() < _P)
|
|
_Xran();
|
|
if (_S.size() - _P < _L)
|
|
_L = _S.size() - _P;
|
|
if (_N < _L)
|
|
_L = _N;
|
|
_Tidy(), _P += _L;
|
|
for (_I = 0; _I < _L; ++_I)
|
|
if (_S[--_P] == '1')
|
|
set(_I);
|
|
else if (_S[_P] != '0')
|
|
_Xinv(); }
|
|
bitset<_N>& operator&=(const bitset<_N>& _R)
|
|
{for (int _I = _Nw; 0 <= _I; --_I)
|
|
_A[_I] &= _R._W(_I);
|
|
return (*this); }
|
|
bitset<_N>& operator|=(const bitset<_N>& _R)
|
|
{for (int _I = _Nw; 0 <= _I; --_I)
|
|
_A[_I] |= _R._W(_I);
|
|
return (*this); }
|
|
bitset<_N>& operator^=(const bitset<_N>& _R)
|
|
{for (int _I = _Nw; 0 <= _I; --_I)
|
|
_A[_I] ^= _R._W(_I);
|
|
return (*this); }
|
|
bitset<_N>& operator<<=(size_t _P)
|
|
{if (_P < 0)
|
|
return (*this >>= -_P);
|
|
const int _D = _P / _Nb;
|
|
if (_D != 0)
|
|
for (int _I = _Nw; 0 <= _I; --_I)
|
|
_A[_I] = _D <= _I ? _A[_I - _D] : 0;
|
|
if ((_P %= _Nb) != 0)
|
|
{for (int _I = _Nw; 0 < _I; --_I)
|
|
_A[_I] = (_A[_I] << _P)
|
|
| (_A[_I - 1] >> (_Nb - _P));
|
|
_A[0] <<= _P, _Trim(); }
|
|
return (*this); }
|
|
bitset<_N>& operator>>=(size_t _P)
|
|
{if (_P < 0)
|
|
return (*this <<= -_P);
|
|
const int _D = _P / _Nb;
|
|
if (_D != 0)
|
|
for (int _I = 0; _I <= _Nw; ++_I)
|
|
_A[_I] = _D <= _Nw - _I ? _A[_I + _D] : 0;
|
|
if ((_P %= _Nb) != 0)
|
|
{for (int _I = 0; _I < _Nw; ++_I)
|
|
_A[_I] = (_A[_I] >> _P)
|
|
| (_A[_I + 1] << (_Nb - _P));
|
|
_A[_Nw] >>= _P; }
|
|
return (*this); }
|
|
bitset<_N>& set()
|
|
{_Tidy(~(_Ty)0);
|
|
return (*this); }
|
|
bitset<_N>& set(size_t _P, bool _X = true)
|
|
{if (_N <= _P)
|
|
_Xran();
|
|
if (_X)
|
|
_A[_P / _Nb] |= (_Ty)1 << _P % _Nb;
|
|
else
|
|
_A[_P / _Nb] &= ~((_Ty)1 << _P % _Nb);
|
|
return (*this); }
|
|
bitset<_N>& reset()
|
|
{_Tidy();
|
|
return (*this); }
|
|
bitset<_N>& reset(size_t _P)
|
|
{return (set(_P, 0)); }
|
|
bitset<_N> operator~() const
|
|
{return (bitset<_N>(*this).flip()); }
|
|
bitset<_N>& flip()
|
|
{for (int _I = _Nw; 0 <= _I; --_I)
|
|
_A[_I] = ~_A[_I];
|
|
_Trim();
|
|
return (*this); }
|
|
bitset<_N>& flip(size_t _P)
|
|
{if (_N <= _P)
|
|
_Xran();
|
|
_A[_P / _Nb] ^= (_Ty)1 << _P % _Nb;
|
|
return (*this); }
|
|
unsigned long to_ulong() const
|
|
{enum {_Assertion = 1 /
|
|
(sizeof (unsigned long) % sizeof (_Ty) == 0)};
|
|
int _I = _Nw;
|
|
for (; sizeof (unsigned long) / sizeof (_Ty) <= _I; --_I)
|
|
if (_A[_I] != 0)
|
|
_Xoflo();
|
|
unsigned long _V = _A[_I];
|
|
for (; 0 <= --_I; )
|
|
_V = _V << _Nb | _A[_I];
|
|
return (_V); }
|
|
string to_string() const
|
|
{string _S;
|
|
_S.reserve(_N);
|
|
for (size_t _P = _N; 0 < _P; )
|
|
_S += test(--_P) ? '1' : '0';
|
|
return (_S); }
|
|
size_t count() const
|
|
{size_t _V = 0;
|
|
for (int _I = _Nw; 0 <= _I; --_I)
|
|
for (_Ty _X = _A[_I]; _X != 0; _X >>= 4)
|
|
_V += "\0\1\1\2\1\2\2\3"
|
|
"\1\2\2\3\2\3\3\4"[_X & 0xF];
|
|
return (_V); }
|
|
size_t size() const
|
|
{return (_N); }
|
|
bool operator==(const bitset<_N>& _R) const
|
|
{for (int _I = _Nw; 0 <= _I; --_I)
|
|
if (_A[_I] != _R._W(_I))
|
|
return (false);
|
|
return (true); }
|
|
bool operator!=(const bitset<_N>& _R) const
|
|
{return (!(*this == _R)); }
|
|
bool test(size_t _P) const
|
|
{if (_N <= _P)
|
|
_Xran();
|
|
return ((_A[_P / _Nb] & ((_Ty)1 << _P % _Nb)) != 0); }
|
|
bool any() const
|
|
{for (int _I = _Nw; 0 <= _I; --_I)
|
|
if (_A[_I] != 0)
|
|
return (true);
|
|
return (false); }
|
|
bool none() const
|
|
{return (!any()); }
|
|
bitset<_N> operator<<(size_t _R) const
|
|
{return (bitset<_N>(*this) <<= _R); }
|
|
bitset<_N> operator>>(size_t _R) const
|
|
{return (bitset<_N>(*this) >>= _R); }
|
|
friend bitset<_N> operator&(const bitset<_N>& _L,
|
|
const bitset<_N>& _R)
|
|
{return (bitset<_N>(_L) &= _R); }
|
|
friend bitset<_N> operator|(const bitset<_N>& _L,
|
|
const bitset<_N>& _R)
|
|
{return (bitset<_N>(_L) |= _R); }
|
|
friend bitset<_N> operator^(const bitset<_N>& _L,
|
|
const bitset<_N>& _R)
|
|
{return (bitset<_N>(_L) ^= _R); }
|
|
friend ostream& operator<<(ostream& _O, const bitset<_N>& _R)
|
|
{for (size_t _P = _N; 0 < _P; )
|
|
_O << (_R.test(--_P) ? '1' : '0');
|
|
return (_O); }
|
|
// TEMPLATE operator>>
|
|
friend istream& operator>>(istream& _I, bitset<_N>& _R)
|
|
{ios_base::iostate _St = ios_base::goodbit;
|
|
bool _Chg = false;
|
|
string _X;
|
|
const istream::sentry _Ok(_I);
|
|
if (_Ok)
|
|
{_TRY_IO_BEGIN
|
|
int _C = _I.rdbuf()->sgetc();
|
|
for (size_t _M = _R.size(); 0 < _M;
|
|
_C = _I.rdbuf()->snextc(), --_M)
|
|
{if (_C == EOF)
|
|
{_St |= ios_base::eofbit;
|
|
break; }
|
|
else if (_C != '0' && _C != '1')
|
|
break;
|
|
else if (_X.max_size() <= _X.size())
|
|
{_St |= ios_base::failbit;
|
|
break; }
|
|
else
|
|
_X.append(1, (char)_C), _Chg = true; }
|
|
_CATCH_IO_(_I); }
|
|
if (!_Chg)
|
|
_St |= ios_base::failbit;
|
|
_I.setstate(_St);
|
|
_R = bitset<_N>(_X);
|
|
return (_I); }
|
|
_Ty _W(size_t _I) const
|
|
{return (_A[_I]); }
|
|
private:
|
|
enum {_Nb = CHAR_BIT * sizeof (_Ty),
|
|
_Nw = _N == 0 ? 0 : (_N - 1) / _Nb};
|
|
void _Tidy(_Ty _X = 0)
|
|
{for (int _I = _Nw; 0 <= _I; --_I)
|
|
_A[_I] = _X;
|
|
if (_X != 0)
|
|
_Trim(); }
|
|
void _Trim()
|
|
{if (_N % _Nb != 0)
|
|
_A[_Nw] &= ((_Ty)1 << _N % _Nb) - 1; }
|
|
void _Xinv() const
|
|
{_THROW(invalid_argument, "invalid bitset<N> char"); }
|
|
void _Xoflo() const
|
|
{_THROW(overflow_error,
|
|
"bitset<N> conversion overflow"); }
|
|
void _Xran() const
|
|
{_THROW(out_of_range, "invalid bitset<N> position"); }
|
|
_Ty _A[_Nw + 1];
|
|
};
|
|
_STD_END
|
|
#ifdef _MSC_VER
|
|
#pragma pack(pop)
|
|
#endif /* _MSC_VER */
|
|
|
|
#endif /* _BITSET */
|
|
|
|
/*
|
|
* Copyright (c) 1994 by P.J. Plauger. ALL RIGHTS RESERVED.
|
|
* Consult your license regarding permissions and restrictions.
|
|
*/
|