单链表中将R结点插入到P结点之后，画出示意图，并写出关键代码

以下是单链表中将R结点插入到P结点之后的示意图:

     +---+    +---+    +---+    +---+    +---+    +---+
...  |   | -> |   | -> | P | -> |   | -> |   | -> |   |  ...
     +---+    +---+    +---+    +---+    +---+    +---+
                                          |
                                          V
                                       +---+
                                       | R |
                                       +---+

关键代码如下:

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

def insert_after_P(head, P, R):
    if not head:
        return head
    new_node = Node(R)
    cur = head
    while cur:
        if cur == P:
            new_node.next = cur.next
            cur.next = new_node
            break
        cur = cur.next
    return head

其中，head为单链表头结点，P为要插入R结点之后的结点，R为要插入的结点。函数的返回值为插入后的单链表头结点。