C++partition_copy分区并复制

2026-04-01 14:25:25 1445阅读 0评论

C++中的`partition_copy`：分区并复制的高效工具

在编程的世界里，处理数据时总免不了需要将数据按照某种规则进行分类和处理。C++标准库为我们提供了一个非常强大的工具——std::partition_copy，它可以将容器中的元素根据指定的条件分为两部分，并将这两部分分别复制到不同的目标容器中。这篇文章将详细介绍如何使用std::partition_copy，以及它在实际应用中的各种应用场景。

基本概念

在开始之前，我们先了解一下std::partition_copy的基本概念。std::partition_copy是C++标准库中的一个算法，定义在头文件 <algorithm> 中。它的主要作用是将输入范围 [first, last) 中的元素分成两个子集，并将这些子集分别复制到两个不同的输出迭代器中。

std::partition_copy 的函数签名如下：

template< class InputIt, class OutputIt1, class OutputIt2, class UnaryPredicate >
std::pair<OutputIt1, OutputIt2> partition_copy( InputIt first, InputIt last,
                                                OutputIt1 d_first_true, OutputIt2 d_first_false,
                                                UnaryPredicate p );

InputIt first, InputIt last：输入范围的起始和结束迭代器。
OutputIt1 d_first_true, OutputIt2 d_first_false：两个输出迭代器，分别用于存放满足条件和不满足条件的元素。
UnaryPredicate p：一个一元谓词，用于判断元素是否应该被放在第一个输出容器中。

使用示例

下面是一个简单的示例，展示了如何使用std::partition_copy来将一个向量中的正数和负数分别复制到两个不同的向量中。

#include <iostream>
#include <vector>
#include <algorithm>

int main() {
    std::vector<int> numbers = {1, -2, 3, -4, 5};
    std::vector<int> positive;
    std::vector<int> negative;

    auto result = std::partition_copy(numbers.begin(), numbers.end(),
                                        std::back_inserter(positive),
                                        std::back_inserter(negative),
                                        [](int n) { return n > 0; });

    std::cout << "Positive numbers: ";
    for (int num : positive) {
        std::cout << num << " ";
    }
    std::cout << std::endl;

    std::cout << "Negative numbers: ";
    for (int num : negative) {
        std::cout << num << " ";
    }
    std::cout << std::endl;

    return 0;
}

在这个示例中，我们使用了std::back_inserter来方便地将元素插入到向量的末尾。std::partition_copy会根据Lambda表达式[](int n) { return n > 0; }的返回值，将正数和负数分别复制到positive和negative向量中。