#include "weighted_set.h" #include #include namespace Math { WeightedSet::WeightedSet(unsigned int seed) : elements() { std::srand(unsigned(std::time(0))); } void WeightedSet::add(unsigned int w, unsigned int e) { while (w > 0) { elements.push_back(e); --w; } } unsigned int WeightedSet::getRandom() { //unsigned int rnd = (int) (totalWeight * (rand() / (RAND_MAX + 1.0))); unsigned int rnd = std::rand() % (elements.size()); return elements[rnd]; } } /* #include int main(int argc, char* argv[]) { Util::Math::WeightedSet set(int(getpid()) ^ int(time(0))); set.add(1, 1); set.add(2, 2); set.add(2, 3); set.add(3, 4); set.add(3, 5); std::cout << set.getRandom() << std::endl; }*/