1025 反转链表

#include<bits/stdc++.h>
using namespace std;

struct Node
{
	int data;//数据域 
	int next;//下一个节点地址 
	int before;//上一个节点地址 
	Node(int data0,int next0)
	{
		data=data0;
		next=next0;
	}
	Node(){
	}
};

unordered_map<int,Node> mp;//保存所有节点，键是节点的地址 

int main(void)
{
	int begin,n,k;
	scanf("%d%d%d",&begin,&n,&k);
	int add,data,next;
	for(int i=1;i<=n;i++)
	{
		scanf("%d%d%d",&add,&data,&next);
		mp[add]=Node(data,next);	
	}
	int before=-1;//第一个节点的前驱结点地址为-1 
	int nowAdd=begin;
	int cntNode=0;//cntNode是链表中的接电脑个数(注意坑点：题目给出的一些节点不在链表中 )， 
	while(nowAdd!=-1)
	{
		cntNode++;
		mp[nowAdd].before=before;
		before=nowAdd;
		nowAdd=mp[nowAdd].next;
	}//设置每个节点的before; 
	
	int nextBegin=begin;//nextBegin下一个子链的头结点地址 
	int pos=0;	//当前操作到了第几个节点 
	int newBegin=begin;//最终的链表的头结点地址 
	nowAdd=begin;
	int preEnd=-1;//上一个子链的尾节点的地址 
	while(cntNode-pos>=k)//剩下的节点个数大于等于k 
	{
		pos+=k;
		for(int i=1;i<k;i++)
		{
			nowAdd=mp[nowAdd].next;
		}	
		
		if(pos==k)
		{
			newBegin=nowAdd;//原链表的第一个子链的尾地址就是新链表的头地址 
		}	
		if(preEnd!=-1)//如果存在上一个子链 
		{
			mp[preEnd].next=nowAdd;
		}
		nextBegin=mp[nowAdd].next;//下一个子链的头地址 
		for(int i=1;i<k;i++)
		{
			mp[nowAdd].next=mp[nowAdd].before;
			nowAdd=mp[nowAdd].before;//反转当前子链 
		}
		preEnd=nowAdd; 
		nowAdd=nextBegin; 
	}
	
	if(preEnd!=-1)//将倒数第二个子链和最后一个子链 接起来 
		mp[preEnd].next=nextBegin;
	
	nowAdd=newBegin;//遍历链表 
	while(nowAdd!=-1)
	{
		printf("%05d %d ",nowAdd,mp[nowAdd].data);
		if(mp[nowAdd].next==-1)
		{
			printf("-1\n");
		}
		else
		{
			printf("%05d\n",mp[nowAdd].next);
		}
		nowAdd=mp[nowAdd].next;
	}
	return 0;
}