“开启掘金成长之旅!这是我参与「掘金日新计划 · 2 月更文挑战」的第 3 天,点击查看活动详情”
描述
给定单向链表的头指针和一个要删除的节点的值,定义一个函数删除该节点。返回删除后的链表的头节点。
1.此题对比原题有改动
2.题目保证链表中节点的值互不相同
3.该题只会输出返回的链表和结果做对比,所以若使用 C 或 C++ 语言,你不需要 free 或 delete 被删除的节点
数据范围:
0<=链表节点值<=10000
0<=链表长度<=10000
示例1
输入: {2,5,1,9},5
返回值: {2,1,9}
说明:给定你链表中值为 5 的第二个节点,
那么在调用了你的函数之后,该链表应变为 2 -> 1 -> 9
示例2
输入:{2,5,1,9},1
返回值:{2,5,9}
说明:给定你链表中值为 1 的第三个节点,
那么在调用了你的函数之后,该链表应变为 2 -> 5 -> 9
思路: 迭代遍历
我们要删除给定的一个元素,那我们首先肯定要找到这个元素,然后考虑怎么去删除它。
删除一个链表节点,肯定是断掉它的前一个节点指向它的指针,然后指向它的后一个节点,即越过了需要删除的这个节点。
断开连接 pre.next = cur.next;
步骤:
- step 1:首先我们加入一个头部节点,方便于如果可能的话删除掉第一个元素。
- step 2:准备两个指针遍历链表,一个指针指向当前要遍历的元素,另一个指针指向该元素的前序节点,便于获取它的指针。
- step 3:遍历链表,找到目标节点,则断开连接,指向后一个。
- step 4:返回时去掉我们加入的头节点。
图示:
c++代码
class Solution {
public:
ListNode* deleteNode(ListNode* head, int val) {
//加入一个头节点
ListNode* res = new ListNode(0);
res->next = head;
//前序节点
ListNode* pre = res;
//当前节点
ListNode* cur = head;
//遍历链表
while (cur != NULL) {
//找到目标节点
if (cur->val == val) {
//断开连接
pre->next = cur->next;
break;
}
pre = cur;
cur = cur->next;
}
//返回去掉头节点
return res->next;
}
};
js代码
* @param head ListNode类
* @param val int整型
* @return ListNode类
*/
function deleteNode(head, val) {
let curNode = head;
if (curNode.val === val) return curNode.next;
let nextNode = curNode.next;
while (nextNode !== null) {
if (nextNode.val === val) {
curNode.next = nextNode.next;
break;
}
curNode = nextNode;
nextNode = nextNode.next;
}
return head;
}
module.exports = {
deleteNode
};