首页 > 代码库 > 【LeetCode】Partition List
【LeetCode】Partition List
Given a linked list and a value x, partition it such that all nodes less thanx come before nodes greater than or equal to x.
You should preserve the original relative order of the nodes in each of the two partitions.
For example,
Given 1->4->3->2->5->2
and x = 3,
return 1->2->2->4->3->5
.
/** * Definition for singly-linked list. * struct ListNode { * int val; * ListNode *next; * ListNode(int x) : val(x), next(NULL) {} * }; */ class Solution { public: ListNode *partition(ListNode *head, int x) { ListNode *smallhead = NULL; ListNode *bighead = NULL; ListNode *tail_small = NULL; ListNode *tail_big = NULL; ListNode *cur = head; if (head == NULL || head->next == NULL) { return head; } while(cur != NULL) { ListNode *tmp_next = cur->next; if (cur->val < x) { if (smallhead == NULL) { smallhead = tail_small = cur; } else { tail_small->next = cur; tail_small = cur; } } else{ if (bighead == NULL) { bighead = tail_big = cur; } else{ tail_big->next = cur; tail_big = cur; } } cur = tmp_next; } if (tail_small == NULL) { tail_big->next = NULL; return bighead; } if (tail_big == NULL) { tail_small->next = NULL; return smallhead; } tail_small->next = bighead; tail_big->next = NULL; return smallhead; } };
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。