(java)Remove Nth Node From End of List

Given a linked list, remove the nth node from the end of list and return its head.

For example,

   Given linked list: 1->2->3->4->5, and n = 2.

   After removing the second node from the end, the linked list becomes 1->2->3->5.

Note:
Given n will always be valid.

Try to do this in one pass.

思路:设置两个指针p，q，和一个计数器count，count用来记录p移动的次数，当p移动的次数等于n时，q与p一起移动。当p移动到最后的时候，q就是倒数第n个位置的前一个位置，此时删除掉q后面的一个指针就行了

代码如下(已通过leetcode)

public class Solution {
public ListNode removeNthFromEnd(ListNode head, int n) {


ListNode p = head;
ListNode q = head;
int count = 0;
int length=0;
while (p != null) {
if (count == n + 1) {
p = p.next;
q = q.next;
} else {
p = p.next;
count++;
}
length++;
}

System.out.print(q.val);
if(length==1) return null;
if(length==n) head=head.next;
if (q.next.next == null)
q.next = null;
else
q.next = q.next.next;


return head;
}
}