如何使用由for循环创建的向量的向量来构建链表?

时间:2019-05-15 10:06:43

标签: c++ loops pointers linked-list

我试图将指针添加到我刚刚使用向量向量创建的2D矩阵中。我想使用的代码可以对数组进行处理,但是我想使用刚创建的向量的向量。所以我的问题是:

  1. 这可能吗?
  2. 如果是这样,我需要更改我想使用的代码吗?

下面是我创建向量向量的代码。 K是向量代表的房间数量,并且已经预先初始化。

    for (int i = 0; i < K; ++i) //a loop for each room
    {
        int M = 0; // initializing rows variable
        int N = 0; // initializing columns variable
        cin >> M >> N;

        vector<vector<int> > matrix(M); //give a matrix with a dimension M*N with all elements set to 0
        for (int i = 0; i < M; i++)
            matrix[i].resize(N);

        for (int i = 0; i < M; i++) //adding each row to the matrix
        {
            for (int j = 0; j < N; j++) //adding each column to the matrix
            {
                cin >> matrix[i][j]; //putting all the elements in the matrix
            }
        }

    }

如果可能的话,这里是我想使用的代码: https://www.geeksforgeeks.org/construct-linked-list-2d-matrix/

我是C ++的新手,如果这是一个荒谬的问题,我深表歉意。

1 个答案:

答案 0 :(得分:2)

如果替换原型,您可以这样做:

Node* construct(int arr[][3], int i, int j,  
                              int m, int n)
{
    //...
}

通过:

Node* construct(const vector<vector<int>> & arr, int i, int j,  
                                                 int m, int n)
{
    //...
}

这样,它应该可以工作,因为您可以使用operator[]访问vectors元素。

我希望它可以解决您的问题。


编辑:

为避免警告,您甚至可以写:

Node* construct(const vector<vector<int>> & arr, size_t i, size_t j,  
                                                 size_t m, size_t n)
{
    //...
}

EDIT2:完整的示例代码

我完全使用了您在问题中提供给我们的代码:

// CPP program to construct a linked list
// from given 2D matrix
#include <bits/stdc++.h>
using namespace std;

// struct node of linked list
struct Node {
    int data;
    Node* right, *down;
};

// returns head pointer of linked list
// constructed from 2D matrix
Node* construct(const vector<vector<int>> & arr, size_t i, size_t j,
                                                 size_t m, size_t n)
{
    // return if i or j is out of bounds
    if (i > n - 1 || j > m - 1)
        return nullptr;

    // create a new node for current i and j
    // and recursively allocate its down and
    // right pointers
    Node* temp = new Node();
    temp->data = arr[i][j];
    temp->right = construct(arr, i, j + 1, m, n);
    temp->down  = construct(arr, i + 1, j, m, n);
    return temp;
}

// utility function for displaying
// linked list data
void display(Node* head)
{
    // pointer to move right
    Node* Rp;

    // pointer to move down
    Node* Dp = head;

    // loop till node->down is not NULL
    while (Dp) {
        Rp = Dp;

        // loop till node->right is not NULL
        while (Rp) {
            cout << Rp->data << " ";
            Rp = Rp->right;
        }
        cout << "\n";
        Dp = Dp->down;
    }
}

// driver program
int main()
{
    // 2D matrix
    vector<vector<int>> arr = { {1, 2, 3}, {4, 5, 6}, {7, 8, 9} };

    size_t m = 3, n = 3;
    Node* head = construct(arr, 0, 0, m, n);
    display(head);
    return 0;
}

我通过更高效,更易读的向量初始化替换了您的代码。

我希望它会对您有所帮助:)