pat advanced 1074

题目：http://pat.zju.edu.cn/contests/pat-a-practise/1074

题目就是坑，一直卡在最后一个case上。最后一个case有这样的检测，所有节点的数量N并不是单链表的长度，也就是说单链表的长度可能小于N，举个栗子，

00100 6 4

00000 4 99999

00100 1 12309

68237 6 -1

33218 3 00000

99999 5 68237

12309 2 33218

一共六个节点，单链表长度也是6，但是下面的栗子：

00100 6 4

00000 4 99999

00100 1 12309

68237 6 00001

33218 3 00000

99999 5 -1

12309 2 33218

一共六个节点，单链表长度却是5.

同时注意链表长度为1的情况，笔者做了测试，oj的数据没有测试链表长度为1的情况。

参考代码：

#include<cstdio>
#include<map>
#include<vector>
using namespace std;

int main()
{
    int N, K, start, t_start, t_value, t_end;
    map<int, int> addrmap, valuemap, rev_valuemap;
    vector<int> vec;
    scanf("%d %d %d", &start, &N, &K);
    for(int i = 0; i < N; ++i)
    {
        scanf("%d %d %d", &t_start, &t_value, &t_end);
        addrmap[t_start] = t_end;
        valuemap[t_start] = t_value;
        rev_valuemap[t_value] = t_start;
    }
    if(N==1)
    {
        printf("%05d %d -1\n", t_start, t_value);
        return 0;
    }
    while(start != -1)
    {
        vec.push_back(valuemap[start]);
        start = addrmap[start];
    }
    int times = vec.size() / K;
    for(int i = 0; i < times; ++i)
    {
        for(int j = K-1; j >= 0; --j)
        {
            if(i==0 && j==K-1)
            {
                printf("%05d %d ", rev_valuemap[vec.at(i*K+j)], vec.at(i*K+j));
            }
            else
            {
                printf("%05d\n", rev_valuemap[vec.at(i*K+j)]);
                printf("%05d %d ", rev_valuemap[vec.at(i*K+j)], vec.at(i*K+j));
            }
        }
    }
    if(vec.size() % K)
    {
        int begin = vec.size() - vec.size() % K;
        while(begin != vec.size())
        {
            printf("%05d\n", rev_valuemap[vec.at(begin)]);
            printf("%05d %d ", rev_valuemap[vec.at(begin)], vec.at(begin));
            ++begin;
        }
    }
    printf("-1\n");
    return 0;
}