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

Partition List

时间:2015-07-31 23:34:56      阅读:116      评论:0      收藏:0      [点我收藏+]

标签:

Given a linked list and a value x, partition it such that all nodes less than x come before nodes greater than or equal to x.

You should preserve the original relative order of the nodes in each of the two partitions.

For example,
Given 1->4->3->2->5->2 and x = 3,

return 1->2->2->4->3->5.

解题思路:

已知一条单向链表和一个固定值x,将链表中小与x的值放置在x的前面,大于或者等于x的值放置在x的链表的后面。其实非常简单,我们定义两个新指针,从头遍历链表,小与x的节点挂在第一个指针后面,大于或者等于x的节点挂在第二个指针后面,最后将两个指针组成的短链表联合起来即可。

代码一:

/**
 * 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==NULL) return NULL;
        if(head->next==NULL) return head;
        ListNode* p=head;
        ListNode* s1=new ListNode(0);
        ListNode* s2=new ListNode(0);
        ListNode* ss1=s1;
        ListNode* ss2=s2;
        while(p)
        {
            if(p->val<x)
            {
                 s1->next=p;
                 s1=s1->next;
            }
            else 
            {
               s2->next=p;
               s2=s2->next; 
            }
           p=p->next;
        }
        
        if(ss2->next)
        {
            ss2=ss2->next;
            s2->next=NULL;
        }
        else
        ss2=NULL;
        if(ss1->next)
        {
            s1->next=ss2;
            ss1=ss1->next;
            return ss1;
        }
        else
        return ss2;
    }
};
代码一,简单直观,但是略有繁琐。下面我们稍作改动:

代码二:

class Solution {
public:
    ListNode* partition(ListNode* head, int x) {
        if(head==NULL) return NULL;
        if(head->next==NULL) return head;
        ListNode* p=head;
        ListNode* s1=new ListNode(0);
        ListNode* s2=new ListNode(0);
        ListNode* ss1=s1;
        ListNode* ss2=s2;
        while(p)
        {
            if(p->val<x)
            {
                 s1=s1->next=p;
            }
            else 
            {
               s2=s2->next=p;
            }
           p=p->next;
        }
        s2->next=NULL;
        s1->next=ss2->next;
        return ss1->next;
    }
};
注意:

代码二中s1=s1->next=p;指的是s1->next=p,之后再s1=p。也就是说s1节点指针跟p的属性相同。这是a=b=c形式,从右向左依次赋值,最后等式中的变量都和最右边的变量值相同;(a=b)=c 形式就不同了,虽然这种形式很少出现,指的是现将b的值付给a,a=b,之后再将c的值赋值给a,a=c;
相当于:
a=b;
a=c;
完成之后,b的值没有改变,c的值也没有改变,只是a=c而已。




版权声明:本文为博主原创文章,未经博主允许不得转载。

Partition List

标签:

原文地址:http://blog.csdn.net/sinat_24520925/article/details/47176881

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