题目描述:
牛客网——洗牌
洗牌在生活中十分常见,现在需要写一个程序模拟洗牌的过程。
现在需要洗2n张牌,从上到下依次是第1张,第2张,第3张一直到第2n张。首先,我们把这2n张牌分成两堆,左手拿着第1张到第n张(上半堆),右手拿着第n+1张到第2n张(下半堆)。接着就开始洗牌的过程,先放下右手的最后一张牌,再放下左手的最后一张牌,接着放下右手的倒数第二张牌,再放下左手的倒数第二张牌,直到最后放下左手的第一张牌。接着把牌合并起来就可以了。
例如有6张牌,最开始牌的序列是1,2,3,4,5,6。首先分成两组,左手拿着1,2,3;右手拿着4,5,6。在洗牌过程中按顺序放下了6,3,5,2,4,1。把这六张牌再次合成一组牌之后,我们按照从上往下的顺序看这组牌,就变成了序列1,4,2,5,3,6。
现在给出一个原始牌组,请输出这副牌洗牌k次之后从上往下的序列。
#include <iostream>
#include <vector>
using namespace std;
int main()
{
int T, n, k;
cin >> T;
// 循环组数
while(T--)
{
cin >> n >> k;
vector<int> vec(2 * n);
for(int i = 0; i < 2 * n; i++)
cin >> vec[i];
// 循环洗牌次数
while(k--)
{
vector<int> tmp(vec.begin(), vec.end());
for(int i = 0; i < n; i++)
{
// 插入左手的牌
vec[2 * i] = tmp[i];
// 插入右手的牌
vec[2 * i + 1] = tmp[i + n];
}
}
for(int i = 0; i < 2 * n; i++)
cout << vec[i] << " ";
cout << endl;
}
return 0;
}