HJ48 从单向链表中删除指定值的节点【python3】

题目描述

输入一个单向链表和一个节点的值，从单向链表中删除等于该值的节点，删除后如果链表中无节点则返回空指针。

链表的值不能重复。

构造过程，例如输入一行数据为:
6 2 1 2 3 2 5 1 4 5 7 2 2
则第一个参数6表示输入总共6个节点，第二个参数2表示头节点值为2，剩下的2个一组表示第2个节点值后面插入第1个节点值，为以下表示:
1 2 表示为
2->1
链表为2->1

3 2表示为
2->3
链表为2->3->1

5 1表示为
1->5
链表为2->3->1->5

4 5表示为
5->4
链表为2->3->1->5->4

7 2表示为
2->7
链表为2->7->3->1->5->4

最后的链表的顺序为 2 7 3 1 5 4

最后一个参数为2，表示要删掉节点为2的值
删除 结点 2
则结果为 7 3 1 5 4

数据范围：链表长度满足 $1\le n\le 1000$ ，节点中的值满足 $0\le val\le 10000$

测试用例保证输入合法

输入描述

输入一行，有以下4个部分：
1 输入链表结点个数
2 输入头结点的值
3 按照格式插入各个结点
4 输入要删除的结点的值

输出描述

输出一行
输出删除结点后的序列，每个数后都要加空格

代码

题目并不能，在于理解题目意思。简单来说，设S为输入数据的列表形式，则S[1]就是头结点的值，S[2:-1]就是我们需要操作的部分。对每一对数值而言，要做的操作是，把值为前一个数的结点，插入到值为后一个数的结点之后。

class ListNode:
    def __init__(self, val:int):
        self.val = val
        self.next = None

# 根据题目构建单链表 将值为val1的节点插入到值为val2的节点后面
def create_list(head:ListNode,val1:int,val2:int):
    cur=head
    # 插入到头结点后面的情况：
    if cur.val==val2:
        tmp=ListNode(val1)
        tmp.next=cur.next
        cur.next=tmp    
    # 非头结点的情况
    while cur.val!=val2:
        cur=cur.next
        if cur.val==val2:
            tmp=ListNode(val1)
            tmp.next=cur.next
            cur.next=tmp
            break

def delete_value(head:List,k:int):
    cur=head
    # 找到值为k的结点的前一个结点
    while cur.next:
        if cur.next.val==k:
            break
        else:
            cur=cur.next
    # 链表内不存在值为k的结点
    if cur.next==None:
        return None
    else:
        cur.next=cur.next.next

while True:
    try:
        S=[int(x) for x in input().split()]
        n=S[0] # 链表结点个数
        headvalue=S[1] # 头结点值
        k=S[-1] # 要删除的结点值

        # 按照格式插入各个节点 S[2:2*n]是需要操作的部分
        head=ListNode(headvalue)
        for i in range(2,2*n,2):
            create_list(head,S[i],S[i+1])
 
        # 删除指定值的节点
        delete_value(head,k)
        # 输出操作后的链表
        cur=head
        while cur:
            print(cur.val,end=' ')
            cur=cur.next



    except:
        break