面试题 02.04. 分割链表 C语言实现

题目描述:

给你一个链表的头节点 head 和一个特定值 x ,请你对链表进行分隔,使得所有 小于 x 的节点都出现在 大于或等于 x 的节点之前。

你不需要 保留 每个分区中各节点的初始相对位置。

题目链接

面试题 02.04. 分割链表 C语言实现_第1张图片

解题思路: 

可能是连续做了好几题链表的题目,看到这道题比较容易的想到了创建两个链表,分别保存小于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->valnext = 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;
}

 为了解决头结点需要特殊考虑的问题,分别在两个链表初始化的时候创建两个哨兵节点,等链接两个链表和返回时跳过该节点即可。

你可能感兴趣的:(力扣刷题,链表,数据结构)