用C++制作随机数生成器模板

2022-05-26 00:00:00 random distribution templates c++

我知道我的代码错误。我应该有uniform_int_distribution<int>,但是我需要一个随机数生成器,它可以工作在任何类型的代码中。 我的意思是,我可以生成int,然后除以10^n得到一个浮点数,但我不喜欢它的优雅。

template <class T>
T aleaGenVal(const T &min,const T &max)
{
    std::random_device dev;
    std::mt19937 rng(dev());

    std::uniform_int_distribution<T> dist(min,max);
    return dist(rng);
}

感谢您的帮助


解决方案

std::uniform_int_distribution仅为一些基础整数类型定义,std::uniform_real_distribution仅为基础浮点类型定义。

您可以在std::conditional_t

之间进行选择

遗憾的是,有许多整型无法与std::uniform_int_distribution一起使用,因此我们必须枚举允许的整型。

template <typename> struct has_uniform_distribution : std::false_type;

template <std::floating_point T> struct has_uniform_distribution<T> : std::true_type;

template <> struct has_uniform_distribution<short> : std::true_type;
template <> struct has_uniform_distribution<unsigned short> : std::true_type;

template <> struct has_uniform_distribution<int> : std::true_type;
template <> struct has_uniform_distribution<unsigned int> : std::true_type;

template <> struct has_uniform_distribution<long> : std::true_type;
template <> struct has_uniform_distribution<unsigned long> : std::true_type;

template <> struct has_uniform_distribution<long long> : std::true_type;
template <> struct has_uniform_distribution<unsigned long long> : std::true_type;

template <typename T>
concept uniform_distribution = has_uniform_distribution<T>::value;

template <uniform_distribution T> // or sfinae over has_uniform_distribution in C++ earlier than C++20
T aleaGenVal(T min, T max)
{
    std::random_device dev;
    std::mt19937 rng(dev());

    using dist_t = std::conditional_t<
        std::is_integral_v<T>, 
        std::uniform_int_distribution<T>, 
        std::uniform_real_distribution<T>
    >;

    dist_t dist(min,max);
    return dist(rng);
}

template <typename T>
T aleaGenVal(T min, T max) = delete;

或者,我们也可以为所有算术类型定义它,方法是使用最宽的生成器类型并缩小结果范围

template <std::arithmetic T> // or sfinae over std::is_arithmetic in C++ earlier than C++20
T aleaGenVal(T min, T max)
{
    std::random_device dev;
    std::mt19937 rng(dev());

    using dist_t = std::conditional_t<
        std::is_integral_v<T>, 
        std::conditional_t<
            std::is_signed_v<T>, 
            std::uniform_int_distribution<long long>, 
            std::uniform_int_distribution<unsigned long long>> 
        std::uniform_real_distribution<T>>;

    dist_t dist(min,max);
    return static_cast<T>(dist(rng));
}

相关文章