删除链表的倒数第n个节点

删除链表的倒数第n个节点

题目描述

给定一个链表，删除链表的倒数第n个节点并返回链表的头指针
例如，

给出的链表为:1->2->3->4->5, n= 2.
删除了链表的倒数第n个节点之后,链表变为1->2->3->5.

备注：
题目保证n一定是有效的
请给出请给出时间复杂度为 O(n)\ O(n) O(n)的算法

代码

import java.util.*;

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

public class Solution {
    /**
     * 
     * @param head ListNode类 
     * @param n int整型 
     * @return ListNode类
     */
    public ListNode removeNthFromEnd (ListNode head, int n) {
        // write code here
        //if(head==null) return head;
        
        //这样定义会在只有一个输入示例的时候出错,因为会略过第一个
//         ListNode tmp = head;
//         ListNode p1 = head;
//         ListNode p2 = head;
        ListNode tmp = new ListNode(0);
        tmp.next = head;
        ListNode p1 = tmp;
        ListNode p2 = tmp;
        
        for(int i = 0;i < n;i++){
            p1 = p1.next;
        }
        
        while(p1.next!=null){
            p1 = p1.next;
            p2 = p2.next;
        }
        p2.next = p2.next.next;
        
        return tmp.next;
        
    }
}