每日一题_725. 分隔链表
题目:
题意分析:
今天的每日一题,感觉也蛮友好的,题目的意思是有一个链表,然后给定一个是数字k,然后让我们把这个链表尽可能的平均分为k个部分,但可想而知,只有当链表的长度刚好整除k的时候,才能均分,这就是说当链表的长度不能整除k时,多出来的节点就得被均摊。
因此总共有k个子链表,其中 q u o t i e n t = n / k r e m a i n d e r = n % k quotient = n / k \quad \quad remainder = n \% k quotient=n/kremainder=n%k quotient代表如果不考虑剩余的remainder个节点时,k个链表中每个子链表的个数,因为还剩remainder个节点,将其平均分摊到前remainder个子链表中,按照这个思路,我们首先得遍历整个链表,得到节点的总的个数n,然后算出quotient和remainder,再第二次遍历链表,根据算好的个数,在合适的地方断开链表,得到最终结果。
class Solution {
public:
vector<ListNode*> splitListToParts(ListNode* head, int k) {
int n = 0;
vector<ListNode*> parts(k, nullptr);
ListNode *temp = head;
while (temp)
{
n ++;
temp = temp->next;
}
int quot = n / k, remain = n % k;
ListNode * curr = head;
for (int i = 0; i < k && curr != nullptr; i ++)
{
parts[i] = curr;
int partSize = quot + (i < remain ? 1 : 0);
for (int j = 1; j < partSize; j ++)
curr = curr->next;
ListNode * next = curr->next;
curr->next = nullptr;
curr = next;
}
return parts;
}
};