1025. 反转链表 (25)
时间限制
300 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作者
CHEN, Yue
给定一个常数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
开辟一个buf[100002] 的结构数组,将data和next 根据其address保存到buf[address]中,然后根据firstaddress将链表上的节点加载到vector中,
最后计算需要翻转的次数num = v.size()/k,用reverse(v.begin()+i*k, v.begin()+(i+1)*k)进行翻转,k从0到num。
#include<stdio.h>
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
struct node
{
int address;
int data;
int next;
}buf[100002];
vector<node> v;
int main()
{
freopen("D://input.txt", "r", stdin);
int firstAddress, n, k;
while(scanf("%d %d %d", &firstAddress, &n, &k) != EOF)
{
for(int i = 0; i < n; i ++)
{
int index;
scanf("%d", &index);
buf[index].address = index;
scanf("%d %d", &buf[index].data, &buf[index].next);
}
if(firstAddress == -1)
printf("-1\n");
else
{
int p = firstAddress;
while(p != -1)
{
v.push_back(buf[p]);
p = buf[p].next;
}
int countNode = v.size();
int num = countNode / k;//反转次数 因为k<=n 所以num>=1
for(int i = 0; i < num; i ++)
std::reverse(v.begin() + i*k, v.begin() + (i+1)*k);
for(int i = 0; i < countNode; i ++)
{
if((i+1) != countNode)
printf("%05d %d %05d\n", v[i].address, v[i].data, v[i+1].address);
else
printf("%05d %d -1\n", v[i].address, v[i].data);
}
}
}
return 0;
}