剑指offer之partition算法

1 问题

partition 算法:

从无序数组中选出枢轴点 pivot,然后通过一趟扫描,以 pivot 为分界线将数组中其他元素分为两部分,使得左边部分的数小于等于枢轴,右边部分的数大于等于枢轴(左部分或者右部分都可能为空),最后返回枢轴在新的数组中的位置。

如果原始数组为[5,9,2,1,4,7,5,8,3,6],那么整个处理的过程如下图

Partition 可不只用在快速排序中,还可以用于 Selection algorithm(在无序数组中寻找第K大的值)中.

2 代码实现

我们按照算法需求简单实现如下

void swap(int* a, int* b)
{
    int temp = *a;
    *a = *b;
    *b = temp;
}

int partition(vector<int>& vector, int start, int end)
{
    if (vector.size() < 1)
    {
        std::cout << "vector is null or vector size is not normal" << std::endl;
        return -1;
    }
    //一般我们写代码不要这样写死,pivot = vector[0],如果遇到这种写死数字的时候我们确认下是否是可以用变量更加合适
    int pivot = vector[start];
    int index = 0;
    for (int i = start + 1; i < end; ++i)
    {
        if (vector[i] <= pivot)
        {
            ++index;
            swap(vector[index], vector[i]);
        }
    }
    swap(vector[0], vector[index]);
    return index;
}

3 优化

上面实现的效率很低,我们需要优化,如果我们考虑用2个指针的思想,保持头尾两个指针向中间扫描,每次在头部找到大于pivot的值,同时在尾部找到小于pivot的值,然后将它们做一个交换

1)第一种优化代码实现

#include <iostream>
#include <vector>

using namespace std;

/*
 *partition算法 记得如果这里是C++我们传递的是vector类型,我们记得要加引用,
 *不然改变不了数据,这里和java传递ArrayList不一样,ArrayList作为参数可以改变集合里面的值,
 *所以C++如果函数传递非基本数据类型,一半都是带引用的
 */
int partitionOne(vector<int>& vector, int start, int end)
{
    if (start > end)
    {
        std::cout << "vector is empty or start > end" << std::endl;
        return -1;
    }
    int pivot = vector[start];
    while (start < end)
    {
        //我们先从尾巴开始
        while (start < end && pivot <= vector[end])
        {
            --end;
        }
        //这里用的数组赋值,而不是直接用swap交换函数,那么下面的2步也是用数组赋值,而不是用swap交换函数
        vector[start] = vector[end];
        while (start < end && pivot >= vector[start])
        {
            ++start;
        }
        vector[end] = vector[start];
    }
    vector[start] = pivot;
    return start;
}

void printVector(vector<int> v)
{
    for (int i = 0; i < v.size(); ++i)
    {
        std::cout << v[i] << "\t";
    }
    std::cout << std::endl;
}

int main()
{
    vector<int> v1;
    //[5,9,2,1,4,7,5,8,3,6]
    v1.push_back(5);
    v1.push_back(9);
    v1.push_back(2);
    v1.push_back(1);
    v1.push_back(4);
    v1.push_back(7);
    v1.push_back(5);
    v1.push_back(8);
    v1.push_back(3);
    v1.push_back(6);

    std::cout << "old data print " << std::endl;
    printVector(v1);

    partitionOne(v1, 0, v1.size() - 1);

    std::cout << "after partitionOne" << std::endl;
    printVector(v1);
    return 0;
}

运行结果如下

old data print
5921475836
after partitionOne
3421575896

然后图解每一步如下

2)第二种优化代码实现

我们使用交换函数,而不是数组赋值,和上面差不多,我们用swap函数的时候,我们单独定义了2个变量i和j保存了start和end,这样在后面的最后一个swap函数的时候进行swap(vector[i], vector[start]);还有这个函数是返回i,而不是start,不然就有问题。

#include <iostream>
#include <vector>

using namespace std;

void swap(int* a, int* b)
{
    int temp = *a;
    *a = *b;
    *b = temp;
}

void printVector(vector<int> v)
{
    for (int i = 0; i < v.size(); ++i)
    {
        std::cout << v[i] << "\t";
    }
    std::cout << std::endl;
}

/*
 *partition算法 记得如果这里是C++我们传递的是vector类型,我们记得要加引用,
 *不然改变不了数据,这里和java传递ArrayList不一样,ArrayList作为参数可以改变集合里面的值,
 *所以C++如果函数传递非基本数据类型,一半都是带引用的
 */
int partitionTwo(vector<int>& vector, int start, int end)
{
    if (start > end)
    {
        return -1;
    }
    int i = start;
    int j = end;
    int pivot = vector[start];
    while (i < j)
    {
        //我们先从尾巴开始
        while (i < j && pivot <= vector[j])
        {
            --j;
        }
        //这里用的数组赋值,而不是直接用swap交换函数,那么下面的2步也是用数组赋值,而不是用swap交换函数
        while (i < j && pivot >= vector[i])
        {
            ++i;
        }
        swap(vector[i], vector[j]);
    }
    swap(vector[i], vector[start]);
    //printVector(vector);
    return i;
}

