leetcode

Solution to some Leetcode problems written in C++
git clone git://git.dimitrijedobrota.com/leetcode.git
Log | Files | Refs | README | LICENSE

1424.cpp (568B)


0 class Solution {
1 typedef tuple<int, int, int> rec_t;
3 public:
4 vector<int> findDiagonalOrder(const vector<vector<int>> &nums) const {
5 static rec_t arr[100001];
6 int n = 0;
7 for (int i = 0; i < nums.size(); i++) {
8 for (int j = 0; j < nums[i].size(); j++) {
9 arr[n++] = {-i - j, i, nums[i][j]};
10 }
11 }
12 sort(begin(arr), begin(arr) + n, greater<rec_t>());
13 vector<int> res;
14 for (int i = 0; i < n; i++)
15 res.push_back(get<2>(arr[i]));
16 return res;
17 }
18 };