LeetCode第19题,中等难度,很经典的一道链表相关的题目。一个多月以前做的,当时一看就知道怎么写,结果代码总是差了点...
原题地址: https://leetcode-cn.com/problems/remove-nth-node-from-end-of-list/
题目描述:
给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
说明:
给定的 n 保证是有效的。
进阶:
你能尝试使用一趟扫描实现吗?
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/remove-nth-node-from-end-of-list
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路:
在根据自己写的代码写自己的解题思路时候,才发现这个题解居然是一遍扫描的。用两个指针,计算步差的方式,是最简单的。
中文官网题解:
https://leetcode-cn.com/problems/remove-nth-node-from-end-of-list/solution/
个人题解:
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
class Solution {
public ListNode removeNthFromEnd(ListNode head, int n) {
ListNode n1 = null;
ListNode n2 = head;
int step = 0;
while (n2.next != null) {
n2 = n2.next;
step++;
if (step >= n) {
if (n1 == null) {
n1 = head;
} else {
n1 = n1.next;
}
}
}
if (step > 0) {
if (n1 == null) {
return head.next;
} else {
n1.next = n1.next.next;
}
return head;
} else {
return null;
}
}
}
结果:
虽然最后通过了这题,但是解答错误了4次(思路正确但是代码错误),最后的结果还不错。