搜索
您的当前位置:首页正文

leetcode25 k个一组反转链表

来源:独旅网

题目
给你一个链表,每 k 个节点一组进行翻转,请你返回翻转后的链表。
k 是一个正整数,它的值小于或等于链表的长度。
如果节点总数不是 k 的整数倍,那么请将最后剩余的节点保持原有顺序。

思路
和反转链表是一个系列的题目,首先写出反转1-k个节点的函数reverseK,返回反转后的头结点。然后根据题意,先数出k各节点,进行翻转,此时原头结点的next指向后续剩余部分K个一组反转的头结点;若不足k个节点则直接返回。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* reverseK(ListNode* head, int k) {
        if(k==1 || head==nullptr || head->next == nullptr)
            return head;
        ListNode* p = reverseK(head->next, k-1);
        head->next->next = head;
        head->next = nullptr;
        return p;
    }
    ListNode* reverseKGroup(ListNode* head, int k) {
        int cnt = k;
        ListNode* p = head, * new_head;
        while(cnt && p)
        {
            cnt--;
            p=p->next;
        }
        if(cnt == 0)
        {
            p = reverseKGroup(p, k);
            new_head = reverseK(head, k);
            head->next = p;
            return new_head;
        }
        else
            return head;
    }
};

因篇幅问题不能全部显示,请点此查看更多更全内容

Top