Day3 LeetCode 203. 移除链表元素 c&python

203题目：

给你一个链表的头节点 head 和一个整数 val ，请你删除链表中所有满足 Node.val == val 的节点，并返回 新的头节点 。

示例 1：

输入：head = [1,2,6,3,4,5,6], val = 6
输出：[1,2,3,4,5]

示例 2：

输入：head = [], val = 1
输出：[]

 思路：选择使用虚拟头节点来解题，这样可以避免判断是否要删除头节点

C语言

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */
struct ListNode* removeElements(struct ListNode* head, int val)
{

    struct ListNode *dummy_head;
    dummy_head=(struct ListNode*)malloc(sizeof(struct ListNode));
    dummy_head->next=head;
    struct ListNode *cur=dummy_head;
    while(cur->next!=NULL)
    {
        if(cur->next->val==val)
        {
            cur->next=cur->next->next;
        }
        else
        {
            cur=cur->next;
        }
    }
    head=dummy_head->next;
    free(dummy_head);
    return head;
}

python语言

# Definition for singly-linked list.
# class ListNode(object):
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution(object):
    def removeElements(self, head, val):
        """
        :type head: ListNode
        :type val: int
        :rtype: ListNode
        """
        dummy_head=ListNode(next=head)
        current=dummy_head
        while current.next:
            if current.next.val==val:
                current.next=current.next.next
            else:
                current=current.next
        return dummy_head.next
        

注意事项：C语言是使用指针进行一系列操作。