链表内指定区间反转---C++实现

/*
将一个节点数为 size 链表 m 位置到 n 位置之间的区间反转，要求时间复杂度 O(n)，空间复杂度 O(1)。
例如：
给出的链表为 1→2→3→4→5→NULL, m=2,n=4
返回 1→4→3→2→5→NULL
数据范围：链表长度 0<size<=1000, 0<m<=n<=size
 */
 
ListNode *reverseBetween(ListNode *head, int m, int n) {
    if (head == NULL || head->next == NULL)
        return head;

    ListNode *q = NULL, *p = head;
    for (int i = 1; i <= m - 1; i++) {
        q = p;
        p = p->next;
    }

    // p此时指向第m个结点
    ListNode *end = p;
    ListNode *pPre = p, *nxt = NULL;
    p = p->next;

    // m->n之间的结点逆序
    for (int i = m + 1; i <= n; ++i) {
        nxt = p->next;
        p->next = pPre;
        pPre = p;
        p = nxt;
    }
    // p指向原链表中第n个结点的下一个结点
    // end表示逆序子链表的尾结点
    end->next = p;

    // pPre指向的是逆序后的子链表头
    // q指向的是第m个结点前一个结点 注意有可能是NULL
    if (q)
        q->next = pPre;  // 不是NULL 链接起来即可
    else
        head = pPre;    // 是NULL 头结点即为子链表头

    return head;
}