如何在C++中扁平化向量或2D向量

如何在C++中扁平化向量或2D向量

给定一个向量的向量(2D向量),任务是扁平化这个2D向量。

示例:

输入: vector = [[1, 2, 3, 4], [5, 6], [7, 8]]

输出: 1 2 3 4 5 6 7 8

输入: vector = [[1, 2], [3], [4, 5, 6, 8]]

输出: 1 2 3 4 5 6 8

推荐:请先在{IDE}上尝试,再转到解决方案。

算法:

  1. 可以使用迭代器来扁平化2D向量。
  2. 在两个数组iStart和iEnd中存储每个向量的起始和结束迭代器。
  3. 创建一个 hasNext() 方法来检查向量是否有下一个元素。
  4. 如果 hasNext() 返回 true,则打印当前元素。

下面是上述方法的实现:

//C++ program to flatten a
// Vector of Vectors or 2D Vector

#include <bits/stdc++.h>
using namespace std;

// Class to flatten the 2d vector
class FlattenVector {

public:
    int n;

    vector<vector<int>::iterator> iStart;
    vector<vector<int>::iterator> iEnd;
    int currIndex;

    // Store ending and starting iterators.
    FlattenVector(vector<vector<int> >& v)
    {

        // Get the number
        // of rows in 2d vector
        n = v.size();
        currIndex = 0;
        iStart.resize(n);
        iEnd.resize(n);

        for (int i = 0; i < n; i++) {
            iStart[i] = v[i].begin();
            iEnd[i] = v[i].end();
        }
    }

    // Returns true if any element is left.
    bool hasNext()
    {
        for (int i = 0; i < n; i++) {
            if (iStart[i] != iEnd[i])
                return true;
        }
        return false;
    }

    int next()
    {
        // Vector at currIndex is printed,
        // increment currIndex.
        if (iStart[currIndex]
            == iEnd[currIndex]) {
            currIndex++;
            return next();
        }

        // Increment iterator
        // and return the value.
        else
            return *iStart[currIndex]++;
    }
};

// Driver code
int main()
{
    vector<vector<int> >
        v{ { 1, 2 },
        { 3 },
        { 4, 5, 6 },
        { 7, 8, 9, 10 } };
    FlattenVector iter(v);

    while (iter.hasNext())
        cout << iter.next() << " ";

    return 0;
}  

输出:

1 2 3 4 5 6 7 8 9 10

时间复杂度: O(N)

辅助空间: O(N)

Python教程

Java教程

Web教程

数据库教程

图形图像教程

大数据教程

开发工具教程

计算机教程