在链表上实现 Partition 以及荷兰国旗问题

作者:Grey

原文地址:

博客园:在链表上实现 Partition 以及荷兰国旗问题

CSDN:在链表上实现 Partition 以及荷兰国旗问题

题目描述

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

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

题目链接见:LeetCode 86. Partition List

主要思路

本题可以借鉴数组的 Partition 操作,参考:荷兰国旗问题与快速排序算法

Partition 操作就是荷兰国旗的一种特殊情况而已。

我们可以把本题的难度稍微升级一下:如何在链表上实现荷兰国旗问题?

第一种解法就是将链表先转换成数组,然后在数组上实现荷兰国旗问题,最后将数组还原为链表并返回,该方法的时间复杂度是O(N),空间复杂度是O(N),不是最优解。

第二种解法是用有限几个变量来实现,在同样O(N)的时间复杂度的情况下,空间复杂度可以做到O(1),设置如下几个变量

ListNode sH = null; // 小于区域的头结点
ListNode sT = null; // 小于区域的尾结点
ListNode eH = null; // 等于区域的头结点
ListNode eT = null; // 等于区域的尾结点
ListNode mH = null; // 大于区域的头结点
ListNode mT = null; // 大于区域的尾结点
ListNode next; // 记录遍历到的结点的下一个结点

接下来开始遍历链表,进行主流程处理,伪代码如下

while (head != null) {
next = head.next;
// 如果head.val < pivot,则通过sH,sT构造小于区域的链表
// 如果head.val == pivot,则通过eH,eT构造小于区域的链表
// 如果head.val > pivot,则通过mH,mT构造小于区域的链表
head = next;
}
// 把三个区域的链表串联起来即可。

构造每个区域的链表的时候,还要考虑链表是第一次被构造还是已经构造了,以小于区域的链表为例,如果是第一次构造,则sH == null,此时需要把sHsT同时初始化:

if (sH == null) {
sH = head;
sT = head;
}

如果不是第一次构造,则

sT.next = head;
sT = head;

开始区域的尾指针直接指向当前结点,然后把尾指针设置未当前结点即可。

其他两个区域的链表处理方式同理。

最后需要把三个区域的链表连接起来:

        // 小于区域的尾巴,连等于区域的头,等于区域的尾巴连大于区域的头
if (sT != null) { // 如果有小于区域
sT.next = eH;
eT = eT == null ? sT : eT; // 下一步,谁去连大于区域的头,谁就变成eT
}
// all reconnect
if (eT != null) { // 如果小于区域和等于区域,不是都没有
eT.next = mH;
}
// 如果小于区域有,小于区域的头就是最终链表的头
// 如果小于区域没有,等于区域的头有,则等于区域的头就是最终链表的头
// 如果小于和等于区域都没有,则大于区域的头就是最终链表的头
return sH != null ? sH : (eH != null ? eH : mH);

完整代码见

public class Code_PartitionList {

    public static class ListNode {
public int val;
public ListNode next; public ListNode(int data) {
this.val = data;
}
} public static ListNode listPartition2(ListNode head, int pivot) {
ListNode sH = null; // small head
ListNode sT = null; // small tail
ListNode eH = null; // equal head
ListNode eT = null; // equal tail
ListNode mH = null; // big head
ListNode mT = null; // big tail
ListNode next; // save next node
// every node distributed to three lists
while (head != null) {
next = head.next;
head.next = null;
if (head.val < pivot) {
if (sH == null) {
sH = head;
sT = head;
} else {
sT.next = head;
sT = head;
}
} else if (head.val == pivot) {
if (eH == null) {
eH = head;
eT = head;
} else {
eT.next = head;
eT = head;
}
} else {
if (mH == null) {
mH = head;
mT = head;
} else {
mT.next = head;
mT = head;
}
}
head = next;
}
// 小于区域的尾巴,连等于区域的头,等于区域的尾巴连大于区域的头
if (sT != null) { // 如果有小于区域
sT.next = eH;
eT = eT == null ? sT : eT; // 下一步,谁去连大于区域的头,谁就变成eT
}
// all reconnect
if (eT != null) { // 如果小于区域和等于区域,不是都没有
eT.next = mH;
}
// 如果小于区域有,小于区域的头就是最终链表的头
// 如果小于区域没有,等于区域的头有,则等于区域的头就是最终链表的头
// 如果小于和等于区域都没有,则大于区域的头就是最终链表的头
return sH != null ? sH : (eH != null ? eH : mH);
}
}

