第七章 ALDS1_6_B:Partition 分割

小酷miki

于 2017-12-24 20:29:15 发布

阅读量354

点赞数

CC 4.0 BY-SA版权

分类专栏：挑战程序设计竞赛2 算法和数据结构挑战程序设计竞赛2 算法和数据结构

本文链接：https://blog.youkuaiyun.com/u013075699/article/details/78887029

挑战程序设计竞赛2 算法和数据结构同时被 2 个专栏收录

62 篇文章

订阅专栏

挑战程序设计竞赛2 算法和数据结构

62 篇文章

订阅专栏

本文介绍了一个简单的基准划分算法实现，该算法以数列中的最后一个元素作为基准，将数列中小于等于基准的元素放置在基准的左侧，大于基准的元素放置在右侧。通过示例代码演示了如何使用 C++ 实现这一算法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题链接

ALDS1_6_B:Partition

问题内容

以最后一个元素为基准（假设是 $A_r$ ），从后往前遍历，将数列A中小于等于 $A_r$ 的数放在 $A_r$ 的左边，大于 $A_r$ 的数放在 $A_r$ 的右边，并输出标记的 $A_r$

思路

模拟整个过程即可

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int MAX = 500000;

int A[MAX], n;
// 分割
int partition(int p, int r) {
    int x = A[r];
    int i = p - 1;
    for (int j = p; j < r; j++) {
        if (A[j] <= x) {
            i++;
            swap(A[i], A[j]);
        }
    }
    swap(A[i + 1], A[r]);
    return i + 1;
}
int main() {

    scanf("%d", &n);
    for (int i = 0; i < n; i++)
        scanf("%d", &A[i]);

    // 记录Ar的下标
    int p = partition(0, n - 1);

    for (int i = 0; i < n; i++) {
        if(i == p)
            printf("[%d]%c", A[i], i == n - 1 ? '\n' : ' ');
        else
            printf("%d%c", A[i], i == n - 1 ? '\n' : ' ');
    }
    return 0;
}