leetcode之array

leetcode之Flipping an image

描述：

class Solution {
public:
    vector<vector<int>> flipAndInvertImage(vector<vector<int>>& A) 
    {
      for(int i=0;i<A.size();++i)
          reverse(A.begin(), A.end());//调换每一行的首尾元素
      for(int i=0;i<A.size();++i)
      {
          for(int j=0;j<A.size();++j)
          {
              if(A[i][j]=0)
                  A[i][j]=1;
              else
                  A[i][j]=0;
          }
      }
      return A;

    }
};

数组的转置

首先新建一个二维数组，行数为原数组的列数，列数为原数组的行数

class Solution {
public:
    vector<vector<int>> transpose(vector<vector<int>>& A) 
    {
        vector<vector<int>> res(A[0].size());//新建一个二维数组

        for(int i=0;i<res.size();++i)
        {
            res[i].resize(A.size());
            res[i].clear();要把原来数组的元素都清空
        }

        for(int i=0;i<A.size();++i)
        {
            for(int j=0;j<A[0].size();++j)
                res[j].push_back(A[i][j]);
        }
        return res;
    }
};