【List-medium】725. Split Linked List in Parts 给定一个链表和一个数字k,要将原链表分开,分为连续的k个部分。

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/

猜你喜欢

转载自blog.csdn.net/xiaojie_570/article/details/92616980
今日推荐