题目描述:
给你一个链表的头节点 head 和一个特定值 x ,请你对链表进行分隔,使得所有 小于 x 的节点都出现在 大于或等于 x 的节点之前。
你不需要 保留 每个分区中各节点的初始相对位置。
解题思路:
可能是连续做了好几题链表的题目,看到这道题比较容易的想到了创建两个链表,分别保存小于x的节点和大于等于x的节点,遍历完之后将前一个链表 链接到 后一个链表上。其中需要注意的是存放大于等于x的链表的末尾节点有可能还在指向其他节点,从而造成环形链表而编译错误,所以在最后的时候要对该链表的末尾置空。
代码如下:
struct ListNode* partition(struct ListNode* head, int x){
struct ListNode* low = (struct ListNode*)malloc(sizeof(struct ListNode));
low->val = -1;
low->next = NULL;
struct ListNode* high = (struct ListNode*)malloc(sizeof(struct ListNode));
high->val = -1;
high->next = NULL;
struct ListNode* low_tail = low;
struct ListNode* high_tail = high;
while(head)
{
if(head->val<x)
{
low_tail->next = head;
low_tail = low_tail->next;
}
else
{
high_tail->next = head;
high_tail = high_tail->next;
}
head = head->next;
}
high_tail->next = NULL;
low_tail->next = high->next;
return low->next;
}
为了解决头结点需要特殊考虑的问题,分别在两个链表初始化的时候创建两个哨兵节点,等链接两个链表和返回时跳过该节点即可。