如何从std::set中提取n个元素的样本

#include <iostream>
#include <set>
#include <vector>
#include <random>

std::vector<int> extractSampleFromSet(const std::set<int>& inputSet, int n) {
    std::vector<int> sample;
    
    // 将std::set中的元素添加到std::vector中
    for (const auto& element : inputSet) {
        sample.push_back(element);
    }
    
    // 使用随机数生成器生成n个不重复的随机索引
    std::random_device rd;
    std::mt19937 gen(rd());
    std::uniform_int_distribution<> dis(0, sample.size() - 1);
    std::set<int> randomIndices;
    
    while (randomIndices.size() < n) {
        randomIndices.insert(dis(gen));
    }
    
    // 根据随机索引提取元素样本
    std::vector<int> extractedSample;
    
    for (const auto& index : randomIndices) {
        extractedSample.push_back(sample[index]);
    }
    
    return extractedSample;
}

int main() {
    std::set<int> inputSet = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10};
    int n = 3;
    
    std::vector<int> sample = extractSampleFromSet(inputSet, n);
    
    std::cout << "提取的元素样本：";
    
    for (const auto& element : sample) {
        std::cout << element << " ";
    }
    
    std::cout << std::endl;
    
    return 0;
}