如何在c++中实现自定义生成器的复位功能？

#include <iostream>
#include <vector>

class MyGenerator {
private:
    std::vector<int> data;
    int currentIndex;

public:
    MyGenerator() {
        // 初始化生成器状态
        data = {1, 2, 3, 4, 5};
        currentIndex = 0;
    }

    void reset() {
        // 复位生成器状态
        currentIndex = 0;
    }

    int getNext() {
        // 生成下一个数据
        int next = data[currentIndex];
        currentIndex++;

        return next;
    }
};

int main() {
    MyGenerator generator;

    // 生成数据
    std::cout << generator.getNext() << std::endl;
    std::cout << generator.getNext() << std::endl;

    // 复位生成器
    generator.reset();

    // 重新生成数据
    std::cout << generator.getNext() << std::endl;
    std::cout << generator.getNext() << std::endl;

    return 0;
}