分割链表 编写程序以 x 为基准分割链表,使得所有小于 x 的节点排在大于或等于 x 的节点之前。如果链表中包含 x,x 只需出现在小于 x 的元素之后(如下所示)。分割元素 x 只需处于“右半部分”即可,其不需要被置于左右两部分之间。
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/partition-list-lcci
示例:
输入: head = 3->5->8->5->10->2->1, > x = 5 输出: 3->1->2->10->5->5->8
整体思路:(文中的意思就是只要把比x小的放在x的左边节点即可) 1.创建两条新链,分别储存小于x和大于x的节点的值 2.小于x的链指向大于x的链的头结点 3.返回产生的新链 结构体定义:
struct ListNode { int val; ListNode *next; ListNode(int x) : val(x), next(NULL) {} };s代码:**
class Solution { public: ListNode* partition(ListNode* head, int x) { ListNode* smaller_list = new ListNode(0); ListNode* greater_list = new ListNode(0); ListNode* p = head; ListNode* p1 = smaller_list; ListNode* p2 = greater_list; while (p) { if (p->val < x) { p1->next = p; p1 = p1->next; } else { p2->next = p; p2 = p2->next; } p = p->next; } //因为p1和p2必然不会为空节点,所以不用判断他们为空 p1->next = greater_list->next; p2->next = nullptr; return smaller_list->next; } };复杂度分析
时间复杂度:O(n)。n为head的长度 空间复杂度:O(1)。