PAT1015 反转链表:记有关静态链表的编程题目注意的地方：检查并去除不在链表中的项

题目引例：

给定一个常数K以及一个单链表L，请编写程序将L中每K个结点反转。例如：给定L为1→2→3→4→5→6，K为3，则输出应该为
3→2→1→6→5→4；如果K为4，则输出应该为4→3→2→1→5→6，即最后不到K个元素不反转。

输入描述：

每个输入包含1个测试用例。每个测试用例第1行给出第1个结点的地址、结点总个数正整数N(<= 105)、以及正整数K(<=N)，即要求反转的
子链结点的个数。结点的地址是5位非负整数，NULL地址用-1表示。
接下来有N行，每行格式为：
Address Data Next
其中Address是结点地址，Data是该结点保存的整数数据，Next是下一结点的地址。

输出描述：

对每个测试用例，顺序输出反转后的链表，其上每个结点占一行，格式与输入相同。

关于本题：

难度系数2
主要麻烦在输出规范上
用数组的下标代表结点的地址，开一个1e6大小的数组存储结点，结点用结构体类型

具体代码：

#include <iostream>
#include <vector>
#include <algorithm>
#include <iomanip>
using namespace std;
const int maxn = 1e6;
struct Node{
    int data;
    int next;
};
Node node[maxn];
struct Ans{
    int ans_addr;
    int ans_data;
    int ans_next;
};
Ans ans;
vector<Ans> v;
int main()
{
    int start_addr,n,k,i,cur_addr,flag = 0;
    cin>>start_addr>>n>>k;
    for(i=0;i<n;i++)
    {
        cin>>cur_addr;
        cin>>node[cur_addr].data>>node[cur_addr].next;
    }
    int temp_start = start_addr;
    n=0;
    while(temp_start != -1)
    {
        n++;
        temp_start = node[temp_start].next;
    }
    while(n>=k)
    {
        flag = 1;
        v.clear();
        int m=k;
        while(m--)
        {
            ans.ans_addr = start_addr;
            ans.ans_data = node[start_addr].data;
            ans.ans_next = node[start_addr].next;
            v.push_back(ans);
            start_addr = ans.ans_next;
        }
        for(i=v.size()-1;i>0;i--)
        {
            v[i].ans_next = v[i-1].ans_addr;
        }
        v[0].ans_next = start_addr;
        for(i=v.size()-1;i>0;i--)
            cout<<setw(5)<<setfill('0')<<v[i].ans_addr<<" "<<v[i].ans_data<<" "<<setw(5)<<setfill('0')<<v[i].ans_next<<endl;
        n=n-k;
        if(n>=k)
        {
            cout<<setw(5)<<setfill('0')<<v[i].ans_addr<<" "<<v[i].ans_data<<" ";
            int temp = start_addr,p=k-1;
            while(p--)
                temp = node[temp].next;
            cout<<setw(5)<<setfill('0')<<temp<<endl;
        }
        else if(n<k&&n>0)
            cout<<setw(5)<<setfill('0')<<v[i].ans_addr<<" "<<v[i].ans_data<<" "<<setw(5)<<setfill('0')<<v[i].ans_next;
        else if(n==0)
            cout<<setw(5)<<setfill('0')<<v[i].ans_addr<<" "<<v[i].ans_data<<" -1";
    }
    if(n>0)
    {
        if(flag == 1)
            cout<<endl;
        while(start_addr!=-1)
        {
            cout<<setw(5)<<setfill('0')<<start_addr<<" "<<node[start_addr].data<<" ";
            if(node[start_addr].next!=-1)
                cout<<setw(5)<<setfill('0')<<node[start_addr].next<<endl;
            else
                cout<<"-1";
            start_addr = node[start_addr].next;
        }
    }
    return 0;
}

其中的这段代码：

int temp_start = start_addr;
    n=0;
    while(temp_start != -1)
    {
        n++;
        temp_start = node[temp_start].next;
    }

就是为了统计此链表中实际的结点个数，不这样做的话90%的测试用例都会出错。