PAT_1074. Reversing Linked List

//
//  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;
}