用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));
}
相关文章