Leetcode 题解 [725.分隔链表](Java)

Leetcode 题解（每日打卡）

[725.分隔链表]

题目链接

题目描述

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

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

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

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

思路

​ 根据题目意思可知，要达到每部分的长度应该尽可能的相等：任意两部分的长度差距不能超过 1 这个要求，并根据题目所给的示例，我们可以猜想出这个题目大概的思路就是先计算出这个链表的长度(len)，然后再用 len/k 得到每个数组内放的元素的个数，然后再用 len%k 得到前多少个数组内放的元素的个数需要加1。

代码实现

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode() {}
 *     ListNode(int val) { this.val = val; }
 *     ListNode(int val, ListNode next) { this.val = val; this.next = next; }
 * }
 */
class Solution {
    public ListNode[] splitListToParts(ListNode head, int k) {
        ListNode p = head;
        ListNode r = p;
        ListNode[] a = new ListNode[k];
        int count = 0;
        while(p != null){
            count++;
            p = p.next;
        }
        p = head;
        int d = count/k; //每个数组内放的元素的个数
        int m = count%k; // 前多少个数组内放的元素的个数加1
        ListNode t ;
        for(int i=0;i<k;i++){
            t = p;
            count = 0;
            for(int j=0;j<d;j++){
                r = p;
                p = p.next;
            }
            if(m > 0){
                m--;
                r = p;
                p = p.next;
            }
            if(r != null) r.next = null;
            a[i] = t;
        }
        return a;
    }
}

时间复杂度: O(n)

空间复杂度: O(1)