PAT乙级1025 反转链表 (25 分)

Posted 2021-08-04 辉小歌

题目地址
最后一个测试点：不是所有的点都是合法的，有的结点地址是负数。

#include<cstdio>
#include<iostream>
#include<map>
#include<vector>
#include<stack>
using namespace std;
const int N=1e5+10; 
struct node
{
	int id;
	int val;
}Node[N];
map<int,int>mp,v;
int start,n,k;
stack<node> st;
int main(void)
{
	cin>>start>>n>>k;
	for(int i=0;i<n;i++)
	{
		int a,b,c; cin>>a>>b>>c;
		mp[a]=c,v[a]=b;
	}
	int index=0;
	while(1)
	{
		Node[index].id=start,Node[index].val=v[start];
		start=mp[start];
        if(Node[index].id>=0) index++;
		if(start==-1) break;
	}
    int start=0; 
	while(1&&k)
	{	
		if(start+k>index) break;
		for(int i=start;i<=start+k-1;i++) 
		{
			 st.push(Node[i]); 
		}
		for(int i=start;i<=start+k-1;i++)
		{
			Node[i]=st.top(); st.pop();
		}
		start+=k;
	}
	for(int i=0;i<index;i++)
	{
		printf("%05d %d ",Node[i].id,Node[i].val);
		if(i+1==index) printf("-1\\n");
		else printf("%05d\\n",Node[i+1].id);
	}
	return 0;
}

#include<cstdio>
#include<iostream>
#include<map>
#include<stack>
using namespace std;
const int N=1e5+10;
struct node
{
	int id;
	int val;
}Node[N];
map<int,int>mp,v;
int start,n,k;
int main(void)
{
	cin>>start>>n>>k;
	for(int i=0;i<n;i++)
	{
		int a,b,c; cin>>a>>b>>c;
		mp[a]=c,v[a]=b;
	}
	int index=0;
	while(1)
	{
		Node[index].id=start,Node[index].val=v[start];
		start=mp[start];
		if(Node[index].id>=0) index++;
		if(start==-1) break;
	}
	start=0;
	while(1)
	{
		if(start+k>index) break;
		stack<node>st;
		for(int i=start;i<start+k;i++) st.push(Node[i]);
		for(int i=start;i<start+k;i++) Node[i]=st.top(),st.pop();
		start=start+k;
	}
	for(int i=0;i<index;i++)
	{
		printf("%05d %d ",Node[i].id,Node[i].val);
		if(i+1==index) printf("-1\\n");
		else printf("%05d\\n",Node[i+1].id);
	}
	return 0;
}