C++怎么实现一个蓄水池抽样算法_C++从未知大小数据流中随机抽取样本

// 检查 k 是否合法
if (k > n) {
    return stream; // 或抛出异常
}

// 初始化：前 k 个元素直接放入蓄水池
for (int i = 0; i < k; ++i) {
    reservoir[i] = stream[i];
}

// 随机种子
std::srand(static_cast<unsigned int>(std::time(nullptr)));

// 从第 k+1 个元素开始处理
for (int i = k; i < n; ++i) {
    // 生成 [0, i] 之间的随机数
    int j = std::rand() % (i + 1);

    // 如果随机数小于 k，则用当前元素替换蓄水池中索引为 j 的元素
    if (j < k) {
        reservoir[j] = stream[i];
    }
}

return reservoir;

std::vector<int> sample = reservoirSampling(dataStream, k);

std::cout << "随机抽取的 " << k << " 个样本：";
for (int val : sample) {
    std::cout << val << " ";
}
std::cout << std::endl;

return 0;

if (k >= n) {
    return stream;
}

for (int i = 0; i < k; ++i) {
    reservoir[i] = stream[i];
}

std::random_device rd;
std::mt19937 gen(rd());
std::uniform_int_distribution<> dis(0, k - 1); // 用于选择替换位置

for (int i = k; i < n; ++i) {
    std::uniform_int_distribution<> dis_i(0, i);
    if (dis_i(gen) < k) {
        reservoir[dis(gen)] = stream[i];
    }
}

return reservoir;