其他分享
首页 > 其他分享> > 链表内指定区间反转——牛客网

链表内指定区间反转——牛客网

作者:互联网

描述

将一个节点数为 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,链表中每个节点的值满足 |val| <= 1000
要求:时间复杂度 O(n),空间复杂度 O(n)
进阶:时间复杂度 O(n),空间复杂度 O(1)

示例1

输入:
{1,2,3,4,5},2,4
返回值:
{1,4,3,2,5}

示例2

输入:
{5},1,1
返回值:
{5}

解题思路

第一次遍历:首先遍历节点,用一个变量cnt来维护当前节点的位置,如果已经遍历到的节点位置在反转区间左边界的前一个,即跳出循环(也可能反转就从第一个节点开始,需要用一个p指针去维护)

第二次遍历,从左边界的下一个节点开始继续遍历,每次遍历都将当前节点进行反转作为头节点,然后指针q向后移动一位,重复上述过程,直到cnt > n(反转区间右边界)则跳出循环

最后判断,若p不为空,则将p.next = newHead,并返回原头节点head,反之,则直接返回反转后链表的头节点即可)

AC代码

import java.util.*;

/*
 * public class ListNode {
 *   int val;
 *   ListNode next = null;
 * }
 */

public class Solution {
    /**
     * 
     * @param head ListNode类 
     * @param m int整型 
     * @param n int整型 
     * @return ListNode类
     */
    public ListNode reverseBetween (ListNode head, int m, int n) {
        // write code here
        if (head == null || head.next == null) {
            return head;
        }
        
        // 用于记录当前已经遍历到的节点位置
        int cur = 0;
        ListNode p = null;
        
        if (cur != m - 1) {
            p = head;
            // 找到开始位置
            while (p != null) {
                cur++;
                if (cur == m - 1) {
                    break;
                }
                p = p.next;
            }   
        }
        cur++;
        
        ListNode newHead = p != null ? p.next : head;
        if (newHead.next == null) {
            return head;
        }
        
        ListNode tail = newHead;
        ListNode q = tail.next;
        while (q != null) {
            cur++;
            if (cur > n) {
                // 后续不用反转
                break;
            }
            ListNode tmp = q.next;
            q.next = newHead;
            newHead = q;
            tail.next = tmp;
            q = tmp;
        }
        
        // 前后衔接
        if (p != null) {
            p.next = newHead;
            return head;
        }
        
        return newHead;
    }
}

原题链接:https://www.nowcoder.com/practice/b58434e200a648c589ca2063f1faf58c?tpId=295&tqId=654&ru=%2Fpractice%2Fb58434e200a648c589ca2063f1faf58c&qru=%2Fta%2Fformat-top101%2Fquestion-ranking&sourceUrl=%2Fexam%2Foj

标签:head,ListNode,反转,next,链表,牛客,null,newHead,节点
来源: https://www.cnblogs.com/pengsay/p/16321453.html