LeetCode 328 | 奇偶链表

LeetCode 328 | 奇偶链表

题目

题解

1. 遍历链表，用两个向量表示两个子链表，然后合并，不是原地修改，花了O(n)的空间。

/**
 * 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:
    ListNode* oddEvenList(ListNode* head) {
        vector<ListNode*> v1;
        vector<ListNode*> v2;

        bool flag=1;
        while(head){
           if(flag){
                v1.push_back(head);
                head=head->next;
           }
            else{
                v2.push_back(head);
                head=head->next;
            }
            flag=!flag;
        }
        ListNode* s=new ListNode(-1);
        auto p=s;
        for(auto c:v1){
            p->next=c;
            p=p->next;
        }
        for(auto c:v2){
            p->next=c;
            p=p->next;
        }
        p->next=nullptr;

        return s->next;

    }
};

2. 原地修改,主要注意哑节点的使用

/**
 * 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:
    ListNode* oddEvenList(ListNode* head) {
        if(!head)return nullptr;
        //奇偶链表的头结点
        ListNode* s1 = new  ListNode(-1);
        ListNode* s2 = new  ListNode(-1);
        //两个工作指针
        auto p1=s1;
        auto p2=s2;

        bool flag=1;//辨别当前是奇还是偶结点
        while(head){
        	//将当前结点从原链表摘除，u指针并移动原链表的头指针
            auto tmp=head;
            head=head->next;
            tmp->next=nullptr;
            if(flag){//当前结点加入奇链表
                p1->next=tmp;
                p1=p1->next;
            }
            else{//...
                p2->next=tmp;
                p2=p2->next;
            }
            flag=!flag;
        }
        //奇偶链表缝合
        p1->next=s2->next;
        return s1->next;
    }
};