LeetCode----チェーンテーブルの最後からN番目のノードを削除

3006 ワード

チェーンテーブルが与えられ、チェーンテーブルの最後からn番目のノードが削除され、チェーンテーブルのヘッダノードが返される.
例:
チェーンテーブルが与えられる:1->2->3->4->5、およびn=2.最後から2番目のノードを削除すると、チェーンテーブルは1->2->3->5になります.
説明:
与えられたn保証は有効である.
ステップ:
スキャンを使って実現してみてもいいですか?
public class RemoveNthFromEnd {

    @Test
    public void removeNthFromEndTest() {
        Assert.assertNull(removeNthFromEnd(new ListNode(1), 1));
        ListNode listNode = new ListNode(1);
        listNode.next = new ListNode(2);
        Assert.assertEquals(2, removeNthFromEnd(listNode, 2).val);
    }

    @Test
    public void removeNthFromEndTest1() {
        ListNode listNode = new ListNode(1);
        listNode.next = new ListNode(2);
        Assert.assertEquals(1, removeNthFromEnd(listNode, 1).val);
    }
    public ListNode removeNthFromEnd(ListNode head, int n) {
        ListNode temp = head;
        Stack stack = new Stack<>();
        while (head != null) {
            stack.push(head);
            head = head.next;
        }

        for (int i = 0; i < n-1; i++) {
            stack.pop();
        }

        ListNode targetNode = stack.pop();
        if (stack.size() == 0) {
            temp = targetNode.next;
        } else if(n == 1) {
            ListNode lastNode = stack.pop();
            lastNode.next = null;
        } else {
            targetNode.val = targetNode.next.val;
            targetNode.next = targetNode.next.next;
        }
        return temp;
    }

}