删除链表的节点(剑指Offer-18)

题面

给定单向链表的头指针和一个要删除的节点的值，定义一个函数删除该节点。

返回删除后的链表的头节点。

**注意：**此题对比原题有改动

示例

示例 1：

1输入: head = [4,5,1,9], val = 5
2输出: [4,1,9]
3解释: 给定你链表中值为 5 的第二个节点，那么在调用了你的函数之后，该链表应变为 4 -> 1 -> 9.

示例 2：

1输入: head = [4,5,1,9], val = 1
2输出: [4,5,9]
3解释: 给定你链表中值为 1 的第三个节点，那么在调用了你的函数之后，该链表应变为 4 -> 5 -> 9.

限制

  • 题目保证链表中节点的值互不相同
  • 若使用 C 或 C++ 语言，你不需要 free 或 delete 被删除的节点

思路

简简单单删除节点。

代码

 1/**
 2 * Definition for singly-linked list.
 3 * struct ListNode {
 4 *     int val;
 5 *     ListNode *next;
 6 *     ListNode(int x) : val(x), next(NULL) {}
 7 * };
 8 */
 9class Solution {
10public:
11    ListNode* deleteNode(ListNode* head, int val) {
12        if(head->val == val) return head->next;
13        ListNode *p = head;
14        while(p->next){
15            if(p->next->val == val){
16                p->next = p->next->next;
17                break;
18            }
19            p = p->next;
20        }
21        return head;
22    }
23};