//
// main.cpp
// PAT_1074. Reversing Linked List
//
// Created by wjq on 17/8/19.
// Copyright © 2017年 wjq. All rights reserved.
//
#include <iostream>
using namespace std;
int start,N,K;
int linklist[100005],num[100005];
struct Output
{
int from,weight,to;
}result[100005];
int idx=0;
int record[100005],rd=0;
int main(int argc, const char * argv[])
{
//输入
int from,weight,to;
scanf("%d%d%d",&start,&N,&K);
for(int i=0;i<N;i++)
{
scanf("%d%d%d",&from,&weight,&to);
linklist[from]=to;
num[from]=weight;
}
//遍历链表并保存每个节点,计算出有效链长度idx
int temp=start;
while(temp!=-1)
{
result[idx].weight=num[temp];
result[idx].from=temp;
temp=linklist[temp];
idx++;
}
//将翻转链表后的每个节点的nextPosition保存到record中
for(int i=0;i<idx/K;i++)
for(int j=K-1;j>=0;j--)
record[rd++]=result[i*K+j].from;
for(int i=(idx/K)*K;i<N;i++)
record[rd++]=result[i].from;
//输出
int inddd=1,outCount=0;
for(int i=0;i<idx/K;i++)
for(int j=K-1;j>=0;j--)
{
if(outCount!=N-1)
printf("%05d %d %05d\n",result[i*K+j].from,result[i*K+j].weight,record[inddd++]);
else
printf("%05d %d %d\n",result[i*K+j].from,result[i*K+j].weight,-1);
outCount++;
}
for(int i=(idx/K)*K;i<idx;i++)
{
if(i!=idx-1)
printf("%05d %d %05d\n",result[i].from,result[i].weight,record[inddd++]);
else
printf("%05d %d %d\n",result[i].from,result[i].weight,-1);
}
return 0;
}
自己写的代码思路不够清晰,看了网上的代码之后重写了一份.
//
// main.cpp
// PAT_1074. Reversing Linked List
//
// Created by wjq on 17/8/19.
// Copyright © 2017年 wjq. All rights reserved.
//
#include <iostream>
#include <algorithm>
using namespace std;
//node[i][0]存储数据,node[i][1]存储下一个节点地址,i表示起始地址
int node[100005][2],list[100005];
int main()
{
//输入
int st,N,K,from,w,to;
scanf("%d%d%d",&st,&N,&K);
for(int i=0;i<N;i++)
{
scanf("%d%d%d",&from,&w,&to);
node[from][0]=w;
node[from][1]=to;
}
//将链表起始地址顺序存储在list数组中
int temp=st,idx=0;
while(temp!=-1)
{
list[idx++]=temp;
temp=node[temp][1];
}
//翻转链表
int i=0;
while(i+K<=idx)
{
reverse(list+i, list+i+K);
i+=K;
}
//输出
for(int i=0;i<idx-1;i++)
printf("%05d %d %05d\n",list[i],node[list[i]][0],list[i+1]);
printf("%05d %d -1\n",list[idx-1],node[list[idx-1]][0]);
return 0;
}