86. 分隔链表

双指针法:

直觉
我们可以用两个指针pbig 和 psmall 来追踪上述的两个链表。两个指针可以用于分别创建两个链表,然后将这两个链表连接即可获得所需的链表。

算法
利用cur指针遍历原链表。若cur 指针指向的元素值 小于 x,该节点应当是 psmall 链表的一部分。因此我们将其移到 psmall 中。否则,该节点应当是pbig 链表的一部分。因此我们将其移到 pbig 中。

遍历完原有链表的全部元素之后,我们得到了两个链表 psmall 和 pbig。原有链表的元素或者在psmall 中或者在 pbig 中,这取决于它们的值。

现在,可以将 psmall 和 pbig 连接,组成所求的链表。

为了算法实现更容易,我们使用了哑结点初始化。不能让哑结点成为返回链表中的一部分,因此在组合两个链表时需要向前移动一个节点。

注意:pbig->next需要置空。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */


struct ListNode* partition(struct ListNode* head, int x){
    struct ListNode* cur = head;
    struct ListNode bigdummy;
    struct ListNode smalldummy;
    bigdummy.next = head;
    smalldummy.next = head;
    
    struct ListNode *psmall, *pbig;
    psmall = &smalldummy;
    pbig = &bigdummy;
    if(head == NULL || head->next == NULL)
        return head;
    while(cur != NULL){
        if(cur->val < x){
            psmall->next = cur;
            psmall = psmall->next;

        }else{
            pbig->next = cur;
            pbig = pbig->next;    
        }
        
        cur = cur->next;   
    }
    
    pbig->next = NULL;
    psmall->next = bigdummy.next;
    
    return smalldummy.next;
}



你可能感兴趣的