解决了链表的荷兰国旗问题,那么原题中的链表 Partition 问题,就迎刃而解了。

由于是 Partition,所以不存在等于区域,只需要考虑小于区域和大于区域,只需要设置如下几个变量即可。

ListNode sH = null; // 小于区域的头
ListNode sT = null; // 小于区域的尾
ListNode mH = null; // 大于区域的头
ListNode mT = null; // 大于区域的尾
ListNode cur = head; // 当前遍历到的结点

接下来开始遍历链表,进行主流程处理,伪代码如下

while (cur != null) {
// 如果head.val < pivot,则通过sH,sT构造小于区域的链表
// 如果head.val > pivot,则通过mH,mT构造小于区域的链表
cur = cur.next;
}
// 把两个区域的链表串联起来即可。

构造每个区域的链表的时候和上述荷兰国旗问题一样。

最后需要把两个区域的链表连接起来:


if (mT != null) {
// 大于区域的尾置空
mT.next = null;
} if (sT != null) {
// 小于区域的尾置空
sT.next = null;
}
// 经过上述操作,两个链表断开连接
if (sH == null) {
// 小于区域的头为空,说明只有大于区域
return mH;
}
// 小于区域的头不为空,小于区域的尾一定也不为空,直接把小于区域的尾连上大于区域的头即可。
sT.next = mH;
return sH;

完整代码见

class Solution {
public static ListNode partition(ListNode head, int x) {
ListNode sH = null;
ListNode sT = null;
ListNode mH = null;
ListNode mT = null;
ListNode cur = head;
while (cur != null) {
if (cur.val < x) {
if (sH == null) {
sH = cur;
} else {
sT.next = cur;
}
sT = cur;
} else {
// cur.val >= x
// 都放到大于等于区域
if (mH == null) {
mH = cur;
} else {
mT.next = cur;
}
mT = cur;
}
cur = cur.next;
}
if (mT != null) {
mT.next = null;
}
if (sT != null) {
sT.next = null;
}
if (sH == null) {
return mH;
}
sT.next = mH;
return sH;
}
}

更多

算法和数据结构笔记

在链表上实现 Partition 以及荷兰国旗问题的更多相关文章

  1. July收集荷兰国旗问题之三路partition

    这道题目和分成两块的partition的扩展.比如有一堆0 1 2 数字组成的数组,要分成 00 00  11 1 1  222 2这样的顺序的. 利用lumoto版的partition能够非常好的解 ...

  2. 快速排序与荷兰国旗及Partition问题

    快速排序与荷兰国旗及Partition问题 需求: 1.Partition过程 给定一个数组arr,和一个整数num.请把小于等于num的数放在数组的左边,大于num的数放在数组的右边. 要求额外空间 ...

  3. 荷兰国旗问题、快排以及BFPRT算法

    荷兰国旗问题 给定一个数组arr,和一个数num,请把小于num的数放数组的左边,等于num的数放在数组的中间,大于num的数放在数组的右边.要求额外空间复杂度O(1),时间复杂度O(N). 这个问题 ...

  4. ACM 荷兰国旗问题

    荷兰国旗问题 时间限制:3000 ms  |  内存限制:65535 KB 难度:1   描述 荷兰国旗有三横条块构成,自上到下的三条块颜色依次为红.白.蓝.现有若干由红.白.蓝三种颜色的条块序列,要 ...

  5. 算法笔记_051:荷兰国旗问题(Java)

    目录 1 问题描述 2 解决方案   1 问题描述 现有n个红白蓝三种不同颜色的小球,乱序排列在一起,请通过两两交换任意两个球,使得从左至右的球依次为红球.白球.蓝球.这个问题之所以叫荷兰国旗,是因为 ...

  6. Coursera Algorithms week2 基础排序 练习测验: Dutch national flag 荷兰国旗问题算法

    第二周课程的Elementray Sorts部分练习测验Interview Questions的第3题荷兰国旗问题很有意思.题目的原文描述如下: Dutch national flag. Given ...

  7. NYOJ_268_荷兰国旗问题

    荷兰国旗问题 时间限制:3000 ms  |  内存限制:65535 KB 难度:1 描写叙述 荷兰国旗有三横条块构成,自上到下的三条块颜色依次为红.白.蓝.现有若干由红.白.蓝三种颜色的条块序列.要 ...

  8. Java实现荷兰国旗问题

    问题描述 现有n个红白蓝三种不同颜色的小球,乱序排列在一起,请通过两两交换任意两个球,使得从左至右的球依次为红球.白球.蓝球.这个问题之所以叫荷兰国旗,是因为将红白蓝三色的小球弄成条状物,并有序排列后 ...

  9. 荷兰国旗 Flag of the Kingdom of the Netherlands

    问题描述:现有n个红白蓝三种不同颜色的小球,乱序排列在一起,请通过两两交换任意两个球,使得从左至右的球依次为红球.白球.蓝球.这个问题之所以叫做荷兰国旗,是因为将红白蓝三色的小球弄成条状物,并有序排列 ...

  10. 148 Sort List 链表上的归并排序和快速排序

    在使用O(n log n) 时间复杂度和常数级空间复杂度下,对链表进行排序. 详见:https://leetcode.com/problems/sort-list/description/ Java实 ...

