【Leetcode】1424. Diagonal Traverse II

题目地址：

https://leetcode.com/problems/diagonal-traverse-ii/

给定一个list of list，每个list里都是整数。要求按照这样的顺序遍历：先将矩阵中的数按照横纵坐标之和是多少来分类，按照和小到和大的进行遍历。和相等的数中，按照从左下到右上的顺序进行遍历。题目保证每一个list都非空。

从下到上，每一行从左到右遍历，按横纵坐标之和分组存入一个哈希表。然后按横纵坐标之和从小到大遍历所有数。代码如下：

class Solution {
 public:
  vector<int> findDiagonalOrder(vector<vector<int>>& g) {
    unordered_map<int, vector<int>> mp;
    int max_sum = 0;
    for (int i = g.size() - 1; i >= 0; --i)
      for (int j = 0; j < g[i].size(); ++j) {
        mp[i + j].push_back(g[i][j]);
        max_sum = max(max_sum, i + j);
      }
    vector<int> res;
    for (int sum = 0; sum <= max_sum; ++sum)
      res.insert(res.end(), mp[sum].begin(), mp[sum].end());

    return res;
  }
};

时空复杂度$O(n)$，$n$为总共数字的数量。