题目引例:
给定一个常数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是下一结点的地址。
输出描述:
对每个测试用例,顺序输出反转后的链表,其上每个结点占一行,格式与输入相同。
关于本题:
难度系数2
主要麻烦在输出规范上
用数组的下标代表结点的地址,开一个1e6大小的数组存储结点,结点用结构体类型
具体代码:
#include <iostream>
#include <vector>
#include <algorithm>
#include <iomanip>
using namespace std;
const int maxn = 1e6;
struct Node{
int data;
int next;
};
Node node[maxn];
struct Ans{
int ans_addr;
int ans_data;
int ans_next;
};
Ans ans;
vector<Ans> v;
int main()
{
int start_addr,n,k,i,cur_addr,flag = 0;
cin>>start_addr>>n>>k;
for(i=0;i<n;i++)
{
cin>>cur_addr;
cin>>node[cur_addr].data>>node[cur_addr].next;
}
int temp_start = start_addr;
n=0;
while(temp_start != -1)
{
n++;
temp_start = node[temp_start].next;
}
while(n>=k)
{
flag = 1;
v.clear();
int m=k;
while(m--)
{
ans.ans_addr = start_addr;
ans.ans_data = node[start_addr].data;
ans.ans_next = node[start_addr].next;
v.push_back(ans);
start_addr = ans.ans_next;
}
for(i=v.size()-1;i>0;i--)
{
v[i].ans_next = v[i-1].ans_addr;
}
v[0].ans_next = start_addr;
for(i=v.size()-1;i>0;i--)
cout<<setw(5)<<setfill('0')<<v[i].ans_addr<<" "<<v[i].ans_data<<" "<<setw(5)<<setfill('0')<<v[i].ans_next<<endl;
n=n-k;
if(n>=k)
{
cout<<setw(5)<<setfill('0')<<v[i].ans_addr<<" "<<v[i].ans_data<<" ";
int temp = start_addr,p=k-1;
while(p--)
temp = node[temp].next;
cout<<setw(5)<<setfill('0')<<temp<<endl;
}
else if(n<k&&n>0)
cout<<setw(5)<<setfill('0')<<v[i].ans_addr<<" "<<v[i].ans_data<<" "<<setw(5)<<setfill('0')<<v[i].ans_next;
else if(n==0)
cout<<setw(5)<<setfill('0')<<v[i].ans_addr<<" "<<v[i].ans_data<<" -1";
}
if(n>0)
{
if(flag == 1)
cout<<endl;
while(start_addr!=-1)
{
cout<<setw(5)<<setfill('0')<<start_addr<<" "<<node[start_addr].data<<" ";
if(node[start_addr].next!=-1)
cout<<setw(5)<<setfill('0')<<node[start_addr].next<<endl;
else
cout<<"-1";
start_addr = node[start_addr].next;
}
}
return 0;
}
其中的这段代码:
int temp_start = start_addr;
n=0;
while(temp_start != -1)
{
n++;
temp_start = node[temp_start].next;
}
就是为了统计此链表中实际的结点个数,不这样做的话90%的测试用例都会出错。