[PAT解题报告] Reversing Linked List

简单题,每k个元素一组翻转链表。链表给的形式比较有意思,PAT的题链表都如此给。就是地址、元素值(一般是整数)、下一个元素的地址或者-1表示空,其实和指针真正的指针没什么区别的。
做法也比较直接,注意如果长度对k取余数非0,最后那几个是不动的。
(1)先计算链表长度n
(2)循环出最近k个元素,截断,得到一个长度恰好为k的链表,和剩余部分,注意维护好两部分的表头
(3) 翻转长度为k的链表
(4) 把(3)翻转的结果接到之前保存的结果上
(5) 转(2)
(6) 如果有“尾巴”,接到之前的结果上。
关键问题,每次连接注意可能当前结果就是第一段,之前可能是空。包括(6)的连接也是如此,因为链表可能根本就不足k个元素。

代码:
#include <cstdio>
#include <cstring>
#include <string>

using namespace std;

struct node {

int val;
int next;
};

node a[100005];
int reverse(node *a, int head) {
int answer = -1;
    while (head >= 0) {
        int next = a[head].next;
        a[head].next = answer;
        answer = head;
        head = next;
    }
    return answer;
}

int main() {
int n, head, k;
    scanf("%d%d%d",&head, &n, &k);
    for (int i = 0; i < n; ++i) {
        int ad;
        scanf("%d",&ad);
        scanf("%d%d",&a[ad].val,&a[ad].next);
    }
    int tail = -1, answer = -1;
    n = 0;
    for (int i = head; i >= 0; ++n, i = a[i].next)
    ;
    for (int i = 0, j ; n >= k;) {
        for (i = head, j = k - 1; j ;i = a[i].next, --j)
        ;
        n -= k;
        j = a[i].next;
        a[i].next = -1;
        int temp = reverse(a, head);
        head = j;
        if (tail < 0) {
            answer = tail =  temp;
        }
        else {
            a[tail].next = temp;
        }
        for (tail = temp; a[tail].next >= 0; tail = a[tail].next)
        ;
    }
    if (tail < 0) {
        answer = tail = head;
    }
    else {
        a[tail].next = head;
    }
    for (int i = answer; i >= 0; i = a[i].next) {
        printf("%05d %d ", i, a[i].val);
        if (a[i].next >= 0) {
            printf("%05d\n",a[i].next);
        }
        else {
            puts("-1");
        }
    }
    return 0;
}

原题链接: http://www.patest.cn/contests/pat-a-practise/1074
全部评论

相关推荐

不愿透露姓名的神秘牛友
11-21 19:05
面试官_我太想进步了:混学生会的,难怪简历这么水
点赞 评论 收藏
分享
头像
11-21 11:39
四川大学 Java
是红鸢啊:忘了还没结束,还有字节的5k 违约金
点赞 评论 收藏
分享
贺兰星辰:不要漏个人信息,除了简历模板不太好以外你这个个人简介是不是太夸大了...
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务