算法: 删除链表的倒数第 N 个结点。
做了几道链表题发现一般都需要设一个预指针(pre),让它的下一个指针指向head,就是pre.next = head。这道题用到了双指针方法。快慢指针,先让快指针走n步,然后让慢指针和快指针一起走,直到快指针走到最后一个节点,那么慢指针所在节点的下一个节点就是要删除的节点。
示例:
输入:head = [1,2,3,4,5], n = 2
输出:[1,2,3,5]
代码如下:
/**
* 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) {
ListNode pre = new ListNode(0);
pre.next = head;
ListNode start = pre,end = pre;
while(n != 0){
start = start.next;
n--;
}
while(start.next != null){
start = start.next;
end = end.next;
}
end.next = end.next.next;
return pre.next;
}
}
之所以返回的是pre.next而不是head,是因为head也可能是被删除的节点。.
下面是评论区大佬的递归方法解出来的,可以参考借鉴。
class Solution {
public ListNode removeNthFromEnd(ListNode head, int n) {
return removeNode(head,n)==n?head.next:head;
}
public int removeNode(ListNode node,int n) {
if(node.next == null) return 1;
int m = removeNode(node.next, n);
if(m == n)
if(m == 1) node.next = null;
else node.next = node.next.next;
return m+1;
}
}
前天做的另一道链表题也是用了预指针,有些相同之处。
算法: 两数相加
给你两个 非空 的链表,表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的,并且每个节点只能存储 一位 数字。
请你将两个数相加,并以相同形式返回一个表示和的链表。
你可以假设除了数字 0 之外,这两个数都不会以 0 开头。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/add-two-numbers
示例:
输入:l1 = [2,4,3], l2 = [5,6,4]
输出:[7,0,8]
解释:342 + 465 = 807.
/**
* 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 addTwoNumbers(ListNode l1, ListNode l2) {
ListNode pre = new ListNode(0);
ListNode cur = pre;
int carry = 0;
while(l1 != null || l2 != null){
int x = l1 == null?0:l1.val;
int y = l2 == null?0:l2.val;
int sum = x + y + carry;
carry = sum / 10;
sum = sum %10;
cur.next = new ListNode(sum);
cur = cur.next;
if(l1 != null){
l1 = l1.next;
}
if(l2 != null){
l2 = l2.next;
}
}
if(carry==1){
cur.next = new ListNode(1);
}
return pre.next;
}
}