Leetcode 725. Split Linked List in Parts

本题自己想的时候, 连每条链表的结点个数都不明白, 这个其实就是这题的思路并没有那么难 : 首先通过给定的 K 可以确定链表的数量, 然后通过除法和取余操作可以确定每条链表的长度, 因为有余数肯定是每条链表平分的话分不到一个结点, 所以从第一个结点开始, 每个结点分一个即可, 分完余数就没有了.

class Solution {
    public ListNode[] splitListToParts(ListNode root, int k) {
        ListNode cur = root;
        int length = 0;
        // 求链表长度
        while(cur != null) {
            length++;
            cur = cur.next;
        }
        // 每条链表的基本长度
        int size = length / k;
        // 余数
        int mod = length % k;
        cur = root;
        ListNode[] ret = new ListNode[k];
        // cur != null 这个条件要加上
        for(int i = 0; cur != null && i < k; i++) {
            ret[i] = cur;
            // 当前链表的实际长度
            int curSize = size + (mod-- > 0 ? 1 : 0);
            // 这里是 curSize - 1, 因为已经给链表第一个结点赋值了
            for(int j = 0; j < curSize - 1; j++) {
                cur = cur.next;
            }
            // 将当前链表从给定的原链表中断开
            ListNode next = cur.next;
            cur.next = null;
            cur = next;
        }
        return ret;
    }
}