首页 > 代码库 > Partition List
Partition List
问题描述:
Given a linked list and a value x, partition it such that all nodes less than x 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.
解题思路:
这里其实就是对链表元素进行快速排序的一次划分操作,主要就是用两个指针,一个指针(p_great)用于处理链表的元素,一个指针(p_less)用于元素分界,即在p_less指针之前的元素(包括p_less指针指向的值)全部小于x值,而在p_less和p_great之间的元素全部都大于或等于x值。p_great之后的元素是还没有处理的链表元素。
在代码中,我使用了一个辅助结点,让该结点的值等于x-1,并让其next指向head头指针,这样就可以保证链表的第一个元素肯定小于x值,避免很多不必要的条件判断。
/** * 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) { if (NULL == head) return head; ListNode *p_less, *p_great, *p_parent, *new_head; new_head = p_parent = p_less = new ListNode(x-1);/*辅助结点*/ p_less->next = head; p_great = head; while (p_great != NULL) { if (p_great->val < x) { if (p_parent == p_less) {/*链表首部的元素小于给定的值*/ p_less = p_less->next; p_parent = p_great; p_great = p_great->next; } else { p_parent->next = p_great->next; p_great->next = p_less->next; p_less->next = p_great; p_less = p_great; p_great = p_parent->next; } } else { p_parent = p_great; p_great = p_great->next; } } if (new_head->next != head) head = new_head->next; free(new_head); new_head = NULL; return head; } };
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。