Leetcode【移除链表中所有值为key的元素】

题目描述:

删除链表中等于给定值 val 的所有节点.

测试用例:

输入:1->2->6->3->4->5->6, val = 6
输出: 1->2->3->4->5

思路:

- 判断参数合法性(单链表是不是为null)
- 定义一个prev=head表示要删除的元素的前驱节点;
- 定义一个cur=head.next代表要删除的元素;
- 若cur.data==key,则prev.next=cur.next;cur=cur.next;
- 否则prev=cur;cur=cur.next;
- 最终判断头结点的val是不是要删除的元素;

代码如下:

class ListNode{
    public int val;
    public ListNode next;

    public ListNode(int val) {
        this.val = val;
        this.next = null;
    }
}
public class Solution {
    public ListNode deleteDuplicates(ListNode head,int val) {
       //判断参数的合法性
        if (head == null){
            return null;
        }
        ListNode prev = head;
        ListNode cur = head.next;//代表当前要删除的节点
        while(cur != null){
            if (cur.val == val){
                prev.next = cur.next;
                cur = cur.next;
            }else {
                prev = cur;
                cur = cur.next;
            }
        }
        if (head.val == val){
            head = head.next;
        }
        return head;
    }
}