【程序员面试金典】链表分割

最新推荐文章于 2023-03-18 14:55:23 发布

原创最新推荐文章于 2023-03-18 14:55:23 发布 · 741 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#程序员面试金典 #C++ #链表 #指针 #算法

C++ 同时被 2 个专栏收录

66 篇文章

订阅专栏

C++面试总结

65 篇文章

订阅专栏

本文介绍了一种链表分割算法，该算法将链表中小于给定值x的节点排在大于或等于x的节点之前，同时保持原有节点顺序不变。通过创建两个临时链表来分别存放小于x和大于等于x的节点，最后将这两个链表连接起来。

【程序员面试金典】链表分割

题目

编写代码，以给定值x为基准将链表分割成两部分，所有小于x的结点排在大于或等于x的结点之前
给定一个链表的头指针 ListNode* pHead，请返回重新排列后的链表的头指针。注意：分割以后保持原来的数据顺序不变。

分析

头结点可能小于x也可能大于等于x，也有可能出现链表中所有结点都小于x或大于等于x的情况。为了减少各种特殊情况的判断，可以创建两个新的链表，分别存放小于x的结点和大于等于x的结点，再将两个链表去掉头结点后连在一起。

解答

/*
struct ListNode {
    int val;
    struct ListNode *next;
    ListNode(int x) : val(x), next(NULL) {}
};*/
class Partition {
public:
    ListNode* partition(ListNode* pHead, int x) {
        // write code here
        if (pHead == NULL) return pHead;
        ListNode *tmpSmallHead = new ListNode(0), *tmpBigHead = new ListNode(0);
        ListNode *smallTail = tmpSmallHead, *smallHead = tmpSmallHead, *ahead = pHead, *bigHead = tmpBigHead, *bigTail = tmpBigHead;
        while(ahead != NULL){
            if(ahead->val < x){
                smallTail->next = ahead;
                smallTail = smallTail->next;
            }
            else{
                bigTail->next = ahead;
                bigTail = bigTail->next;
            }
            ahead = ahead->next;
        }
        if (smallTail == smallHead){
            bigHead = tmpBigHead->next;
            delete tmpBigHead;
            delete tmpSmallHead;
            return bigHead;
        }
        else if (bigTail == bigHead){
            smallHead = tmpSmallHead->next;
            delete tmpBigHead;
            delete tmpSmallHead;
            return smallHead;
        }
        smallTail->next = bigHead->next;
        bigTail->next = NULL;
        smallHead = tmpSmallHead->next;
        delete tmpBigHead;
        delete tmpSmallHead;
        return smallHead;
    }
};