25. K 个一组翻转链表

最新推荐文章于 2024-02-05 11:29:19 发布

网忆小叶

最新推荐文章于 2024-02-05 11:29:19 发布

阅读量145

点赞数

分类专栏： LeetCode 文章标签： LeetCode 25. K 个一组翻转链表 Reverse Nodes

原文链接：https://blog.csdn.net/qq_42103091/article/details/105113700

版权

LeetCode 专栏收录该内容

27 篇文章 1 订阅

订阅专栏

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.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode reverseKGroup(ListNode head, int k) {

        ListNode resHead = null;
        ListNode resCursor = null;

        Stack<ListNode> stack = new Stack<>();

        ListNode cursor = head;
        while (cursor != null) {

            for (int i = 0; i < k; i++) {

                if (cursor == null) {
                    if (resHead == null) {
                        resHead = head;
                    }
                    return resHead;
                }

                stack.push(cursor);
                cursor = cursor.next;
            }

            while (!stack.isEmpty()) {

                ListNode curNode = stack.pop();

                if (resHead == null) {
                    resHead = curNode;
                }

                if (resCursor == null) {
                    resCursor = curNode;
                } else {
                    resCursor.next = curNode;
                    resCursor = resCursor.next;
                }
            }

            if (resCursor == null) {
                return head;
            } else {
                resCursor.next = cursor;
            }
        }

        return resHead;
    }
}