如何遍历对N维整数数组元素任意加1的所有组合并调用函数

针对你需要遍历所有通过对数组a的任意数量元素加1得到的数组，并调用函数f的需求，这里提供两种高效的实现方式，均基于编译期已知的数组维度N（模板参数）。

核心思路 #

每个数组元素有两种状态：保持原值、加1。总共有2^N种组合（包括所有元素都不加1的原数组），我们需要枚举所有这些组合并逐一调用f。

方法一：位掩码枚举（简洁高效） #

利用整数的二进制位作为掩码，每一位对应数组的一个元素，标记该元素是否需要加1。代码实现如下：

#include <array>

template<int N>
void f(std::array<int, N> a) {
    // 这里是你的业务逻辑
}

template<int N>
void loop(std::array<int, N> a) {
    const int total_combinations = 1 << N; // 编译期计算2^N
    for (int mask = 0; mask < total_combinations; ++mask) {
        std::array<int, N> b = a;
        for (int i = 0; i < N; ++i) {
            // 检查掩码第i位是否为1，是则对应元素加1
            if (mask & (1 << i)) {
                b[i] += 1;
            }
        }
        f(b);
    }
}

int main() {
    std::array<int, 2> a = {0, 1};
    loop(a); // 会调用f({0,1}), f({1,1}), f({0,2}), f({1,2})
    return 0;
}

代码说明

• 1 << N在编译期计算出总组合数，无运行时开销；
• 每个mask对应一种组合：比如N=2时，mask从0到3，二进制分别为00、01、10、11，正好对应你需要的4种数组；
• 遍历掩码的每一位，判断是否要对对应元素加1，生成目标数组后调用f。

方法二：递归枚举（逻辑直观） #

通过递归逐个处理数组元素，每个元素分“不加1”和“加1”两种情况递归，直到处理完所有元素后调用f：

#include <array>

template<int N>
void f(std::array<int, N> a) {
    // 这里是你的业务逻辑
}

template<int N>
void loop_recursive(std::array<int, N> a, int pos = 0) {
    if (pos == N) {
        f(a);
        return;
    }
    // 情况1：当前元素不加1，递归处理下一个元素
    loop_recursive(a, pos + 1);
    // 情况2：当前元素加1，递归处理下一个元素
    a[pos] += 1;
    loop_recursive(a, pos + 1);
}

template<int N>
void loop(std::array<int, N> a) {
    loop_recursive(a);
}

int main() {
    std::array<int, 2> a = {0, 1};
    loop(a); // 同样会生成你需要的4种调用
    return 0;
}

代码说明

• 递归从第0个元素开始，每个分支对应元素的两种状态；
• 当递归到pos == N时，所有元素状态已确定，调用f；
• 编译期已知递归深度N，编译器会进行优化，无需担心栈溢出问题。

两种方法对比 #

• 位掩码方式：代码更简洁，运行时效率略高，适合熟悉位运算的场景；
• 递归方式：逻辑更直观，容易理解和维护，无需处理位运算细节。

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