LeetCode 25 Reverse Nodes in k-Group

论坛 期权论坛 编程之家     
选择匿名的用户   2021-6-2 16:24   525   0

心血来潮参加了一次比赛结果签到之后人没了,继续康复训练吧= =

做了一段时间的题,也看了一段时间别人更优的解法,思路很自然的变清晰了。

题意:给出一个链表的head和一个数值k,要求将链表里面每k个连续的node进行翻转,函数返回值为新的head。

思路:大体框架突出一个递归调用思想,既然每次返回新的head,就没有必要用循环结构来让指针指指指最后还指飞了,特判条件也很多。

于是考虑每k个点处理一次,先求next_head = reverseKGroup(now->next, k),再处理好当前的k个点,将当前的head的next指向next_head并返回翻转后的new_head即可。

本来对于这个k个点也想要用递归来做的,但是考虑到优化空间复杂度到O(1)级别,直接用循环处理。

链表的处理无外乎记录一下cur和next两个点,不过由于这次需要将指针方向调转,需要一个中间变量tmp来记录next的位置,以免无法找到next->next。

对于一般情况而言,指针的记录状态如上图所示,但是注意cur->next这个指针,我在这里的做法是不对它进行维护,而是对于每一次的(cur,next)对,以next为中心来进行处理。这是因为对于边界情况,也就是head而言,其next指针需要指向的是next_head而不是之前与之紧邻的上一个点。

那么如上图中红色指针所示,对于循环过程中的每一个(cur,next)对,只需要进行红色指针的维护即可。

代码如下:

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* reverseKGroup(ListNode* head, int k) {
        if (k == 1)
            return head;//boundary
        int sz = 1;
        ListNode* new_head = head;
        while(new_head != NULL){
            ++sz;
            new_head = new_head->next;
            if (sz == k)
                break;
        }
        if (new_head == NULL)
            return head;
        ListNode* new_next = reverseKGroup(new_head->next, k);
        ListNode* now = head, *tmp_next, *next;
        next = now->next;
        while(--sz){
            tmp_next = next->next;//temporary variant
            next->next = now;//reverse for *next
            now = next;//maintain the value of *cur
            next = tmp_next;//go to NEXT loop
        }
        head->next = new_next;//finished
        return new_head;
    }
};

1A了,爽到。

分享到 :
0 人收藏
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

积分:3875789
帖子:775174
精华:0
期权论坛 期权论坛
发布
内容

下载期权论坛手机APP