24.小R的随机播放顺序【已解决】

一比一翻译题意（删除后插入最后）

#include <iostream>
#include <vector>

std::vector<int> solution(int n, std::vector<int> a) {
    // PLEASE DO NOT MODIFY THE FUNCTION SIGNATURE
    // write code here
    std::vector<int> ans;
    for(int i = 0; i < a.size(); i++){
        ans.push_back(a[i]);
        if(i != a.size() - 1){
            int temp = a[i + 1];
            auto it = a.begin() + i + 1; 
            a.erase(it);
            a.push_back(temp);
        } 
    }
    return ans;
}

int main() {
    std::vector<int> result1 = {5, 2, 4, 1, 3};
    std::vector<int> result2 = {4, 3, 1, 2};
    std::vector<int> result3 = {1, 3, 5, 2, 6, 4};

    std::cout << (solution(5, {5, 3, 2, 1, 4}) == result1) << std::endl;
    std::cout << (solution(4, {4, 1, 3, 2}) == result2) << std::endl;
    std::cout << (solution(6, {1, 2, 3, 4, 5, 6}) == result3) << std::endl;
}