分隔链表--双指针操作

0x01.问题

给定一个链表和一个特定值 x,对链表进行分隔,使得所有小于 x 的节点都在大于或等于 x 的节点之前。
你应当保留两个分区中每个节点的初始相对位置。
输入示例: head = 1->4->3->2->5->2, x = 3
输出示例:1->2->2->4->3->5

C++结构体:
/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
 C++函数形式:   ListNode* partition(ListNode* head, int x) 

0x02.简要分析

要保证前面的元素都比x小,后面的元素都比x大,且相对的位置不变,我们只需要使用两个指针,小的元素往第一个里面添加,大的往第二个里面添加,最后再合并起来就可以了。

需要注意的地方:

  • head节点遍历完后,需要对第二个指针的next初始化,因为这将是返回链表的尾部。
  • 用哑节点处理两个链表。
  • 接的时候也要注意接的是哑节点的next

0x03.解决代码

class Solution {
public:
    ListNode* partition(ListNode* head, int x) {
        ListNode*minx=new ListNode(0);
        ListNode*maxx=new ListNode(0);
        ListNode*before=minx;
        ListNode*after=maxx;
        while(head){
            if(head->val<x){
                before->next=head;
                before=before->next;
            }
            else{
                after->next=head;
                after=after->next;
            }
            head=head->next;
        }
        after->next=NULL;
        before->next=maxx->next;
        return minx->next;
    }
};

ATFWUS --Writing By 2020–03–24

发布了159 篇原创文章 · 获赞 179 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/ATFWUS/article/details/105064189
今日推荐