创建N个嵌套for循环

时间:2015-05-17 18:27:49

标签: c++ algorithm for-loop recursion nested-loops

有没有办法创建表单的for循环

for(int i = 0; i < 9; ++i) {
    for(int j = 0; j < 9; ++i) {
    //...
        for(int k = 0; k < 9; ++k) { //N-th loop

在编译时不知道N.理想情况下,我试图找出一种循环通过数字向量的单独元素的方法,以便在用不同的数字替换一定数量的数字时创建每个可能的数字。

7 个答案:

答案 0 :(得分:9)

您可以使用递归代替基本条件 -

void doRecursion(int baseCondition){

   if(baseCondition==0) return;

   //place your code here

   doRecursion(baseCondition-1);
}  

现在您不需要在编译时提供baseCondition值。您可以在调用doRecursion()方法时提供它。

答案 1 :(得分:5)

这是一个很好的小类,可以通过基于范围的for循环迭代多索引:

#include<array>

template<int dim>
struct multi_index_t
{
    std::array<int, dim> size_array;
    template<typename ... Args>
    multi_index_t(Args&& ... args) : size_array(std::forward<Args>(args) ...) {}

    struct iterator
    {
        struct sentinel_t {};

        std::array<int, dim> index_array = {};
        std::array<int, dim> const& size_array;
        bool _end = false;

        iterator(std::array<int, dim> const& size_array) : size_array(size_array) {}

        auto& operator++()
        {
            for (int i = 0;i < dim;++i)
            {
                if (index_array[i] < size_array[i] - 1)
                {
                    ++index_array[i];
                    for (int j = 0;j < i;++j)
                    {
                        index_array[j] = 0;
                    }
                    return *this;
                }
            }
            _end = true;
            return *this;
        }
        auto& operator*()
        {
            return index_array;
        }
        bool operator!=(sentinel_t) const
        {
            return !_end;
        }
    };

    auto begin() const
    {
        return iterator{ size_array };
    }
    auto end() const
    {
        return typename iterator::sentinel_t{};
    }
};

template<typename ... index_t>
auto multi_index(index_t&& ... index)
{
    static constexpr int size = sizeof ... (index_t); 
    auto ar = std::array<int, size>{std::forward<index_t>(index) ...};
    return multi_index_t<size>(ar);
}

基本思想是使用一个包含多个dim索引的数组,然后实现operator++以适当地增加这些索引。

将其用作

for(auto m : multi_index(3,3,4))
{
    // now m[i] holds index of i-th loop
    // m[0] goes from 0 to 2
    // m[1] goes from 0 to 2
    // m[2] goes from 0 to 3
    std::cout<<m[0]<<" "<<m[1]<<" "<<m[2]<<std::endl;
}

<强> Live On Coliru

答案 2 :(得分:4)

您可以使用递归函数:

void loop_function(/*params*/,int N){
for(int i=0;i<9;++i){
    if(N>0) loop_function(/*new params*/,N-1);
}

这将递归调用loop_function N次,而每个函数将迭代调用loop_function

以这种方式编程可能有点困难,但它应该做你想做的事情

答案 3 :(得分:2)

您可以使用递归调用:

void runNextNestedFor(std::vector<int> counters, int index)
{
     for(counters[index] = 0; counters[index] < 9; ++counters[index]) {
       // DO
       if(index!=N)
          runNextNestedFor(counters, index+1);
     }
}

第一次称它为:

std::vectors<int> counters(N);
runNextNestedFor(counters, 0);

答案 4 :(得分:1)

我写了一些C ++ 11代码,为自己实现了一个N嵌套的for循环。以下是可用作单个.hpp导入的代码的主要部分(我将其命名为nestedLoop.hpp):

OriginalLength

以下是预期输出的示例:

#ifndef NESTEDLOOP_HPP
#define NESTEDLOOP_HPP
#include <vector>

namespace nestedLoop{

    class nestedLoop {
        public:
            //Variables
            std::vector<int> maxes;
            std::vector<int> idxes; //The last element is used for boundary control
            int N=0;
            int nestLevel=0;

            nestedLoop();
            nestedLoop(int,int);
            nestedLoop(int,std::vector<int>);

            void reset(int numberOfNests, int Max);
            void reset(int numberOfNests, std::vector<int> theMaxes);

            bool next();
            void jumpNest(int theNest);

        private:
            void clear();
    };

    //Initialisations
    nestedLoop::nestedLoop(){}

    nestedLoop::nestedLoop(int numberOfNests, int Max) {
        reset(numberOfNests, Max);
    }

    nestedLoop::nestedLoop(int numberOfNests, std::vector<int> theMaxes) {
        reset(numberOfNests,  theMaxes);
    }

    void nestedLoop::clear(){
        maxes.clear();
        idxes.clear();
        N = 0;
        nestLevel = 0;
    }

    //Reset the scene
    void nestedLoop::reset(int numberOfNests, int Max){
        std::vector<int> theMaxes;
        for(int i =0; i < numberOfNests; i++) theMaxes.push_back(Max);
        reset(numberOfNests, theMaxes);
    }

    void nestedLoop::reset(int numberOfNests, std::vector<int> theMaxes){
        clear();
        N = numberOfNests;

        maxes=theMaxes;

        idxes.push_back(-1);
        for(int i=1; i<N; i++) idxes.push_back(theMaxes[i]-1);
    }

    bool nestedLoop::next(){
        idxes[N-1]+=1;

        for(int i=N-1; i>=0; i--){
            if(idxes[i]>=maxes[i]) {
                idxes[i] = 0;

                if(i){ //actually, if i > 0 is needed
                    idxes[i-1] += 1;
                }else{
                    return false;
                }
            }else{
                nestLevel = i;
                break;
            }
        }
        return true;
    }

    void nestedLoop::jumpNest(int theNest){
        for(int i = N-1; i>theNest; i--) {
            idxes[i] = maxes[i]-1;
        }
    }
}
#endif // NESTEDLOOP_HPP

答案 5 :(得分:0)

我将在给出的示例代码上以面值取得OP,并假设要求的是一个通过任意基数为10的解决方案。 (我基于评论和#34;理想情况下,我试图想出一种方法来循环遍历数字向量的单独元素,以创建每个可能的数字&#34;。

此解决方案具有循环,该循环通过基数10中的数字向量计数,并将每个连续值传递给辅助函数(doThingWithNumber)。出于测试目的,我让这个助手只打印出数字。

#include <iostream>

using namespace std;

void doThingWithNumber(const int* digits, int numDigits)
{
    int i;
    for (i = numDigits-1; i>=0; i--)
        cout << digits[i];
    cout << endl;
}

void loopOverAllNumbers(int numDigits)
{
    int* digits = new int [numDigits];
    int i;
    for (i = 0; i< numDigits; i++) 
        digits[i] = 0;

    int maxDigit = 0;
    while (maxDigit < numDigits) {
        doThingWithNumber(digits, numDigits);
        for (i = 0; i < numDigits; i++) {
            digits[i]++;
            if (digits[i] < 10)
                break;
            digits[i] = 0;
        }
        if (i > maxDigit)
            maxDigit = i;
    }
}

int main()
{
    loopOverAllNumbers(3);
    return 0;
}

答案 6 :(得分:0)

我使用这个解决方案:

unsigned int dim = 3;
unsigned int top = 5;
std::vector<unsigned int> m(dim, 0);
for (unsigned int i = 0; i < pow(top,dim); i++)
{
    // What you want to do comes here 
    //      |
    //      |
    //      v
    // -----------------------------------
    for (unsigned int j = 0; j < dim; j++)
    {
        std::cout << m[j] << ",";
    }
    std::cout << std::endl;
    // -----------------------------------

    // Increment m
    if (i == pow(top, dim) - 1) break;
    unsigned int index_to_increment = dim - 1;
    while(m[index_to_increment] == (top-1)) {
        m[index_to_increment] = 0;
        index_to_increment -= 1;
    }
    m[index_to_increment] += 1;
}

它当然可以进行优化和调整,但它运行良好,您不需要将参数传递给递归函数。使用单独的函数来递增多索引:

typedef std::vector<unsigned int> ivec;
void increment_multi_index(ivec &m, ivec const & upper_bounds)
{
    unsigned int dim = m.size();
    unsigned int i = dim - 1;
    while(m[i] == upper_bounds[i] - 1 && i>0) {
        m[i] = 0;
        i -= 1;
    }
    m[i] += 1;
}

int main() {

    unsigned int dim = 3;
    unsigned int top = 5;
    ivec m(dim, 0);
    ivec t(dim, top);
    for (unsigned int i = 0; i < pow(top,dim); i++)
    {
        // What you want to do comes here 
        //      |
        //      |
        //      v
        // -----------------------------------
        for (unsigned int j = 0; j < dim; j++)
        {
            std::cout << m[j] << ",";
        }
        std::cout << std::endl;
        // -----------------------------------

        // Increment m
        increment_multi_index(m, t);
    }

}