Распечатать все комбинации трехмерного вектора - PullRequest
Новая
       28

Распечатать все комбинации трехмерного вектора

0 голосов
/ 04 декабря 2018

Предположим, у меня есть vector<vector<vector<int> > > result.

Единственный размер, который я знаю перед этим, это внутренний и внешний вектор, которые имеют размер k.

, если я печатаю result Я получаю это (для k = 3): 

i = 0

0 1 2
3 4 5

i = 1

6 7 8
9 10 11
12 13 14

i = 2

15 16 17
18 19 20

Что мне нужно сделать, это напечатать каждую комбинацию k строк из каждого вектора векторов i.Другими словами, мне нужен следующий вывод: 

0 1 2
6 7 8
15 16 17

0 1 2
6 7 8
18 19 20

0 1 2
9 10 11
15 16 17

...

3 4 5
12 13 14
18 19 20

Надеюсь, я был уверен в желаемом выводе.Я пробовал тысячу разных циклов, пытаясь сохранить в другом vector<vector<int> >, но пока безуспешно.Я действительно потерян, и любая помощь будет сильно оценена.

Код для генерации вышеуказанного вывода приведен здесь:

(извините, я знаю, что это ужасный код, но он былближе всего я смог продемонстрировать свою проблему в коде MCVE) 

#include <iostream>
#include <vector>
using namespace std;

int main(){
    vector<vector<vector<int> > > result;
    int k = 3;
    vector<vector<int> > randomVectors;
    //I'll create seven random vectors
    //In my original problem, I don't have this number beforehand
    int number = 0;
    for(int i = 0; i < 7; i++){
        vector<int> temp;
        for(int j = 0; j < k; j++){
            temp.push_back(number);
            number++;
        }
        randomVectors.push_back(temp);
    }
    //Vector of vector to assign to "result"
    vector<vector<int> > randomVectors_0;
    randomVectors_0.push_back(randomVectors[0]);
    randomVectors_0.push_back(randomVectors[1]);
    vector<vector<int> > randomVectors_1;
    randomVectors_1.push_back(randomVectors[2]);
    randomVectors_1.push_back(randomVectors[3]);
    randomVectors_1.push_back(randomVectors[4]);
    vector<vector<int> > randomVectors_2;
    randomVectors_2.push_back(randomVectors[5]);
    randomVectors_2.push_back(randomVectors[6]);

    result.push_back(randomVectors_0);
    result.push_back(randomVectors_1);
    result.push_back(randomVectors_2);

    cout << "Printing the 3D vector" << endl;
    for(int i = 0; i < k; i++){
        cout << "i = " << i << endl << endl;
        for(int j = 0; j < result[i].size(); j++){
            for(int m = 0; m < k; m++){
                cout << result[i][j][m] << " ";
            }
            cout << endl;
        }
        cout << endl;
    }
    return 0;
}

Версия компилятора: gcc (tdm-1) 4.7.1

1 Ответ

0 голосов
/ 04 декабря 2018

Я бы сделал rows_to_print вектор, который начинается со всех 0.Затем, после цикла, оно будет увеличивать последнее значение на 1. Если это значение больше, чем размер последнего вектора, затем сбросить его на 0 и увеличить следующее значение вверх по списку и т. Д. - все готовоцикл, когда каждое значение в rows_to_print больше, чем размер каждого из векторов: 

void print_rows(std::vector<size_t> rows, std::vector<std::vector<std::vector<int>>> v) {
    for(size_t x = 0; x < v.size(); x++) {
        for(size_t y = 0; y < v.at(x).at(rows.at(x)).size(); y++) {
            std::cout << v.at(x).at(rows.at(x)).at(y) << ' ';
        }
        std::cout << std::endl;
    }
}

bool increment_rows(std::vector<size_t> &rows, std::vector<std::vector<std::vector<int>>> v) {
    if(!rows.size()) return false; //empty rows, BAD
    rows.at(rows.size() - 1)++;
    for(int x = rows.size() - 1; x >= 0; x--) {
        if(rows.at(x) >= v.at(x).size()) {
            if(x <= 0) { return false; } //first row is done, then we're done!
            rows.at(x-1)++; //increment previous row and set us back to 0 (overflow)
            rows.at(x) %= v.at(x).size();
        }
    }
    return true;
}


int main() {
    ...

    std::vector<size_t> rows_to_print(k, 0);
    print_rows(rows_to_print, result);
    while(increment_rows(rows_to_print, result)) {
        print_rows(rows_to_print, result);
    }
}

Смотрите его в действии здесь: ideone

...