【PAT甲级 链表处理】1074 Reversing Linked List (25 分)

乙级也有这题，以前也写过，以前的写法更简单……
【PAT乙级、C++】1025 反转链表 (25分)
这次用了双指针和栈……

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

struct Node{
    
    
    int data;
    int next;
}node[100010];

int start, N, K; 

int main() {
    
    
    cin >> start >> N >> K;
    for(int i = 0;i < N;++i){
    
    
        int now;
        cin >> now;
        cin >> node[now].data >> node[now].next;
    }
    int p = start, q = start; 
    stack<int> S;  // 反转的临时栈
    vector<int> V; // 存放结果的顺序表
    while(true) {
    
    
        int i;
        for(i = 0;i < K && q != -1;++i) // q比p快K步
            q = node[q].next;
        if(q == -1 && i != K){
    
     // 如果q走到了末尾，而且q走到末尾的步数不是等于K（就像N=6 K=3的情况）
            while(p != q){
    
     // 就将最后面的一段原样拼接
                V.push_back(p);
                p = node[p].next;
            }
            break;
        }
        while(p != q) {
    
     // 将要反转的部分的地址入栈
            S.push(p);
            p = node[p].next;
        }
        while(!S.empty()){
    
     // 然后出栈得到的就是反转的链表
            V.push_back(S.top());
            S.pop();
        }
    }

    // 输出
    for(int i = 0;i < V.size();++i){
    
    
        int now = V[i];
        if(i != V.size() - 1){
    
    
            int next = V[i+1];
            printf("%05d %d %05d\n", now, node[now].data, next);
        } else {
    
    
            printf("%05d %d %d\n", now, node[now].data, -1);
        }
    }

    return 0;
}