剑指Offer-面试题18-删除链表的节点

2020-05-11

题目描述

给定单向链表的头指针和一个要删除的节点的值，定义一个函数删除该节点。
返回删除后的链表的头节点。
注意：此题对比原题有改动

示例1:

1
2
3

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

示例2:

1
2
3

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

说明：

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

思路

头节点单独处理：若删除的节点是头节点，则直接返回头节点的下一个节点
如果不是头节点，则遍历链表，找到要删除的节点删除，返回头节点即可。

代码

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode deleteNode(ListNode head, int val) {
        //判断链表是否为空
        if(head == null){
            return null;
        }
        //头节点单独处理
        if(head.val == val){
            head = head.next;
            return head;
        }
        //若不是头节点，则遍历链表
        ListNode node = head;
        while(node.next != null){
            //找到要删除的节点，删除后break
            if(node.next.val == val){
                node.next = node.next.next;
                break;
            }
            else{
                node = node.next;
            }
        }
        return head;
    }
}

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/shan-chu-lian-biao-de-jie-dian-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。