给定一个链表,从列表末尾删除第n个节点并返回其头。
Input: 1->2->3->4->5, n=2
Output: 1->2->3->5
Method 1 将链表转为list删掉指定位置数字之后, 转为链表输出
1. 将链表中的数字存入list中
{l=[] ; while head: l.append(head.val)
head=head.next}
2.删除倒数第n个数字然后将list中的数字以链表形式按顺序输出
{del l[-n]; l=l[::-1]; res=None; for i in l: res=ListNode(i,res) ; return res}
也可以在将数字存入list中时用insert每次都添加到第一位,然后删除第n-1个数字, 不用逆序l, 直接输出, 这样省掉逆序步骤, 运行所占空间比之前的快。
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution:
def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
l=[]
while head:
l.insert(0,head.val)
head=head.next
del l[n-1]
res=None
for _ in l:
res=ListNode(_,res)
return res
时间复杂度O(2n-1), n个节点, 因为存储为list要遍历所有节点 , 输出为链表时有n-1个节点。
Method 2 直接对链表进行操作
1. 先要知道链表中有多少个节点
{temp=head; i=1; while temp.next: i+=1 temp=temp.next}
2. 建立一个res指针和head相同,如果得到的i和n相同,则证明删除的是第一个节点,此时直接返回res.next。遍历head中的每个节点,直到第n-i个位置时,跳过当前节点的下一个节点。因为要返回res所以要建立一个临时指针temp对res中的节点进行操作。
{res=head; if i==n: return res.next; temp=res; for _ in range(n-i-1):(因为从0开始) temp=temp.next; temp.next=temp.next.next; return res}
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution:
def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
res=temp=head
i=1
while temp.next:
i+=1
temp=temp.next
temp=res
if i==n:
return res.next
for _ in range(i-n-1):
temp=temp.next
temp.next=temp.next.next
return res
时间复杂度O(n), n个节点。
Java
- 先得出List长度length
- 设置一个指针将length-n的结点的下一位指向.next.next
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode() {}
* ListNode(int val) { this.val = val; }
* ListNode(int val, ListNode next) { this.val = val; this.next = next; }
* }
*/
class Solution {
public ListNode removeNthFromEnd(ListNode head, int n) {
if(head==null) return null;
int length=0;
ListNode temp=head;
while(temp!=null){//get List's length
temp=temp.next;
length++;
}
if(length==n) return head.next;
int index=length-n;
temp=head;
for(int i=1;i<index;i++){
temp=temp.next;
}
temp.next=temp.next.next;
return head;
时间复杂度O(2n)