C++如何使用std::partition对容器进行分区_C++容器分区与std::partition使用

std::partition 用于按条件重排容器元素,使满足条件的在前半部分。它接受起始迭代器、结束迭代器和一元谓词,返回指向最后满足条件元素后位置的迭代器。

在C++中,std::partition 是一个非常实用的算法,用于将容器中的元素按照某个条件重新排列,使得满足条件的元素位于容器的前半部分,不满足条件的元素位于后半部分。它定义在 gorithm> 头文件中,适用于所有支持迭代器的序列容器。

std::partition 基本用法

std::partition 接受三个参数:起始迭代器、结束迭代器和一个一元谓词(返回 bool 的函数或 lambda)。它会重排元素,使谓词为 true 的元素移到前面,false 的移到后面。函数返回一个迭代器,指向最后一个满足条件的元素之后的位置。

示例代码:

#include 
#include 
#include 

int main() {
    std::vector vec = {1, 2, 3, 4, 5, 6, 7, 8};

    auto it = std::partition(vec.begin(), vec.end(), [](int n) {
        return n % 2 == 0; // 偶数放在前面
    });

    std::cout << "分区后:";
    for (auto i = vec.begin(); i != vec.end(); ++i) {
        if (i == it) std::cout << " | "; // 分隔符
        std::cout << *i << " ";
    }
    std::cout << std::endl;

    return 0;
}

输出可能为:8 2 6 4 | 5 3 7 1,其中偶数在前,奇数在后,中间由返回的迭代器 it 分隔。

与 std::stable_partition 的区别

std::partition 不保证相对顺序,性能较高;而 std::stable_partition 保持满足条件和不满足条件元素各自的原始顺序,但开销更大。

如果你需要保持顺序,比如保留输入中偶数出现的先后顺序,应使用 std::stable_partition

实际应用场景

常见用途包括:

  • 将正数和负数分开
  • 分离空字符串与非空字符串
  • 按对象状态分类(如激活/未激活)
  • 预处理数据以便后续操作(如快速筛选)

例如,分离正负数:

std::vector nums = {-1, 3, -5, 6, -2, 8};
auto bound = std::partition(nums.begin(), nums.end(), [](int x) {
    return x > 0;
});
// 此时正数在 [begin, bound),负数在 [bound, end)

注意事项

使用 std::partition 需注意以下几点:

  • 容器必须支持随机访问迭代器(如 vector、deque),list 容器建议使用其成员函数 list::partition
  • 谓词应尽量避免副作用,确保行为可预测
  • 分区后,仅保证分组正确,组内顺序不确定
  • 如果所有元素都满足条件,返回值为 end();都不满足则返回 begin()

基本上就这些。std::partition 是一个简洁高效的工具,适合在不需要完全排序的情况下对数据进行逻辑划分。掌握它能让你在处理容器时更加灵活高效。