LeetCode 86. 分隔链表 Partition List(C语言)

题目描述:

给定一个链表和一个特定值 x,对链表进行分隔,使得所有小于 x 的节点都在大于或等于 x 的节点之前。
你应当保留两个分区中每个节点的初始相对位置。

示例:

输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5

题目解答:

方法1:双指针

两个指针分别记录比x大的节点和比x小的节点组成的链表。另外申请头结点方便加入新节点。
运行时间0ms,代码如下。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */
struct ListNode* partition(struct ListNode* head, int x) {
    if(head == NULL || head->next == NULL)
        return head;
    struct ListNode* t1 = (struct ListNode*)malloc(sizeof(struct ListNode));
    struct ListNode* t2 = (struct ListNode*)malloc(sizeof(struct ListNode));
    struct ListNode *temp1 = t1, *temp2 = t2;
    while(head) {
        if(head->val < x) {
            temp1->next = head;
            temp1 = temp1->next;
        }
        else {
            temp2->next = head;
            temp2 = temp2->next;
        }
        head = head->next;    
    }
    temp2->next = NULL;
    temp1->next = t2->next;
    temp1 = t1->next;
    free(t1);
    free(t2);
    return temp1;
}

你可能感兴趣的:(LeetCode,C语言)