码迷,mamicode.com
首页 > 其他好文 > 详细

【LeetCode】25. K 个一组翻转链表

时间:2020-06-21 17:48:12      阅读:38      评论:0      收藏:0      [点我收藏+]

标签:lang   problems   ref   展开   总数   lock   使用   lin   public   

链接:

https://leetcode-cn.com/problems/reverse-nodes-in-k-group

描述:

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

示例:
给你这个链表:1->2->3->4->5
当 k = 2 时,应当返回: 2->1->4->3->5
当 k = 3 时,应当返回: 3->2->1->4->5

说明:
你的算法只能使用常数的额外空间。
你不能只是单纯的改变节点内部的值,而是需要实际进行节点交换。

ListNode* reverseKGroup(ListNode* head, int k) {}

思路:

C++

展开后查看
/**
 * 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* dummyHead = new ListNode(-1);
        dummyHead->next = head;
        ListNode* pre = dummyHead;
        while(pre->next != NULL){
            ListNode* start = pre->next;
            ListNode* end = pre;
            for(int i = 0; i < k; i++){
                end = end->next;
                if(end == NULL){
                    return dummyHead->next;
                }
            }
            ListNode* next = end->next;
            end->next = NULL;
            pre->next = reverse(start);
            start->next = next;
            pre= start;
        }
        return dummyHead->next;
    }
    ListNode* reverse(ListNode* head){
        ListNode* pre = NULL;
        ListNode* p = head;
        while(p != NULL){
            ListNode* next = p->next;
            p->next = pre;
            pre = p;
            p = next;
        }
        return pre;
    }
};

Java

展开后查看
/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode reverseKGroup(ListNode head, int k) {
        ListNode dummyHead = new ListNode(-1);
        dummyHead.next = head;
        ListNode pre = dummyHead;
        while(pre.next != null){
            ListNode start = pre.next;
            ListNode end = pre;
            for(int i = 0; i < k; i++){
                end = end.next;
                if(end == null){
                    return dummyHead.next;
                }
            }
            ListNode next = end.next;
            end.next = null;
            pre.next = reverse(start);
            start.next = next;
            pre = start;
        }
        return dummyHead.next;
    }
    ListNode reverse(ListNode head){
        ListNode pre = null;
        ListNode p = head;
        while(p != null){
            ListNode next = p.next;
            p.next = pre;
            pre = p;
            p = next;
        }
        return pre;
    }
}

【LeetCode】25. K 个一组翻转链表

标签:lang   problems   ref   展开   总数   lock   使用   lin   public   

原文地址:https://www.cnblogs.com/crazyBlogs/p/13173169.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!