Picking an Object randomly based on another Object and probability?

Question

I have a base class called bait that has children classes some bait names, suppose bait1,bait2...etc and another fishes classes fish1,fish2,...etc

I want the user to pick a bait and for each bait there is a probability that he catches a certain fish eg. if he chose bait1, there's a probability 40% he catches fish1 and a probability 30% he catches fish2 and a probability 25% he catches fish3 and a probability 5% he catches fish4, and if he chooses bait2 same as example but with different probabilities, any idea how to implement this ?

sorry I fixed the problem with the previous example

Answer 1

Here is solution for bait 1 using std::discrete_distribution, you can see a live example here:

#include <iostream>
#include <map>
#include <random>

int main()
{
    std::random_device rd;
    std::mt19937 gen(rd());
    std::discrete_distribution<> bait1({40,30,25,5});
    std::map<int, int> m;
    for(int n=0; n<10000; ++n) {
        ++m[bait1(gen)];
    }
    for(auto p : m) {
        std::cout << "fish " <<  (p.first+1) << " generated " << p.second << " times\n";
    }
}

sample output:

fish 1 generated 3964 times
fish 2 generated 2981 times
fish 3 generated 2544 times
fish 4 generated 511 times

modifying this for bait 2 etc... should be straight forward