码迷,mamicode.com
首页 > 其他好文 > 详细

【LeetCode-链表】分隔链表

时间:2020-04-10 21:16:33      阅读:69      评论:0      收藏:0      [点我收藏+]

标签:problem   return   class   nullptr   申请   示例   struct   等于   pre   

题目描述

给定一个链表和一个特定值 x,对链表进行分隔,使得所有小于 x 的节点都在大于或等于 x 的节点之前。
你应当保留两个分区中每个节点的初始相对位置。
示例:

输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5

题目链接: https://leetcode-cn.com/problems/partition-list/

思路

x会将链表分割成两部分,左链表中的节点值全都小于x,右链表中的节点值全都大于等于x,则我们可以创建两个链表smallList和bigList,smallList存储左链表,bigList存储右链表。从头开始遍历原链表,当当前节点值小于x时,将当前结点加入smallList;否则加入到bigList。代码如下:

/**
 * 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) {
        if(head==nullptr){
            return nullptr;
        }

        ListNode* smallList = new ListNode(0);  // 数值<x的链表头
        ListNode* bigList = new ListNode(0);    // 数值>=x的链表头
        ListNode* curSmall = smallList;
        ListNode* curBig = bigList;
        ListNode* curNode = head;
        while(curNode!=nullptr){
            if(curNode->val<x){
                curSmall->next = curNode;
                curSmall = curNode;
                curNode = curNode->next;
            }else{
                curBig->next = curNode;
                curBig = curNode;
                curNode = curNode->next;
            }
        }
        curSmall->next = bigList->next;
        curBig->next = nullptr; //  这句别忘了
        return smallList->next;
    }
};
  • 时间复杂度:O(n)
    遍历了链表一遍。
  • 空间复杂度:O(1)
    申请的额外内存不随输入规模的增大而增大。

【LeetCode-链表】分隔链表

标签:problem   return   class   nullptr   申请   示例   struct   等于   pre   

原文地址:https://www.cnblogs.com/flix/p/12676093.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!