PAT 乙级 1025 反转链表

335 阅读3分钟

「这是我参与2022首次更文挑战的第7天,活动详情查看:2022首次更文挑战」。

题目描述

给定一个常数 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

思路分析

首先定义静态链表再初始化,由首地址begin遍历整条链表,然后再记录有效节点个数再排序,最后输出链表,由于是每k个节点翻转一次,因此可以把n个节点分为n/k个完整的块,如果n%k不等于0,则最后会出现不完整的块,不完整和完整的块都需要分情况讨论,详情请看注释

AC代码

include<bits/stdc++.h>
using namespace std;
const int maxn=100010; 
struct Node
{
	int address,data,next;
	int order;  //结点在链表中的地址,无效地址为maxn 
}node[maxn]; 

bool cmp(Node a,Node b)
{
	return a.order<b.order;
}

int main()
{
	for(int i=0;i<maxn;i++)  //初始化order,若有无效结点则可以区分开来 
		node[i].order=maxn;
	
	int begin,n,k,address;
	cin>>begin>>n>>k;
	while(n--)
	{  //输入
		cin>>address;
		cin>>node[address].data>>node[address].next;
	    node[address].address=address;
	}
	
	int p=begin,count=0;
	while(p!=-1)
	{  //找出有效结点 
		node[p].order=count++;
		p=node[p].next; 
	} 
	
	sort(node,node+maxn,cmp);  //对所有结点进行排序
	n=count;  //为方便仍使用n
	  
	//链表已形成,准备反转输出 
	for(int i=0;i<n/k;i++)
	{  //枚举n/k块 
		for(int j=(i+1)*k-1;j>i*k;j--) //第i完整块从后往前输出 
			printf("%05d %d %05d\n",node[j].address,node[j].data,node[j-1].address);
		printf("%05d %d ",node[i*k].address,node[i*k].data); //以下为每块最后一个结点的输出
		if(i<n/k-1)  //若不是最后一个完整块 
			printf("%05d\n",node[(i+2)*k-1].address); 
		else
		{  //最后一个完整快 
			if(n%k==0)  //而且还是最后一个结点 
				printf("-1\n");
			else
			{  //不是最后一个结点,即有不完整的块 
				printf("%05d\n",node[(i+1)*k].address);
				for(int i=n/k*k;i<n;i++)
				{
					printf("%05d %d ",node[i].address,node[i].data);
					if(i<n-1) printf("%05d\n",node[i].next);
					else printf("-1\n");
				} 
			} 
		} 
	} 
	return 0;
}

总结

这个题主要考察链表的相关知识,日后仍需巩固😁