pat advanced 1074

来源:互联网 发布:王柳雯的淘宝店名 编辑:程序博客网 时间:2024/05/17 00:10

题目: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;}


0 0