[CareerCup] 18.3 Randomly Generate Integers 무작위 숫자 생성

3573 단어

18.3 Write a method to randomly generate a set of m integers from an array of size n. Each element must have equal probability of being chosen.
 
이 문제는 우리가 한 수조에서 m개의 숫자를 무작위로 추출하도록 한다. 모든 숫자가 추출될 확률이 같도록 요구한다. 사실 이 문제는 이전의 18.2 Shuffle Cards의 방법을 사용한다. 마찬가지로 우리는 귀속과 교체 두 가지 방법으로 할 수 있다. 귀속의 사고방식은 역소법으로 i+1=m로 거슬러 올라갈 때 수조의 앞 m개의 숫자를 추출하여 결과res에 넣고 한 층씩 되돌아온다.돌아오는 과정에서 매번 [0,i] 사이에서 무작위 수를 꺼낸 다음에 이 수가 m보다 작으면res[k]의 값은nums[i]로 업데이트합니다. 이것은 카드를 씻는 것과 같습니다. 이렇게 하면 모든 수가 꺼질 확률이 같습니다. 코드는 다음과 같습니다.
 
해법 1:
vector<int> pick(vector<int> &nums, int m, int i) {
    if (i + 1 < m) return {};
    else if (i + 1 == m) {
        vector<int> res(m);
        for (int k = 0; k < m; ++k) {
            res[k] = nums[k];
        }
        return res;
    } else {
        vector<int> res = pick(nums, m, i - 1);
        int k = rand() % (i + 1);
        if (k < m) res[k] = nums[i];
        return res;
    }
}

 
물론 대응하는 교체된 묘사법도 있다. 사고방식은 모두 같다.
 
해법 2:
vector<int> pick(vector<int> &nums, int m) {
    vector<int> res(m);
    for (int i = 0; i < m; ++i) {
        res[i] = nums[i];
    }
    for (int i = m; i < nums.size(); ++i) {
        int k = rand() % (i + 1);
        if (k < m) res[k] = nums[i];
    }
    return res;
}

 
CareerCup All in One 문제 요약

좋은 웹페이지 즐겨찾기