leetcode_8_链表分割

weixin_52872520

已于 2023-04-12 13:43:09 修改

阅读量32

点赞数

分类专栏： leetcode 文章标签：链表 leetcode 数据结构

于 2023-04-12 10:49:10 首次发布

本文链接：https://blog.csdn.net/weixin_52872520/article/details/130100013

版权

leetcode 专栏收录该内容

28 篇文章 0 订阅

订阅专栏

把小于val的数值尾插到一个链表，大于等于尾插到另一个链表，然后再链接起来。

/*
struct ListNode {
    int val;
    struct ListNode *next;
    ListNode(int x) : val(x), next(NULL) {}
};*/
class Partition {
public:
    ListNode* partition(ListNode* pHead, int x) {
    struct ListNode* greaterGuard,*greaterTail,*lessGuard,*lessTail;
    greaterGuard = greaterTail = (struct ListNode*)malloc(sizeof(struct ListNode));
    lessGuard = lessTail = (struct ListNode*)malloc(sizeof(struct ListNode));
    greaterTail->next = lessTail->next = NULL;


    struct ListNode* cur = pHead;
    while(cur)
    {
        if(cur->val < x)
        {
            lessTail->next = cur;
            lessTail = lessTail->next;
        }   
        else 
        {
            greaterTail->next = cur;
            greaterTail = greaterTail->next;
        }
        cur = cur->next;
    }
    lessTail->next = greaterGuard->next;
    greaterTail->next = NULL;


    pHead = lessGuard->next;
    free(lessGuard);
    free(greaterGuard);


    return pHead;
    }
};