在已实现的单链表(Singly Linked List)基础上,实现一个复杂功能——单链表的分段

在已有的单链表结构基础上，实现链表的分段功能通常意味着我们需要将一个大链表分割成几个较小的、相互独立的部分，每个部分形成一个新的链表。这个过程可能会基于特定条件，比如节点值的范围、特定的索引位置或者是用户给定的长度等。下面是一个简单的Python示例，展示如何通过遍历链表并维护新的链表头节点来实现这一功能：

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

def segment_list(head, segment_size):
    # 如果链表为空或segment_size为0，直接返回None
    if not head or segment_size == 0:
        return None
    
    dummy = ListNode(0)  # 创建虚拟头节点
    dummy.next = head
    current = dummy
    segment_head = ListNode()  # 存储当前段的头部

    for _ in range(segment_size):  # 分割前几个元素
        if current.next is None:
            break
        current = current.next
    else:
        segment_head.next = dummy.next  # 将第一个段连接到新链表
        dummy.next = None  # 更新原链表的起点

    while current:
        # 拷贝当前段到新的链表，然后移动指针
        new_segment_head = ListNode(current.val)
        new_segment_head.next = segment_head.next
        segment_head.next = new_segment_head

        for _ in range(segment_size - 1):  # 继续处理下一个段，直到无法再分
            if current.next is None:
                break
            current = current.next
        else:
            # 如果还有剩余节点，更新当前段的结尾，并准备下一段
            segment_head.next = current.next
            current.next = None

    return segment_head.next  # 返回新的链表的头部

# 使用示例
head = ListNode(1, ListNode(2, ListNode(3, ListNode(4))))
result = segment_list(head, 2)  # 将链表分为每两个节点一组