leetcode Partition List

来源:互联网 发布:艾力欧与知世怀孕虐文 编辑:程序博客网 时间:2024/04/29 14:47
class Solution {public:    ListNode* partition(ListNode* head, int x) {        ListNode* smallHead = new ListNode(0);// 增加头结点,简化代码        ListNode* notSmallHead = new ListNode(0);        ListNode* notSmallPtr = notSmallHead;        ListNode* smallPtr = smallHead;        for (ListNode* ptr = head; ptr;){            ListNode* tmp = ptr->next;            if (ptr->val < x){                smallPtr->next = ptr;                smallPtr = ptr;            }            else{                notSmallPtr->next = ptr;                notSmallPtr = ptr;                ptr->next = NULL;            }            ptr = tmp;        }        smallPtr->next = notSmallHead->next;        return smallHead->next;;    }};

0 0