题目:
分析:要分成一段一段,首先要知道链表有多长,然后才能计算每一段的个数,而且前面的部分长度应大于等于后面的长度,后面部分可以出现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;
}
}