1. 程式人生 > >陣列學習9——使用vector對二維陣列進行初始化的理解

陣列學習9——使用vector對二維陣列進行初始化的理解

這個題目不難,但是這裡涉及了二維陣列的基礎知識,值得記錄。

什麼時候對二維陣列必須初始化在使用push_back的時候Push的不是一個數組(而是一個元素)的時候。

因為這個時候你知道怎麼分割槽!!所有的元素都是一股腦堆在一起。

*************************************************************************************************************************************************

class Solution {
public:
    vector<vector<int>> matrixReshape(vector<vector<int>>& nums, int r, int c) {
        vector<int> t;


        //對二維陣列一定要進行初始化

        //初始化一個r行,每行的資料是這樣的:vector<int>(c,0) 二維陣列。
       
// vector<vector<int>> newA(r,vector<int>(c,0));
        
        //對二維陣列進行初始化
       
vector<vector<int>> newA;
        newA.resize(r);
        for(int i=0;i<r;i++) newA[i].resize(c);


        int count=nums[0].size();
        if(count*nums.size()!=r*c)
        {
            return nums;
        }
        if(count*nums.size()==r*c)
        {
            //把原矩陣中的所有的元素存入到臨時容器t中


            for(int i=0;i<nums.size();i++)
            {
                for(int j=0;j<nums[0].size();j++)
                {
                    t.push_back(nums[i][j]);
                }
            }
            //開始填充到新矩陣
            for(int i=0;i<r;i++)
            {
                for(int j=0;j<c;j++)
                {
                    newA[i][j]=t[i*c+j];
                }
            }
            return newA;
        }
    }
};