1.题目
问题描述
小R有一个特殊的随机播放规则。他首先播放歌单中的第一首歌，播放后将其从歌单中移除。如果歌单中还有歌曲，则会将当前第一首歌移到最后一首。这个过程会一直重复，直到歌单中没有任何歌曲。
例如，给定歌单 [5, 3, 2, 1, 4]，真实的播放顺序是 [5, 2, 4, 1, 3]。
保证歌曲中的id两两不同。
测试样例
样例1：
输入：n = 5 ,a = [5, 3, 2, 1, 4]
输出：
[5, 2, 4, 1, 3]
样例2：
输入：n = 4 ,a = [4, 1, 3, 2]
输出：
[4, 3, 1, 2]
样例3：
输入：n = 6 ,a = [1, 2, 3, 4, 5, 6]
输出：
[1, 3, 5, 2, 6, 4]
2.思路
用队列存放原始歌单，模拟题目中的播放规则
3.代码
#include <iostream>
#include <vector>
#include <queue>
using namespace std;
std::vector<int> solution(int n, std::vector<int> a) {
// PLEASE DO NOT MODIFY THE FUNCTION SIGNATURE
// write code here
queue<int> original_playlist; // 原始歌单
vector<int> final_playlist; // 最终播放顺序
// 将原始歌单放入队列中
for (int i = 0; i < a.size(); i++) {
original_playlist.push(a[i]);
}
while (original_playlist.size() != 0) {
// 播放歌单中的第一首歌
int cur = original_playlist.front();
final_playlist.push_back(cur);
// 播放后将其从歌单中移除
original_playlist.pop();
// 当前第一首歌移到最后一首
cur = original_playlist.front();
original_playlist.pop();
original_playlist.push(cur);
}
return final_playlist;
}
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;
}