【Leetcode】面试题 02.04. 分割链表
题目
给你一个链表的头节点 head 和一个特定值 x ,请你对链表进行分隔,使得所有 小于 x 的节点都出现在 大于或等于 x 的节点之前。
你不需要 保留 每个分区中各节点的初始相对位置。
思路
- 初始化两个链表 一个保存小于节点x的值 一个保存 大于或者等于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) {
// 初始化两个链表 一个保存小于节点x的值 一个保存 大于或者等于x的节点值
// 然后将两个链表链接起来
// 本题默认带头节点
ListNode *p = new ListNode(0);// 链表头节点 数值是0
ListNode *q = new ListNode(0);
ListNode *pHead = p;
ListNode *qHead = q;
while(head != NULL)
{
if(head->val < x)
{
p->next = head;
p = p->next;
}
else
{
q->next = head;
q = q->next;
}
head = head->next;
}
q->next = NULL;// 尾节点指针置为空
// 链接两个链表 跳过第二个链表的头节点
p->next = qHead->next;
return pHead->next;
}
};