微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

如何使用condition检查typename T是否为C中的float类型的整数类型

我打算写一个模板来生成随机数据的向量.问题是
std :: uniform_int_distribution只接受整数类型,而float类型的std :: uniform_real_distribution.我想把两者结合起来.这是我的代码.
#include <vector>
#include <random>
#include <algorithm>
#include <iterator>
#include <functional>

template<typename T>
std::vector<T> generate_vector(size_t N,T lower = T(0),T higher = T(99)) {
    // Specify the engine and distribution. 
    if constexpr (std::is_integral<T>) {
    std::uniform_int_distribution<T> distribution(lower,higher);
    }
    else if constexpr (std::is_floating_point<T>) {
    std::uniform_real_distribution<T> distribution(lower,higher);
    }
    std::mt19937 engine; // Mersenne twister MT19937
    auto generator = std::bind(distribution,engine);
    std::vector<T> vec(N);
    std::generate(vec.begin(),vec.end(),generator);
    return vec;

我很困惑如何在if条件下实现语句.整数类型应包括:short,int,long,long long,unsigned short,unsigned int,unsigned long或unsigned long long.浮点类型包括float,double或long double.

任何帮助建议?

解决方法

在C17之前的编译器中,您可以使用模板特化来实现if-else逻辑.
// Declare a class template
template <bool is_integral,typename T> struct uniform_distribution_selector;

// Specialize for true
template <typename T> struct uniform_distribution_selector<true,T>
{
   using type = typename std::uniform_int_distribution<T>;
};

// Specialize for false
template <typename T> struct uniform_distribution_selector<false,T>
{
   using type = typename std::uniform_real_distribution<T>;
};


template<typename T>
std::vector<T> generate_vector(size_t N,T higher = T(99))
{
   // Select the appropriate distribution type.
   using uniform_distribution_type = typename uniform_distribution_selector<std::is_integral<T>::value,T>::type;

   uniform_distribution_type distribution(lower,higher);
   std::mt19937 engine;
   auto generator = std::bind(distribution,engine);
   std::vector<T> vec(N);
   std::generate(vec.begin(),generator);
   return vec;
}

原文地址:https://www.jb51.cc/c/118847.html

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。

相关推荐