给定一个常数K以及一个单链表L,请编写程序将L中每K个结点反转。例如:给定L为1→2→3→4→5→6,K为3,则输出应该为3→2→1→6→5→4;如果K为4,则输出应该为4→3→2→1→5→6,即最后不到K个元素不反转。
输入格式:
每个输入包含1个测试用例。每个测试用例第1行给出第1个结点的地址、结点总个数正整数N(<= 105)、以及正整数K(<=N),即要求反转的子链结点的个数。结点的地址是5位非负整数,NULL地址用-1表示。
接下来有N行,每行格式为:
Address Data Next
其中Address是结点地址,Data是该结点保存的整数数据,Next是下一结点的地址。
输出格式:
对每个测试用例,顺序输出反转后的链表,其上每个结点占一行,格式与输入相同。
输入样例:00100 6 4 00000 4 99999 00100 1 12309 68237 6 -1 33218 3 00000 99999 5 68237 12309 2 33218输出样例:
00000 4 33218 33218 3 12309 12309 2 00100 00100 1 99999 99999 5 68237 68237 6 -1
这个题刚开始时最后一人case过不了,最后才发现是没有考虑到输入的N个节点,有某个或某些不在链表上,所以最后要输出的总的节点数小于N。加一个计数器就可以了。
代码如下:
#include<iostream>
#include<vector>
#include<algorithm>
#include<iterator>
#include<utility>
using namespace std;
typedef pair<int, int> P;
vector<P> coll(100000);
struct node
{
int adress;
int data;
int next;
node(int a, int b, int c) :adress(a), data(b), next(c){};
};
vector<node> vec;
int main()
{
#ifdef ONLINE_JUDGE
#else
freopen("D:\\in.txt", "r", stdin);
freopen("D:\\out.txt", "w", stdout);
#endif
int adress, data, next;
int n, k, first;
while (scanf("%d%d%d", &first, &n, &k) != EOF)
{
int num = n;
while (n--)
{
scanf("%d%d%d", &adress, &data, &next);
P p;
p.first = data;
p.second = next;
coll[adress] = p;
}
int index = first;
int cnt(0);//在链表上的结点个数
while (index != -1)
{
cnt++;
vec.push_back(node(index, coll[index].first, coll[index].second));
index = coll[index].second;
}
int t = cnt / k;
for (int i = 0; i < t; i++)
std::reverse(vec.begin() + i*k, vec.begin() + (i + 1)*k);
for (int i = 0; i < vec.size()-1;i++)
{
printf("%05d %d %05d\n", vec[i].adress, vec[i].data, vec[i + 1].adress);
}
printf("%05d %d %d\n", vec[vec.size() - 1].adress, vec[vec.size() - 1].data, -1);
}
return 0;
}