题目描述
给定一个链表,删除链表的倒数第n个节点并返回链表的头指针
例如,
给出的链表为:1->2->3->4->5, n= 2. 删除了链表的倒数第n个节点之后,链表变为1->2->3->5.
备注:
题目保证n一定是有效的
请给出请给出时间复杂度为\ O(n) O(n)的算法
解:
首先链表是单向的,所以想要直接删除可以反转后删除再反转回来,但是效率过低,所以我们采用数学方法。倒数第n个,也就是正序(length-n+1),所以我们只需要一次遍历统计次数即可完成length的获得。然后正序遍历到(length-n)个结点进行重指向即可完成。时间上是O(n),满足要求。这道题的n必定有效,所以倒是不用判断n和length的大小关系。
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) {
if(head==null){
return null;
}
//获取length
int len=0;
ListNode index=head;
while(index!=null){
len++;
index=index.next;
}
//重指向
if((len-n)==0){
head=head.next;
return head;
}
index=head;
for(int i=1;i<(len-n);i++){
//注意head从1开始,i次循环指向i+1结点
index=index.next;
}
index.next=index.next.next;
return head;
}
}