给定一个链表和一个特定值 x,对链表进行分隔,使得所有小于 x 的节点都在大于或等于 x 的节点之前。
你应当保留两个分区中每个节点的初始相对位置。
示例:
输入: head = 1->4->3->2->5->2, x = 3 输出: 1->2->2->4->3->5
1 #include "_000库函数.h" 2 3 struct ListNode { 4 int val; 5 ListNode *next; 6 ListNode(int x) : val(x), next(NULL) {} 7 }; 8 9 10 //感觉用两条链表可能会更简单点,但内存会大点, 11 //的确,内存多了点 12 class Solution { 13 public: 14 ListNode* partition(ListNode* head, int x) { 15 ListNode* p1 = new ListNode(-1);//用来存放小于x的数 16 ListNode* p2 = new ListNode(-1);//用来存放其他数字 17 ListNode* q1 = p1; 18 ListNode* q2 = p2; 19 while (head) {//无头结点 20 ListNode* q = new ListNode(0); 21 q->val = head->val; 22 if (head->val < x) { 23 q1->next = q; 24 q1 = q; 25 } 26 else { 27 q2->next = q; 28 q2 = q; 29 } 30 head = head->next; 31 } 32 q1->next = p2->next;//组合为一个链表 33 return p1->next;//不需要头结点 34 } 35 }; 36 37 //用插入法,在原链表上操作 38 class Solution { 39 public: 40 ListNode* partition(ListNode* head, int x) { 41 ListNode* p = new ListNode(-1); 42 p->next = head; 43 head = p;//加一个头结点 44 ListNode* pro, *cur; 45 pro = p;//要插入的位置 46 cur = p;//移动的位置 47 p = p->next; 48 while (p) { 49 if (p->val < x) { 50 ListNode* q = new ListNode(0); 51 q->val = p->val; 52 cur->next = p->next;//删除该数 53 p = cur->next; 54 q->next = pro->next;//插入 55 pro->next = q; 56 if (cur == pro) 57 cur = cur->next; 58 pro = q; 59 } 60 else { 61 p = p->next; 62 cur = cur->next; 63 } 64 } 65 return head->next; 66 } 67 }; 68 69 void T086() { 70 Solution s; 71 vector<int>v; 72 ListNode *head = new ListNode(0); 73 ListNode *p = head; 74 v = { 1,4,3,2,5,2 }; 75 for (auto a : v) { 76 ListNode *q = new ListNode(0); 77 q->val = a; 78 p->next = q; 79 p = q; 80 } 81 p = head->next; 82 while (p) { 83 cout << p->val << "->"; 84 p = p->next; 85 } 86 cout << endl; 87 p = s.partition(head->next,3); 88 while (p) { 89 cout << p->val << "->"; 90 p = p->next; 91 } 92 cout << endl; 93 }