LeetCode-092-反轉連結串列 II

雄獅虎豹發表於2021-10-22

反轉連結串列 II

題目描述:給你單連結串列的頭指標 head 和兩個整數 left 和 right ,其中 left <= right 。請你反轉從位置 left 到位置 right 的連結串列節點,返回 反轉後的連結串列 。

示例說明請見LeetCode官網。

來源:力扣(LeetCode)
連結:https://leetcode-cn.com/probl...
著作權歸領釦網路所有。商業轉載請聯絡官方授權,非商業轉載請註明出處。

解法一:利用棧

首先,如果head為null或者head只有一個節點,直接返回head

否則, 宣告一個新的頭節點newHead,宣告一個棧reverseNodes用來放leftright位置之間的節點(用於逆序),具體處理過程如下:

  • 遍歷head中的節點;
  • left之前的節點一次放入新連結串列中;
  • leftright之間的節點先放入棧reverseNodes中;
  • rightNode記錄right位置後節點的位置;
  • 最後,將棧reverseNodes中的節點一次放入新的連結串列中,然後將rightNode放到新連結串列的最後。

最後,返回newHead.next即為反轉後的連結串列。

import com.kaesar.leetcode.ListNode;

import java.util.Stack;

public class LeetCode_092 {
    public static ListNode reverseBetween(ListNode head, int left, int right) {
        if (head == null || head.next == null) {
            return head;
        }
        // 宣告一個新的頭節點
        ListNode newHead = new ListNode(-1);
        ListNode leftNode = newHead, rightNode = head;
        // 記錄是否已經走過left和right位置
        boolean findLeft = false, findRight = false;
        // 將left和right之間的節點放入棧中
        Stack<ListNode> reverseNodes = new Stack<>();
        int count = 1;
        while (head != null) {
            if (findLeft && findRight) {
                break;
            }
            if (findLeft) {
                if (count == right) {
                    reverseNodes.add(head);
                    rightNode = head.next;
                    break;
                } else {
                    reverseNodes.add(head);
                    head = head.next;
                }
            } else {
                if (count == left) {
                    findLeft = true;
                    reverseNodes.add(head);
                    if (count == right) {
                        rightNode = head.next;
                        findRight = true;
                        break;
                    }
                } else {
                    leftNode.next = head;
                    leftNode = leftNode.next;
                }
                head = head.next;
            }
            count++;
        }
        // 最後將棧中的節點逆序放入新的連結串列中
        while (!reverseNodes.isEmpty()) {
            leftNode.next = reverseNodes.pop();
            leftNode = leftNode.next;
        }
        leftNode.next = rightNode;
        return newHead.next;
    }

    public static void main(String[] args) {
        ListNode head = new ListNode(3);
        head.next = new ListNode(5);

        ListNode result = reverseBetween(head, 1, 2);
        while (result != null) {
            System.out.print(result.val + " ");
            result = result.next;
        }
    }
}
【每日寄語】 最初所擁有的只是夢想和毫無根據的自信而已,但是所有的一切都從這裡開始。

相關文章