1551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant//===----------------------------------------------------------------------===// 2551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant// 3551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant// The LLVM Compiler Infrastructure 4551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant// 5b64f8b07c104c6cc986570ac8ee0ed16a9f23976Howard Hinnant// This file is dual licensed under the MIT and the University of Illinois Open 6b64f8b07c104c6cc986570ac8ee0ed16a9f23976Howard Hinnant// Source Licenses. See LICENSE.TXT for details. 7551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant// 8551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant//===----------------------------------------------------------------------===// 9551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant 10551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant// <random> 11551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant 12551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant// template<class IntType = int> 13551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant// class discrete_distribution 14551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant 15551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant// template<class InputIterator> 16551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant// discrete_distribution(InputIterator firstW, InputIterator lastW); 17551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant 18551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant#include <random> 19551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant#include <cassert> 20551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant 21551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnantint main() 22551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant{ 23551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant { 24551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant typedef std::discrete_distribution<> D; 25551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant double p0[] = {1}; 26551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant D d(p0, p0); 27551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant std::vector<double> p = d.probabilities(); 28551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant assert(p.size() == 1); 29551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant assert(p[0] == 1); 30551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant } 31551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant { 32551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant typedef std::discrete_distribution<> D; 33551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant double p0[] = {10}; 34551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant D d(p0, p0+1); 35551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant std::vector<double> p = d.probabilities(); 36551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant assert(p.size() == 1); 37551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant assert(p[0] == 1); 38551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant } 39551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant { 40551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant typedef std::discrete_distribution<> D; 41551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant double p0[] = {10, 30}; 42551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant D d(p0, p0+2); 43551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant std::vector<double> p = d.probabilities(); 44551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant assert(p.size() == 2); 45551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant assert(p[0] == 0.25); 46551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant assert(p[1] == 0.75); 47551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant } 48551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant { 49551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant typedef std::discrete_distribution<> D; 50551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant double p0[] = {30, 10}; 51551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant D d(p0, p0+2); 52551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant std::vector<double> p = d.probabilities(); 53551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant assert(p.size() == 2); 54551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant assert(p[0] == 0.75); 55551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant assert(p[1] == 0.25); 56551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant } 57551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant { 58551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant typedef std::discrete_distribution<> D; 59551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant double p0[] = {30, 0, 10}; 60551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant D d(p0, p0+3); 61551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant std::vector<double> p = d.probabilities(); 62551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant assert(p.size() == 3); 63551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant assert(p[0] == 0.75); 64551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant assert(p[1] == 0); 65551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant assert(p[2] == 0.25); 66551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant } 67551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant { 68551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant typedef std::discrete_distribution<> D; 69551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant double p0[] = {0, 30, 10}; 70551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant D d(p0, p0+3); 71551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant std::vector<double> p = d.probabilities(); 72551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant assert(p.size() == 3); 73551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant assert(p[0] == 0); 74551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant assert(p[1] == 0.75); 75551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant assert(p[2] == 0.25); 76551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant } 77551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant { 78551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant typedef std::discrete_distribution<> D; 79551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant double p0[] = {0, 0, 10}; 80551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant D d(p0, p0+3); 81551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant std::vector<double> p = d.probabilities(); 82551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant assert(p.size() == 3); 83551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant assert(p[0] == 0); 84551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant assert(p[1] == 0); 85551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant assert(p[2] == 1); 86551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant } 87551d8e4ddb087da09eeb315d5d072c37412544b4Howard Hinnant} 88