1. 题目原址
https://leetcode.com/problems/split-linked-list-in-parts/
2. 题目描述
3. 题目大意
给定一个链表和一个数字k,要将原链表分开,分为连续的k个部分。
4. 解题思路
- 首先要得到原链表的长度,以及计算每一部分要存储多少元素,和当前元素可不可以平均分到k个部分,如果不能,那么多出来的元素要放到变量remaining中,用于接下来下一次遍历
- 遍历链表,每一部分都通过计数看是否打到了存储数据的个数,如果打到了,就将当前链表截断。
5. AC代码
class Solution {
public ListNode[] splitListToParts(ListNode root, int k) {
ListNode temp = root;
int length = 0;
while(temp != null) {
temp = temp.next;
length ++;
}
int each = length / k; // 每一部分要存储多少数据
int remaining = length % k;
ListNode[] ret = new ListNode[k];
temp = root;
for(int i = 0; i < k && temp != null;i++) {
ret[i] = temp;
int total = each - 1 + (remaining -- > 0 ? 1 : 0);
ListNode prev = temp;
while (total -- >= 0 && temp != null) {
prev = temp;
temp = temp.next;
}
prev.next = null; // 截断了
}
return ret;
}
}
6. 相似题目
【1】 61. Rotate List 题目原址:https://leetcode.com/problems/rotate-list/
【2】 328. Odd Even Linked List 题目原址:https://leetcode.com/problems/odd-even-linked-list/