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

链表内指定区间反转

https://www.nowcoder.com/practice/b58434e200a648c589ca2063f1faf58c

/**
 * struct ListNode {
 *	int val;
 *	struct ListNode *next;
 *	ListNode(int x) : val(x), next(nullptr) {}
 * };
 */
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param head ListNode类 
     * @param m int整型 
     * @param n int整型 
     * @return ListNode类
     */
    ListNode* reverseBetween(ListNode* head, int m, int n) {
        // write code here
	  //链表为空,或链表只有一个节点,或者m==n
        if (head == nullptr || head->next == nullptr || m == n) return head;
	  //计算要反转几次
        int count = n - m;
	  //辅助指针p1,p2,p3,pbegin保存m位置的节点,pre保存它前一个节点
        ListNode* p1 = nullptr;
        ListNode* p2 = nullptr;
        ListNode* p3 = nullptr;
        ListNode* pre = nullptr;
        ListNode* pbegin = head;
	  //如果开始反转的位置不是表头,那么就要找到m位置的节点
        if (m != 1) {
            while (pbegin && --m) {
                pre = pbegin;
                pbegin = pbegin->next;
            }
        }
	  //p2就是m位置的节点
        p2 = pbegin;
        p3 = p2->next;
        while (count--&&p3) {
            p2->next = p1;
            p1 = p2;
            p2 = p3;
            p3 = p3->next;
        }
	  //此时,p2就是n位置的节点,p3是n位置的下一个节点
        p2->next = p1;
	  //m位置的节点下一个要指向p3(n位置的下一个节点)
        pbegin->next = p3;
	  //如果不是从第一个位置开始反转,那么pre的下一个要指向p2
        if (pre) pre->next = p2;
        return pre?head:p2;
    }
};

全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务