题目描述
编写代码,以给定值x为基准将链表分割成两部分,所有小于x的结点排在大于或等于x的结点之前
给定一个链表的头指针 ListNode* pHead,请返回重新排列后的链表的头指针。注意:分割以后保持原来的数据顺序不变。
思路:
1.我们可以定义两个链表(结构体类型变量)(比如small,large),分别放小于x的结点、大于或等于x的结点。
2.对原链表进行遍历,与x进行比较大小,对两个新链表进行尾插(所以还要分别定义两个指针哦)
3.遍历完后,结点位置都已经放好了,我们再把两个链表连起来就可以了
注:
这里要用到 . 操作符,它是结构体类型变量访问内部成员的;而 -> 是通过指针访问结构体内部的变量
struct ListNode {
int val;
struct ListNode *next;
ListNode(int x) : val(x), next(NULL) {}
};
class Partition {
public:
ListNode* partition(ListNode* pHead, int x) {
//先给两个链表,定义结构体类型的变量
ListNode small(0);
ListNode large(0);
//给两个链表分别定义指针(进行尾插)
ListNode* largeNode = &large;
ListNode* smallNode = &small;
while (pHead)
{
if (pHead->val < x)
{
smallNode->next = pHead;
smallNode = pHead;
}
else
{
largeNode->next = pHead;
largeNode = pHead;
}
pHead = pHead->next;
}
largeNode->next = NULL;//要让最后的结点指向空
smallNode->next = large.next;//将两个链表连接
return small.next;
}
};