牛客网做题笔记--洗牌问题

题目描述：

洗牌在生活中十分常见，现在需要写一个程序模拟洗牌的过程。 现在需要洗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次之后从上往下的序列。

输入描述:

第一行一个数T(T ≤ 100)，表示数据组数。对于每组数据，第一行两个数n,k(1 ≤ n,k ≤ 100)，
接下来一行有2n个数a1,a2,...,a2n(1 ≤ ai ≤ 1000000000)。表示原始牌组从上到下的序列。

输出描述:

对于每组数据，输出一行，最终的序列。数字之间用空格隔开，不要在行末输出多余的空格。

输入例子:

3
3 1
1 2 3 4 5 6
3 2
1 2 3 4 5 6
2 2
1 1 1 1

输出例子:

1 4 2 5 3 6
1 5 4 3 2 6
1 1 1 1

题目分析：

我们主要需要执行的操作就是洗牌这个过程，将洗完的牌返回以供继续洗牌，所以我们可以构造一个洗牌函数，根据传进来的参数给定其洗多少次，返回最后的那副牌即可。洗牌函数较简单，所以这道题目不难。

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
vector<int> sort(vector<int>& res);
int main() {
        int time;
        int n,k;
        cin>>time;
        while(time){
                cin>>n>>k;
                vector<int> res(2*n);
                for(unsigned int i = 0;i < 2*n;++i){
                        cin >> res[i];
                }
                while(k){
                        res = sort(res);
                        k--;
                }
                for(int i = 0;i < res.size();i++){
                        if(i != res.size()-1)
                                cout<<res[i]<<" ";
                        else
                                cout<<res[i];
                }
                cout<<endl;
        time--;
    }
    return 0;
}
vector<int> sort(vector<int>& res){
        int size = res.size();
        int i = size/2;
        int j = 1;
        vector<int> r;
        while(i){
                r.push_back(res[size - j]);
                r.push_back(res[(size/2) - j]);
                j++;
                i--;
        }
        for(int i = 0;i < (r.size()/2);i++){
            swap(r[i],r[r.size()-i-1]);
        }
        return r;
}

洗牌函数就是这么一个过程：

1. 将传进来的牌分成两部分（0～i == i～size），因为给出的测试用例肯定是偶数，所以我们也不用担心奇数情况，降低了难度
2. 先将右边的牌的最后一张放入到新的牌中，再将左边的牌的最后一张放入，两边标记均往上跑一步
3. 这么重复，直到跳出循环，也就是没牌的时候了
4. 按照题目规则，将这些牌反序存储在数组中，就得到了新的一副牌

在主函数中，就是根据k的次数，执行k次洗牌过程，跳出k次洗牌这个过程之后，就将正确的结果输出即可。

注：但是这个题目在牛客上还没人用python做出过，我开始也是用python写，因为list很适合这些数组操作，但是最后发现不能按照它的要求输出一行，貌似python只有print函数。（如果有了解的，希望提醒一下博主，万分感谢。）

补充：

在别人代码上更改过的python：

import sys
def xipai(a,n):
    r=[a[i+j*n] for i in range(n) for j in range(2)]
    return r
while True:
    t = sys.stdin.readline().strip()
    if t == '':
        break
    t = int(t)
    if t > 100:
        break
    else:
        for i in range(t):
        s = sys.stdin.readline().strip()
        [n,k]=s.split()
        n=int(n)
        k=int(k)
        if n > 100 or k >100:
            break
        s = sys.stdin.readline().strip()
        if s == '':
            break
        a=s.split()
        r=a[:]
        if len(r) != 2*n:
            continue
        while k>0 and k<=100 and len(r) == (2*n):
            i+=1
            r=xipai(r, n)
            if r==a:
                k=k%i
            k-=1
            if len(r) == (2*n):
            for i in range(2*n-1):
                print int(r[i]),
            print r[2*n-1]

牛客上的这道题目貌似有点问题，貌似都是提示数组越界或者非法访问，在牛客的讨论区也有一些人遇到这种类似的问题，望以后解决。