Leetcode(25) K个一组翻转链表

题目描述
给出一个链表,每 k 个节点一组进行翻转,并返回翻转后的链表。
k 是一个正整数,它的值小于或等于链表的长度。如果节点总数不是 k 的整数倍,那么将最后剩余节点保持原有顺序。
示例 :
给定这个链表:1->2->3->4->5
当 k = 2 时,应当返回: 2->1->4->3->5
当 k = 3 时,应当返回: 3->2->1->4->5

解题思路

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* reverseKGroup(ListNode* head, int k) {
        ListNode *L = head;
        ListNode *res = new ListNode(0);
        ListNode *pre = res;
        res->next = head;
        while(L)
        {
            int n = k;
            ListNode *inside = L;
            // 检查是否可以翻转
            while(n && inside)
            {
                inside = inside->next;
                n--;
            }
            if(n == 0)
            {
                //说明符合k哪么反转就OK了
                n = k;
                ListNode *p1 = inside;
                ListNode *p2 = L;
                ListNode *p3 = p2;
                while(n)
                {
                    p3 = p2->next;
                    p2->next = p1;
                    p1 = p2;
                    p2 = p3;
                    n --;
                }
                pre->next = p1;
                pre = L;
            }

            L = inside;
        }
        return res->next;        
    }
};

猜你喜欢

转载自blog.csdn.net/weixin_43624053/article/details/84953122
今日推荐