leetcode 725. 分隔链表 中等 链表

题目：

分析：要分成一段一段，首先要知道链表有多长，然后才能计算每一段的个数，而且前面的部分长度应大于等于后面的长度，后面部分可以出现null。也就是说如果遇到链表长度/组数 有余数的话，应该往前面的部分加并且每部分长度差不能超过1

代码：

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode[] splitListToParts(ListNode root, int k) {
        ListNode cur = root;
        int listLength = 0;
        while (cur != null) {
            cur = cur.next;
            listLength++;
        }

        int length = listLength / k;
        int remainder = listLength % k;

        ListNode[] result = new ListNode[k];
        cur = root;
        for (int i = 0; i < k; ++i) {
            ListNode head = cur;
            //如果有余数，那就前面部分每一部分加多1个，因为每部分长度差不能超过1，余数是几就加几个部分
            for (int j = 0; j < length + (i < remainder ? 1 : 0) - 1; ++j) {
                if (cur != null) cur = cur.next;
            }
            if (cur != null) {
                ListNode end = cur;
                cur = cur.next;
                end.next = null;
            }
            result[i] = head;
        }
        return result;
    }
}