题目.png

解决思路

直接求出链表长度，除以k得到每个链表长度d，同样求出余数a，即前a个元素长度为d+1，遍历链表分割即可。

/**
 * Definition for singly-linked list.
 * type ListNode struct {
 *     Val int
 *     Next *ListNode
 * }
 */
func splitListToParts(root *ListNode, k int) []*ListNode {
    count := 0
    for cur := root; cur != nil; cur = cur.Next {
        count ++
    }
    a := count / k
    b := count % k
    
    list := make([]*ListNode, 0, k)
    for cur, i := root, 0; i < k; i++ {
        length := a
        if b > 0 {
            length ++
            b --
        }
        var previous *ListNode = nil
        list = append(list, cur)
        for j := 0; j < length; j++{
            if j == length - 1 {
                previous = cur
            }
            cur = cur.Next
        }
        if previous != nil {
            previous.Next = nil
        }
    }
    return list
}