您的位置:首页 > 其它

判断n是否是N的整数次方(N是2的N次方)

2014-09-24 10:48 155 查看
昨天翻了下自己以前的博客,看到这样一道题:" 判断一个数是否是4的整数次方",就想能不能用模板的方式将4扩展到2,4,8,...2^n的所有情况呢?于是将题目改为:"判断n是否是N的整数次方,其中N必须是2的整数次方且大于1,如果传的N不符合规范,则编译错误"

template <size_t N> bool isPowerOf(size_t n)


以下是代码实现

template <bool> struct StaticAssertFailed;
template <>struct StaticAssertFailed<true>{ enum{ val = true};};
#define STATIC_ASSERT(exp) StaticAssertFailed<(bool)exp>::val

template<size_t N> struct isPowOf2{ enum {val  =  !(N & (N-1))};};

template<size_t N>
struct __LogOf2
{
enum {val = 1 + __LogOf2<(N>>1)>::val};
};
template<> struct __LogOf2<1>{	enum {val = 0};};

template<size_t N>
struct LogOf2
{
enum {val = __LogOf2<N>::val};
enum {bEnsure = STATIC_ASSERT(isPowOf2<N>::val)};
};

template<size_t offset,size_t N,bool bCont>
struct __Mask{ enum { val = 1 | __Mask<offset,N + offset,N < 32>::val << offset };};
template<size_t offset,size_t N> struct __Mask<offset,N,false> { enum { val = 0};};

template<size_t N>
struct Mask{
enum { val = __Mask<LogOf2<N>::val,0,true>::val};
};

/*
判断n是否是N的整数次方
其中N必须是2的整数次方且大于1,如果传的N不符合规范,则编译错误
*/
template <size_t N>
bool isPowerOf(size_t n)
{
return  ((n&( n-1)) ==0) && (n & Mask<N>::val);
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: