1025 反转链表 (25)(25 分)
给定一个常数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是下一结点的地址。
输出格式:
对每个测试用例,顺序输出反转后的链表,其上每个结点占一行,格式与输入相同。
这个题有两个测试点没过第一次提交,倒数第二个测试点超时,最后一个错误,超时让我重新写了一遍,错误就是后来还会有这个类似于 -1 9 00500 和 00500 8 12345这种节点,很明显这种节点是无效的,做题的时候没有想到这一点让我思考了很长时间,无效节点只加一个的话有可能在程序上没错误,多加几个就出现错误,就是这样,昨天晚上躺床上想的,今天中午起来一改就A了,所以做题遇到窘境的时候就该静下心来好好思考一下,到底是哪里错了。
#include <iostream> #include <string> #include <algorithm> #include <cstdio> #include <cstdlib> #include <cstring> #include <vector> #include <cmath> #include <list> #include <deque> #define max 100005 using namespace std; struct node{ int frontad; int data; int backad; }; struct Data { int data; int next; }; int main() { int firstad; int k; int n,s,cur; node temp; Data D[max]; int count1=0; vector<node> in; vector<node> ou; scanf("%d %d %d",&firstad,&n,&k); for(int i=0;i<n;i++) { cin>>temp.frontad>>temp.data>>temp.backad; in.push_back(temp); if(temp.frontad==-1) { continue; } D[temp.frontad].data=temp.data; D[temp.frontad].next=temp.backad; } cur=firstad; while(1) { node semp; semp.frontad=cur; semp.data=D[cur].data; semp.backad=D[cur].next; ou.push_back(semp); if(D[cur].next==-1) break; cur=D[cur].next; } for(s=0;s<ou.size()/k;s++) { reverse(ou.begin()+s*k,ou.begin()+(s+1)*k); } for(int i=0;i<ou.size();i++){ if(i==ou.size()-1){ printf("%05d %d -1\n",ou[i].frontad,ou[i].data); } else{ printf("%05d %d %05d\n",ou[i].frontad,ou[i].data,ou[i+1].frontad ); } } return 0; }