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

partition-list

时间:2018-11-21 15:51:37      阅读:163      评论:0      收藏:0      [点我收藏+]

标签:val   ios   return   class   clu   tno   public   ace   节点   

题意略:

说一下自己的两个坑点:当为空表或者只有一个节点时,应该返回head而不是NULL

#include<iostream>
using namespace std;
 struct ListNode {
     int val;
     ListNode *next;
     ListNode(int x) : val(x), next(NULL) {}
 };

class Solution {
public:
    ListNode *partition(ListNode *head, int x) {
        if (head == NULL||head->next==NULL)return head;
        ListNode *head1 = new ListNode(-1);
        ListNode *head2 = new ListNode(-1);
        ListNode *end1 = head1;
        ListNode *end2 = head2;
        ListNode *p = head;
        while (p){
            if (p->val < x){
                end1 = end1->next = p;
            }
            else{
                end2 = end2->next = p;
            }
            p = p->next;
        }
        end2->next = NULL;
        end1->next = head2->next;
        return head1->next;
    }
};

 

partition-list

标签:val   ios   return   class   clu   tno   public   ace   节点   

原文地址:https://www.cnblogs.com/ALINGMAOMAO/p/9994698.html

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