24_大数据top k-快排分割

try my best c

已于 2022-05-31 20:50:32 修改

阅读量3.5k

点赞数

分类专栏： sl数据结构学习笔记文章标签： c++ 排序算法算法

于 2022-03-15 10:56:06 首次发布

本文链接：https://blog.csdn.net/m0_67778699/article/details/123497005

版权

sl数据结构学习笔记专栏收录该内容

18 篇文章 0 订阅

订阅专栏

理论讲解

在这里插入图片描述

代码实现

#include <iostream>
using namespace std;
// 快排分割函数
int Partation(int arr[], int begin, int end)
{
    int val = arr[begin];
    int i = begin;
    int j = end;

    while (i < j)
    {
        while(i < j&& arr[j] < val)
            j--;

        if (i < j)
        {
            arr[i] = arr[j];
            i++;
        }

        while (i < j && arr[i] > val)
            i++;

        if (i < j)
        {
            arr[j] = arr[i];
            j--;
        }
    }

    arr[i] = val;
    return i;
}

// 求top k的函数
void SelectTopK(int arr[], int begin, int end, int k)
{
    int pos = Partation(arr, begin, end);
    if (pos == k - 1)
    {
        return;
    }
    else if (pos > k - 1)
    {
        SelectTopK(arr, begin, pos - 1, k);
    }
    else
    {
        SelectTopK(arr, pos + 1, end, k);
    }
}

int main()
{
    int arr[] = { 64, 45, 52, 80, 66, 68, 0, 2, 18, 75 };
    int size = sizeof arr / sizeof arr[0];

    // 求值最小的前3个元素
    int k = 3;
    SelectTopK(arr, 0, size - 1, k);

    for (int i = 0; i < k; i++)
    {
        cout << arr[i] << " ";
    }
    cout << endl;
}