int main()
{
    vector<int> v1;
    //[5,9,2,1,4,7,5,8,3,6]
    v1.push_back(5);
    v1.push_back(9);
    v1.push_back(2);
    v1.push_back(1);
    v1.push_back(4);
    v1.push_back(7);
    v1.push_back(5);
    v1.push_back(8);
    v1.push_back(3);
    v1.push_back(6);

    std::cout << "old data print " << std::endl;
    printVector(v1);

    partitionTwo(v1, 0, v1.size() - 1);

    std::cout << "after partitionOne" << std::endl;
    printVector(v1);
    return 0;
}

运行结果如下

old data print
5921475836
after partitionOne
4321575896

4 总结

我们使用partition算法的时候,从我们上面代码第一次调用来看,我们选择的第一个数字5作为中间轴,然后执行一次后,我们的 partition函数返回的start或者i值都是4,然后我们最后一步把5也插入了vector[4]那里,就是说明我们左边有4个值比当前数字5作为中间轴都小,也能说明这左边的4个值和中间轴数5都是数组里面最小的5个值,如果我们需要求出一个数组里面最小的5个值,我们只需要partition算法返回值是4就行,然后在左边的数组的前5个数字就是这个数组里面最小的5个数,所以这里的数组里面最小的多少K个数确保partition返回的index或者start的关系是:index = K - 1; 或者start = K -1关系,也就是说partition函数返回index或者start值的时候,数组里面从坐标0到index或者start的值就是数组里面最小的元素,也就是index+1个元素。

我们使用partition算法是双指针思想

(0)

相关推荐

  • 「排序算法」—图解双轴快排

    首发公众号:bigsai 前言 在排序算法中,快排是占比非常多的一环,但是快排其思想一直被考察研究,也有很多的优化方案.这里主要讲解双轴快排的思想和实现. 首先,双轴快排也是一种快排的优化方案,在JD ...

  • 美团面试:请手写一个快排,被我怼了!

    大家好,我是田维常,十年码农给你分享后端开发技术,记得关注我. 前面分享8篇,关于2017年,我去上海美团面试遇到的技术问题. 美团面试:熟悉哪些JVM调优参数,幸好我准备过! 美团面试:讲清楚MyS ...

  • 剑指offer计划5(查找算法中等版)---java

    剑指offer计划5(查找算法中等版)---java 1.1.题目1 剑指 Offer 04. 二维数组中的查找 1.2.解法 其实就是暴力解法的升级版,从最后一行开始判断,通过num当前的大小, 如 ...

  • 剑指offer计划7(搜索与回溯算法简单版)---java

    1.1.题目1 剑指 Offer 26. 树的子结构 1.2.解法 这题看了解法,感叹真的6,代码量减了很多. (A != null && B != null) && ...

  • 编程语言剑指offer计划28(搜索与回溯算法困难)---java

    1.1.题目1 剑指 Offer 37. 序列化二叉树 1.2.解法 这题给我笑死了,我看到题解有个解法,我愿称之为神. public class Codec { private TreeNode r ...

  • 【剑指Offer】数值的整数次方

    题目描述 给定一个double类型的浮点数base和int类型的整数exponent.求base的exponent次方. 保证base和exponent不同时为0 解法1 最直接的思路,计算base的 ...

  • 【剑指Offer】链表中倒数第k个结点

    题目描述 输入一个链表,输出该链表中倒数第k个结点. 解法 基本思路是使用两个辅助指针p, q,让p先走k - 1步后,p, q两个指针再一起走 这样当p指针走到链表的末尾时,q指针刚好走到的就是倒数 ...

  • 【剑指Offer】反转链表

    题目描述 输入一个链表,反转链表后,输出新链表的表头. 解法1 可以使用三个辅助指针pHead, last,next pHead记录当前节点,last记录上一个节点,next记录下一个节点 首先使用n ...

  • 剑指 Offer 14- I. 剪绳子

    我服了.动态规划杀我. 可以说一说解决动态规划的思路(只做了两三道就总结了emmm) 1.识别动态规划问题 --重叠子问题:大问题可以分为一个个子问题.和分治策略分割的子问题不同(分治问题的子问题是相 ...

  • 剑指offer

    03 数组中重复的数字 public int findRepeatNumber(int[] nums){ //排序后的数组,重复元素必然相邻 Arrays.sort(nums); //结果集 int ...

  • 剑指 Offer 30. 包含min函数的栈

    定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中,调用 min.push 及 pop 的时间复杂度都是 O(1). 示例:MinStack minStack = ne ...