如何用C++实现不同长度向量的对应元素相加？

首先明确：线性代数中的标准向量加法仅适用于长度相同的向量——对应位置元素相加，结果向量长度与原向量一致。如果向量长度不同，从数学定义上来说是不合法的操作。

但如果是工程场景下需要处理不同长度的向量，通常有两种通用处理策略：

策略1：截断到最短向量长度 #

只对所有向量共有的索引范围（从0到最短向量长度-1）内的元素相加，超出范围的元素直接丢弃，结果向量长度等于最短向量的长度。

策略2：补零到最长向量长度 #

将所有较短向量的末尾补0，使其长度与最长向量一致，再对所有位置的元素相加，结果向量长度等于最长向量的长度。

C++实现：支持任意数量、任意数值类型的向量加法 #

下面是两个模板函数，分别实现上述两种策略，同时支持任意数值类型（int、float、double等）和任意数量的向量：

实现策略1（截断到最短长度） #

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

template<typename T>
std::vector<T> add_vectors_truncate(const std::initializer_list<std::vector<T>>& vectors) {
    if (vectors.empty()) {
        return {};
    }

    // 找出所有向量中最短的长度
    size_t min_len = vectors.begin()->size();
    for (const auto& vec : vectors) {
        min_len = std::min(min_len, vec.size());
    }

    std::vector<T> result(min_len, 0);
    for (size_t i = 0; i < min_len; ++i) {
        T sum = 0;
        for (const auto& vec : vectors) {
            sum += vec[i];
        }
        result[i] = sum;
    }

    return result;
}

int main() {
    std::vector<int> v1 = {0, 7, 3, 4};
    std::vector<int> v2 = {1, 2, 4, 1, 6};
    std::vector<int> v3 = {5, 1};

    auto result = add_vectors_truncate({v1, v2, v3});
    for (int num : result) {
        std::cout << num << " ";
    }
    // 输出：6 10
    return 0;
}

实现策略2（补零到最长长度） #

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

template<typename T>
std::vector<T> add_vectors_pad_zero(const std::initializer_list<std::vector<T>>& vectors) {
    if (vectors.empty()) {
        return {};
    }

    // 找出所有向量中最长的长度
    size_t max_len = vectors.begin()->size();
    for (const auto& vec : vectors) {
        max_len = std::max(max_len, vec.size());
    }

    std::vector<T> result(max_len, 0);
    for (size_t i = 0; i < max_len; ++i) {
        T sum = 0;
        for (const auto& vec : vectors) {
            // 索引在向量范围内就加对应元素，否则加0（默认）
            if (i < vec.size()) {
                sum += vec[i];
            }
        }
        result[i] = sum;
    }

    return result;
}

int main() {
    std::vector<int> v1 = {0, 7, 3, 4};
    std::vector<int> v2 = {1, 2, 4, 1, 6};
    std::vector<double> v3 = {5.5, 1.2, 3.0};

    auto result = add_vectors_pad_zero({v1, v2, v3});
    for (double num : result) {
        std::cout << num << " ";
    }
    // 输出：6.5 10.2 10.0 5.0 6.0
    return 0;
}

额外提示 #

如果严格遵循线性代数规则，应该在函数中先检查所有向量长度是否一致，不一致则抛出异常或返回空向量，避免不符合数学定义的运算。比如可以在函数开头添加检查：

// 检查所有向量长度是否一致（可选，用于严格遵循数学定义）
size_t vec_len = vectors.begin()->size();
for (const auto& vec : vectors) {
    if (vec.size() != vec_len) {
        throw std::invalid_argument("All vectors must have the same length for standard vector addition.");
    }
}

内容的提问来源于stack exchange，提问作者aaries