题解 | #链表内指定区间反转#

import java.util.*;

/*
 * public class ListNode {
 *   int val;
 *   ListNode next = null;
 *   public ListNode(int val) {
 *     this.val = val;
 *   }
 * }
 */

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     *
     * @param head ListNode类
     * @param m int整型
     * @param n int整型
     * @return ListNode类
     */
    public ListNode reverseBetween(ListNode head, int m, int n) {
        ListNode p = head;
        ListNode pre = null; // 当前节点的上一个节点
        ListNode m1 = null; // 反转区间的前一个节点
        ListNode reverseHead = null; // 反转区间反转后链表的头节点
        ListNode reverseTail = null; // 反转区间反转后，链表的尾节点

        int count = 0;
        while (p != null) {
            count++;
            ListNode next = p.next; // 下一个节点
            if (count >= m && count <= n) {
                if (count == m) {
                    reverseTail = p;
                }
                if (count == n) {
                    reverseHead = p;
                }
                p.next = pre;
            }
            // 记录反转区间的前一个节点
            if (count == m - 1) {
                m1 = p;
                p.next = reverseHead;
            }
            // 记录反转区间的后一个节点
            if (count == n + 1) {
                if (reverseTail != null) {
                    reverseTail.next = p;
                }
            }
            pre = p; // 更新前一个节点为当前节点
            p = next; // 指针往后移
            // n为最后一个节点的情况
            if (next == null && count == n) {
                if (reverseTail != null) {
                    reverseTail.next = null;
                }
            }
        }
        // m为第一个节点的情况
        if (m == 1) {
            head = reverseHead;
        }
        if (m1 != null) {
            m1.next = reverseHead;
        }
        return head;
    }
}