1025 反转链表
给定一个常数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(<= 10^5^)、以及正整数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
题解:这道题刚开始有两个测试点过不去一直纠结,看了好长时间,才发现是每k个都要反转,
也就是给定L为1→2→3→4→5→6,K为3,则输出应该为3→2→1→6→5→4,开始时候一直没有注
意到这个,导致有两个测试点过不去。思路的话,使用两个数组,一个数组来存放每个节点的地址,
一个数组来存放每个节点的下一个地址,再遍历一次链表得出节点个数(可能有节点不在其中),
同时将节点按照链表顺序存储在一个新数组里,每次对k个节点使用reverse来进行倒置,最后输出
即可。
知识点复习:
① reverse,用于反转容器中的元素,其两个参数分别为反转开始的位置和反转结束的位置
(如反转string a, 即为 reverse(a.begin(),a.end()),该函数存在于头文件#include<algorithm>下。
② printf("%05d",a);使用"%05d"可以使a按照5位输出,不足5位在前面补0。
③ memset(),有三个参数,分别是开始位置,将该范围内的数据赋的值(一般为0,用作数据初始化),长度大小。
该函数在#include<cstring>下,写法为 memset(a,0,N);
代码如下:
1 #include <iostream> 2 #include <algorithm> 3 using namespace std; 4 int main() { 5 int first, k, n, temp; 6 cin >> first >> n >> k; 7 int data[100005], next[100005], list[100005]; 8 for (int i = 0; i < n; i++) { 9 cin >> temp; 10 cin >> data[temp] >> next[temp]; 11 } 12 int sum = 0; 13 while (first != -1) { 14 list[sum++] = first; 15 first = next[first]; 16 } 17 for (int i = 0; i < (sum - sum % k); i += k) 18 reverse(begin(list) + i, begin(list) + i + k); 19 for (int i = 0; i < sum - 1; i++) 20 printf("%05d %d %05d\n", list[i], data[list[i]], list[i + 1]); 21 printf("%05d %d -1", list[sum - 1], data[list[sum - 1]]); 22 return 0; 23 }
下面是我刚开始理解错题意的错误代码,希望对大家有一定借鉴意义,或者也可以按照我这个思路自己写一个reverse
1 #include<iostream> 2 #include<cstring> 3 #define N 1000005 4 5 using namespace std; 6 7 int a[N], b[N]; 8 9 int main() 10 { 11 memset(a,0,N); 12 memset(b,0,N); 13 int head, n, k, address, data, next, number = 1, p, q, num = 1; 14 scanf("%d %d %d",&head, &n, &k); 15 while(n--){ 16 scanf("%d %d %d",&address, &data, &next); 17 a[address] = next; 18 b[address] = data; 19 } 20 21 address = head; 22 while(a[address]!= -1){ 23 address = a[address]; 24 num++; 25 } 26 if( num >= k){ 27 p = head; 28 q = a[head]; 29 while( number != k){ 30 a[p] = a[a[p]]; 31 a[q] = head; 32 head = q; 33 q = a[p]; 34 number++; 35 } 36 } 37 address = head; 38 while(a[address]!= -1){ 39 printf("%05d %d %05d\n",address, b[address], a[address]); 40 address = a[address]; 41 } 42 printf("%05d %d %d",address, b[address], a[address]); 43 return 0; 44 }