每K个一组反转链表

链接：https://www.nowcoder.com/questionTerminal/a632ec91a4524773b8af8694a51109e7
来源：牛客网
 

给出一个链表，每 k 个节点一组进行翻转，并返回翻转后的链表。k 是一个正整数，它的值小于或等于链表的长度。如果节点总数不是 k 的整数倍，那么将最后剩余节点保持原有顺序。

说明：

1. 你需要自行定义链表结构，将输入的数据保存到你的链表中；

2. 你不能只是单纯的改变节点内部的值，而是需要实际的进行节点交换；

3. 你的算法只能使用常数的额外空间。

 

输入描述:

第一行输入是链表的值
第二行输入是K的值，K是大于或等于1的整数

输入形式为：
1 2 3 4 5
2

输出描述:

当 k = 2 时，应当输出:
2 1 4 3 5

当 k = 3 时，应当输出:
3 2 1 4 5

当k=6时，应当输出：
1 2 3 4 5

示例1

输入

1 2 3 4 5
2

输出 

2 1 4 3 5

 

#include<iostream>
#include<vector>
using namespace std;
int main()
{
    vector<int>a;
    int b;
    int k;
    while(cin>>b)
    {
        a.push_back(b);
        if(cin.get() == '\n')break;
    }
    cin>>k;
    if(k<=a.size())
    {
        int in = a.size();
        int fg = 0;
        for(int i = 0 ; i < a.size();)
        {
            if(i+k>a.size())
            {
                in = i;
                break;
            }
            for(int j = k-1; j >= 0; j--)
            {
                if(fg==0)
                {
                    cout<<a[i+j];
                    fg = 1;
                }else cout<<" "<<a[i+j];
            }
            i+=k;
        }
        if(in<a.size())
        {
            for(int i = in; i < a.size(); i++)cout<<" "<<a[i];
        }
    }else
    {
        for(int i = 0; i < a.size(); i++)
        {
            if(i==0)cout<<a[i];
            else cout<<" "<<a[i];
        }
    }
    cout<<endl;
    return 0;
}