刷题日记-分隔链表

给你一个头结点为 head 的单链表和一个整数 k ,请你设计一个算法将链表分隔为 k 个连续的部分。

每部分的长度应该尽可能的相等:任意两部分的长度差距不能超过 1 。这可能会导致有些部分为 null 。

这 k 个部分应该按照在链表中出现的顺序排列,并且排在前面的部分的长度应该大于或等于排在后面的长度。

返回一个由上述 k 部分组成的数组。

示例 1:

输入:head = [1,2,3], k = 5
输出:[[1],[2],[3],[],[]]
解释:
第一个元素 output[0] 为 output[0].val = 1 ,output[0].next = null 。
最后一个元素 output[4] 为 null ,但它作为 ListNode 的字符串表示是 [] 。
示例 2:

输入:head = [1,2,3,4,5,6,7,8,9,10], k = 3
输出:[[1,2,3,4],[5,6,7],[8,9,10]]
解释:
输入被分成了几个连续的部分,并且每部分的长度相差不超过 1 。前面部分的长度大于等于后面部分的长度。

提示:

链表中节点的数目在范围 [0, 1000]
0 <= Node.val <= 1000
1 <= k <= 50
思路
其实比起一道链表题,本题更像是对除法和取余操作的应用的考察。
首先遍历一遍链表获得链表长度。经过分析可知,最终分割后,如果不能平分,会得到两种长度的链表,两种链表长度相差1。len对k取余的值是长链表的个数。k减去长链表的个数是短链表的个数。len除以k的商是短链表的长度。短链表的长度加一是长链表的长度。此外有一种特殊情况就是输入为空链表,此时只需要返回一个全为空指针的数组即可。
在得到上述数据后,可以遍历分割链表并将其放入数组,依照顺序先处理长链表再处理短链表
在处理短链表是需要注意,短链表的长度可能为0,这时需要额外处理
代码

public ListNode[] splitListToParts(ListNode head, int k) {
        if(head==null){
            ListNode result[] = new ListNode[k];
            for(int i=0;i<k;i++){
                result[i]=null;
            }
            return result;
        }
        ListNode cur = head;
        int len = 0;
        while(cur!=null){
            len++;
            cur = cur.next;
        }
        int small = len/k; //短的子链表有多少节点
        int large = small+1;//长的子链表有多少节点
        int largrNum = len%k;//长的子链表的个数
        int smallNum = k-largrNum;//短的子链表的个数
        ListNode result[] = new ListNode[k];
        cur = head;
        int i=0;
        for(;i<largrNum;i++){
            ListNode dummy = new ListNode(-1);
            dummy.next = cur;
            ListNode pre = dummy;
            ListNode curHead = cur;
            for(int j=0;j<large;j++){
                pre = pre.next;
                cur = cur.next;
            }
            pre.next = null;
            result[i]=curHead;
        }
        if (small==0){
            for(;i<k;i++){
                result[i]=null;
            }
            return result;
        }
        else{
            for(;i<k;i++){
            ListNode dummy = new ListNode(-1);
            dummy.next = cur;
            ListNode pre = dummy;
            ListNode curHead = cur;
            for(int j=0;j<small;j++){
                pre = pre.next;
                cur = cur.next;
            }
            pre.next = null;
            result[i]=curHead;
        }
        return result;
        }
        }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值