LeetCode——分割链表

原始连接https://blog.csdn.net/gjpzl/article/details/80151354

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

你应当保留两个分区中每个节点的初始相对位置。

示例:

关于这题自己的思路开始有个误区——以为要建立一个原链表的拷贝,其val值和原链表相同,但其实是将原链表中的节点按值大小重排。

或者以为要两次遍历链表,第一个把所有小于x的节点串起来,并记录下第一个大于等于x的节点,然后从该节点往后遍历,将大于等于x的节点接着串。看了别人的解答才发现是自己太愚蠢..............

构建两个新链表,小于x的放在链表1,大于或等于x的放在链表2,最后将链表1的表尾指向链表2的表头即可。

#include<cstdlib>
class Solution {
public:
    ListNode *partition(ListNode *head, int x) {
        ListNode* head1, *head2, *node1, *node2, *temp;
        ListNode less_head(0);  // 设置两个临时的头节点
        ListNode more_head(0); 
        
        head1 = &less_head;
        head2 = &more_head;
        node1 = head1;
        node2 = head2;
        temp = head;
        while(temp != NULL)
        {
            if(temp->val < x)
            {
                node1->next = temp;
                node1 = temp;
            }
            else
            {
                node2->next = temp;
                node2 = temp;
            }
            temp = temp->next;
        }
        
        node1->next = head2->next;
        node2->next = NULL;
        return head1->next;
    }
};

奇怪的是,我是照着别人代码写的,过程一样,我的执行结果是左图,他的是右图,差这么多啊???!

其实没必要用temp,直接用head,因为反正原链表节点的连接顺序已经被改变了,而且这个head对于返回新链表的头节点也没有用。

其他方法

还有一种方法是构建一个新链表,将所有小于给定值的节点取出放进去,并修改next指针指向以删除原链表中这些节点,剩下大于等于x的节点,再将新链表next指针指向原链表。

调错总结:

开始时显示这个错误,原来是while循环中忘记移动节点了,,,temp=temp->next愚蠢!

改了还是报这个错,再一看原来是没有将第二个链表最后的node指向NULL,有可能形成环,无穷无尽呐!!


猜你喜欢

转载自blog.csdn.net/eartha1995/article/details/81000763
今日推荐