标签:pre cur 22 2021.9 pos next 链表 长度
725. 分隔链表
给你一个头结点为
head
的单链表和一个整数k
,请你设计一个算法将链表分隔为 k 个连续的部分。每部分的长度应该尽可能的相等:任意两部分的长度差距不能超过 1 。这可能会导致有些部分为 null 。
这 k 个部分应该按照在链表中出现的顺序排列,并且排在前面的部分的长度应该大于或等于排在后面的长度。
返回一个由上述 k 部分组成的数组。
题目链接: 725. 分隔链表 - 力扣(LeetCode) (leetcode-cn.com)
解题思路
- 首先计算已给单链表的长度,才能计算出每个部分的平均长度
len // k
得到每个部分的平均长度,len % k
得到多余的长度- 对多余长度部分的处理,给前几个部分每个部分加一个节点,直到将多余的部分用完
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution:
def splitListToParts(self, head: ListNode, k: int) -> List[ListNode]:
# 添加哑结点方便处理
dumb = ListNode(0,head)
cur = dumb
# 记录整个单链表的长度
length = 0
while cur.next:
length += 1
cur = cur.next
div = length // k
mod = length % k
# res用于返回最后的分隔结果
res = []
# 记录当前分配到哪一个部分,共需分配k个部分,pos的取值为0~k-1
pos = 0
pre = dumb
cur = head
while pos < k:
res.append(cur)
# 记录当前段落分配到了哪个节点
temp_pos = 0
while temp_pos < div:
pre = pre.next
cur = cur.next
temp_pos += 1
# 多分配一个节点
if pos < mod:
pre = pre.next
cur = cur.next
pos += 1
# 下面这行代码是将单链表在pre指针这边断开,没有这行代码每一个部分都将保存到原链表直到结尾
pre.next = None
pre = ListNode(0,cur)
return res
如果没有第37行代码,可以看到结果如下所示。
标签:pre,cur,22,2021.9,pos,next,链表,长度 来源: https://www.cnblogs.com/waitting975/p/15319340.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。