C++每日一练2-分割链表

分割链表
编写程序以 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)。