725. 分隔链表
给定一个头结点为 root 的链表, 编写一个函数以将链表分隔为 k 个连续的部分。
每部分的长度应该尽可能的相等: 任意两部分的长度差距不能超过 1,也就是说可能有些部分为 null。
这k个部分应该按照在链表中出现的顺序进行输出,并且排在前面的部分的长度应该大于或等于后面的长度。
返回一个符合上述规则的链表的列表。
举例: 1->2->3->4, k = 5 // 5 结果 [ [1], [2], [3], [4], null ]
解题思路:
做链表的题,我有一个感觉,或者说本办法,即将被移动的或者摘取的节点,先摘除下来,然后在进行移动操作,怎么说呢,我们分析链表题型,无外乎让我们对链表进行拆分,或者说对链表的节点进行挪动,这里总结之,即都涉及到对一个或者一系列的节点进行操作,那么我们可以先忽略对这些节点后续进行如何操作,先把节点从链表上摘取下来呈一个游离状态,并将原始的链表缺出来的地方进行弥补,即处理好next指针,是置为null呢还是置为下下个节点。
LeetCode中链表的题有对单个节点进行挪动的,也有对一系列节点进行挪动的,单个节点挪动,没有什么特殊的trick,多节点挪动往往会涉及到一些trick,比如328. 奇偶链表,这题要求将奇数点放在一起,偶数点放在一起,如果我们直接在遍历链表的过程中对奇数和偶数节点进行移动,不是说不好做,只能说非常容易出错,其实我们可以换个思路,我们将原始链表上的节点单独拎出来,访问到奇数节点,将奇数节点摘除放到奇数链表上,偶数节点同理,在遍历完原始链表后,将这两个链表直接拼接在一起即可。即这里提到一个trick,不在原始链表上移动节点,而是将重新构建一个链表,将原始链表上的节点挪动到这个链表上。
此题的思路: 按照上面基本思路,解此题只需要注意一点,即保留一个rest变量,将rest一份一份分配到前置链表上。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
vector<ListNode*> splitListToParts(ListNode* root, int k) {
// if (!root) return {};
int len = 0;
ListNode *cur = root;
while (cur) {
cur = cur->next;
++len;
}
int rest = len % k, per = len / k;
vector<ListNode*> res(k);
cur = root;
for (int i = 0; i < k; ++i) {
if (!cur) break;
ListNode *newNode = cur;
res[i] = newNode;
int cnt = per - 1;
if (rest != 0) {
--rest;
cnt = per;
}
for (int j = 0; j < cnt; ++j) {
cur = cur->next;
}
ListNode* next = cur->next;
cur->next = NULL;
cur = next;
}
return res;
}
};
相关题
写在后面
链表题,摘节点,补缺口,移节点,插入节点。