566. Reshape the Matrix

简单题

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

        if(nums.size()==0||nums[0].size()==0)
            return nums;

        if(nums.size()*nums[0].size()!=r*c)
            return nums;

        vector<int> temp(c,0);
        vector<vector<int>> result(r,temp);

        int new_i = 0;
        int new_j = 0;
        for(int i=0;i<nums.size();i++)
        {
            for(int j=0;j<nums[i].size();j++)
            {
                //cout<<new_i<<" "<<new_j<<endl;
                result[new_i][new_j] = nums[i][j];
                new_j++;
                if(new_j%c==0)
                {
                    new_j=0;
                    new_i++;
                }

            }
        }

        return result;
    }
};