题解 | #调整牛群顺序#

调整牛群顺序

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

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 n int整型 
     * @return ListNode类
     */
    public ListNode moveNthToEnd (ListNode head, int n) {
        if (n == 1) {
            return head;
        }
        int sz = 1;
        ListNode end = head;
        // 1.找到尾节点,并得到链表长度sz
        while (end.next != null) {
            sz ++;
            end = end.next;
        }
        // n=正序第n个节点
        n = sz - n + 1;
        ListNode pre = head;
        ListNode cur = head;
        // val记录被删除节点的值
        int val = -1;
        while (cur != null) {
            n--;
            // 2.n=0,删除目标节点
            if (n == 0) {
                // 考虑删除头节点的情况
                if (cur == head) {
                    val = head.val;
                    head = head.next;
                } else {
                    val = cur.val;
                    pre.next = cur.next;
                }
                break;
            }
            pre = cur;
            cur = cur.next;
        }
        // 3.新增节点放置链表尾部
        end.next = new ListNode(val);
        return head;
    }
}

本题考查的依旧是链表的遍历和删除,我的思路是大致可以分为三步:

1.找到链表的尾节点 end 和 长度 sz(直接遍历即可得到),从而得出顺序的 n = sz - n + 1

2.从头遍历链表,删除第 n 个节点,并记录该节点的 val,跳出循环

3.新增值为 val 的新节点,放置链表的尾部

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务