随机推荐

  1. 操作服务器的神奇工具Tmux

    Tmux 是什么? 会话与进程 命令行的典型使用方式是,打开一个终端窗口(terminal window,以下简称"窗口"),在里面输入命令.用户与计算机的这种临时的交互,称为一次 ...

  2. 不懂 Kubernetes 实现云原生是什么体验?

    云原生的本质和最终效果 要明白什么是云原生,就要先弄明白云计算是什么有什么问题,云计算将计算资源.网络.存储等基础设施统一管理,通过资源规模化和自动化管理,实现降低资源的成本和提高资源的管理效率,云计 ...

  3. Windows 10无法显示无线网络连接

    最近刚刚升级了一下操作系统,升级到了1903版本.正好又有一个HP的打印机安装了一下.结果,发现居然无法管理无线网络了.如果看不到图,请点我. 右击选择连接,也无法显示SSID. 驱动是从这个官网下载 ...

  4. linux使用iptables屏蔽ip地址

    一.iptables命令介绍: netfilter/iptables(简称为iptables)组成Linux平台下的包过滤防火墙,与大多数的Linux软件一样,这个包过滤防火墙是免费的,在安装系统的时 ...

  5. 《吐血整理》高级系列教程-吃透Fiddler抓包教程(22)-如何使用Fiddler生成Jmeter脚本-下篇

    1.简介 今天这篇文章其实和上一篇差不多也是利用一个fiddler的插件进行Jmeter脚本的导出,开始宏哥想要合在一起写一篇文章,可是结果实践的时候,两个插件还是有区别的,因此为了不绕晕小伙伴或者童 ...

  6. Minio VS Ceph

    文章转载自:https://blog.csdn.net/Moolight_shadow/article/details/123204412

  7. Centos7新增静态路由

    文章转载自:https://blog.51cto.com/loong576/2417561 环境说明: 一.临时方式 1. 查看路由和ip [root@centos7 ~]# route -n Ker ...

  8. Elasticsearch Painless script编程

    我们之前看见了在Elasticsearch里的ingest node里,我们可以通过以下processor的处理帮我们处理我们的一些数据.它们的功能是非常具体而明确的.那么在Elasticsearch ...

  9. Logstash: 启动监控及集中管理

    在本篇文章里,我将详细介绍如果启动Logstash的监控及集中管理. 前提条件 安装好Logstash,设置Elasticsearch及Kibana的安全密码. 如何监控Logstash? 我们安装如 ...

  10. kvm上已安装的虚拟机修改为桥接网络

    kvm上安装的虚拟机默认使用的nat网络格式,现在已经调整kvm主机为桥接方式了,但是已经安装的虚拟机还是nat方式,所以需要修改一下 让KVM虚拟主机使用桥接网络br0 修改虚拟机的配置文件,默认存 ...