25. K 个一组翻转链表

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;
    }
}

你可能感兴趣的:(LeetCode,LeetCode,25.,K,个一组翻转链表,Reverse,Nodes)