标签：head 分隔 int nullptr next 链表 725 ListNode curr

725. 分隔链表

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    vector<ListNode*> splitListToParts(ListNode* head, int k) {
        vector<ListNode*> ans;
        ListNode* tnode = head;
        int n = 0;
        while(tnode!= nullptr)
        {
            n++;
            tnode = tnode -> next;
        }
        if(n == 0){
            for(int i=0;i<k;i++){
                ans.emplace_back(nullptr);
            }
            return ans;
        }
        int n1 =  n / k;
        int n2 = n % k;     
        for(int i =0;i<k;i++){
            int n3 = n1;
            if( i < n2) 
                n3 ++;
            ListNode* newnode = new ListNode();
            ListNode* t_ans = newnode;
            for(int j=0; j <n3&&head!=nullptr;j++){
                ListNode* node = new ListNode(head->val);
                newnode->next = node;
                newnode = newnode -> next;
                head = head->next;
            }
            ans.emplace_back(t_ans->next);
        }
        return ans;
    }
};

官方代码如下，可以看出有些优化

• 在创建结果数组时，就将初始值设置成nullptr，

• 循环时记着把节点不为空加进去

官方代码

class Solution {
public:
    vector<ListNode*> splitListToParts(ListNode* head, int k) {
        int n = 0;
        ListNode *temp = head;
        while (temp != nullptr) {
            n++;
            temp = temp->next;
        }
        int quotient = n / k, remainder = n % k;

        vector<ListNode*> parts(k,nullptr);
        ListNode *curr = head;
        for (int i = 0; i < k && curr != nullptr; i++) {
            parts[i] = curr;
            int partSize = quotient + (i < remainder ? 1 : 0);
            for (int j = 1; j < partSize; j++) {
                curr = curr->next;
            }
            ListNode *next = curr->next;
            curr->next = nullptr;
            curr = next;
        }
        return parts;
    }
};

标签：head,分隔,int,nullptr,next,链表,725,ListNode,curr
来源： https://blog.csdn.net/qq_40967961/article/details/120408603

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。