1025. 反转链表 (25)
时间限制
300 ms
内存限制
32000 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
#include <cstdio>
#include <stack>
#include <vector>
using namespace std;
typedef struct Node
{
int data;
int next;
};
int main()
{
Node node[100000];
stack<int> stk;
vector<int> vec;
int head, n, k;
int i, j, addr, data, next;
// Get input
scanf("%d %d %d", &head, &n, &k);
for(i = 0; i < n; i++)
{
scanf("%d %d %d", &addr, &data, &next);
node[addr].data = data;
node[addr].next = next;
}
// Get number of valid nodes
addr = head;
i = 0;
while(addr != -1)
{
i++;
addr = node[addr].next;
}
n = i;
// Reverse the total n/k groups of nodes, and put it into the vector
addr = head;
for(i = 0; i < n / k; i++)
{
for(j = 0; j < k; j++)
{
stk.push(addr);
addr = node[addr].next;
}
while(!stk.empty())
{
vec.push_back(stk.top());
stk.pop();
}
}
// Put the last nodes of less than k into the vector.
while(addr != -1)
{
vec.push_back(addr);
addr = node[addr].next;
}
// Print out the result.
for(i = 1; i <= vec.size(); i++)
{
if(i != vec.size())
printf("%05d %d %05d\n", vec[i - 1], node[vec[i - 1]].data, vec[i]);
else
printf("%05d %d -1", vec[i - 1], node[vec[i - 1]].data);
}
return 0;
}