86. Partition List。

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的放在后面,注意不要乱了原来的顺序。也是一道比较简单的题,我的第一个思路就是使用stack的特性,使用两个栈。遍历链表,小于x的放在一个栈中,大于或者等于放在另一个栈中。然后遍历两个栈,如果存放小数值的栈不为空则将该栈中的节点出栈并使用头插法连接起来,然后遍历存放大数值的栈,将其连接起来即可。

class Solution {
public:
    ListNode* partition(ListNode* head, int x) {
        if(!head) {
            return NULL;
        }

        ListNode* result = new ListNode(0);

        stack<ListNode*> lt;//小于x的
        stack<ListNode*> gt;//大于x的

        while(head) {
            if(head->val < x) {
                lt.push(head);
            } else {
                gt.push(head);
            }
            head = head->next;
        }

        while(!lt.empty() || !gt.empty()) {
            if(!gt.empty()) {//如果大的数没有遍历完则遍历大的
                gt.top()->next = result->next;
                result->next = gt.top();
                gt.pop();
            } else {
                lt.top()->next = result->next;
                result->next = lt.top();
                lt.pop();
            }
        }

        return result->next;
    }
};

第二种方法就是使用两个指针,遍历一遍链表,使用一个指针将小于x的节点连接起来,用另一个节点将大于或者等于x的节点连接起来,最后将两个链表按照小的在前,大的在后的顺序整合起来即可。

#include <iostream>
#include <stack>
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) {

        ListNode* ltDummy = new ListNode(0);
        ListNode* gtDummy = new ListNode(0);
        ListNode* ltTail = ltDummy;
        ListNode* gtTail = gtDummy;

        while(head) {
            if(head->val < x) {
                ltTail->next = head;
                ltTail = head;
            } else {
                gtTail->next = head;
                gtTail = head;
            }
            head = head->next;
        }

        ltTail->next = gtDummy->next;
        gtTail->next = NULL;//最后一个可能不为空,所以需要赋值为空

        return ltDummy->next;


        /*if(!head) {
            return NULL;
        }

        ListNode* result = new ListNode(0);

        stack<ListNode*> lt;//小于x的
        stack<ListNode*> gt;//大于x的

        while(head) {
            if(head->val < x) {
                lt.push(head);
            } else {
                gt.push(head);
            }
            head = head->next;
        }

        while(!lt.empty() || !gt.empty()) {
            if(!gt.empty()) {//如果大的数没有遍历完则遍历大的
                gt.top()->next = result->next;
                result->next = gt.top();
                gt.pop();
            } else {
                lt.top()->next = result->next;
                result->next = lt.top();
                lt.pop();
            }
        }

        return result->next;*/

    }
};

int main() {
    Solution s;


    ListNode node1(1);
    ListNode node2(4);
    ListNode node3(3);
    ListNode node4(2);
    ListNode node5(5);
    ListNode node6(2);

    node1.next = &node2;
    node2.next = &node3;
    node3.next = &node4;
    node4.next = &node5;
    node5.next = &node6;

    ListNode* p = s.partition(&node1,3);

    while(p) {
        cout << p->val;
        cout << endl;
        p = p->next;
    }

}

您可以使用Java 8中引入的Stream API中的partitioningBy()方法来将列表分区为两个集合。该方法需要一个Predicate参数来分区元素,并返回一个Map<Boolean, List<T>>对象,其中Boolean键表示元素是否满足分区条件,List<T>表示满足或不满足条件的元素列表。下面是示例代码: import java.util.stream.Collectors; import java.util.List; import java.util.Map; import java.util.Collection; public class MyClass { public static <T> Map<Boolean, Collection<T>> partition(Collection<T> inputCollection, Predicate<? super T> condition) { return inputCollection.stream().collect(Collectors.partitioningBy(condition)); } } 在这个方法中,参数inputCollection是要分区的集合,参数condition是分区的条件,返回一个Map<Boolean, Collection<T>>对象,其中Boolean键表示元素是否满足分区条件,Collection<T>表示满足或不满足条件的元素集合。 您可以使用以下代码调用此方法: List<String> inputList = Arrays.asList("foo", "bar", "baz", "qux"); Predicate<String> condition = str -> str.startsWith("b"); Map<Boolean, Collection<String>> result = partition(inputList, condition); 在这个例子中,输入列表包含四个字符串元素。我们首先定义一个Predicate对象,该对象返回true如果字符串以“b”开头。我们然后调用partition()方法将输入列表分区为两个集合,其中一个集合包含以“b”开头的字符串,另一个集合包含其他字符串。最终,我们将返回的Map对象存储在result变量中。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值