資料結構
網路
關係型資料庫管理系統
作業系統
Java
MS Excel
iOS
HTML
CSS
Android
Python
C 程式設計
C++
C#
MongoDB
MySQL
Javascript
PHP
物理
化學
生物學
數學
英語
經濟學
心理學
社會研究
時尚研究
法律研究C++ 中的對角遍歷 II
假設我們有一個名為 nums 的列表列表，我們需要按對角線順序顯示 nums 中的所有元素。
因此，如果輸入如下所示：
那麼輸出將是 [1,6,2,8,7,3,9,4,12,10,5,13,11,14,15,16]
為此，我們將按照以下步驟操作：
定義一個數組 ret
定義一個二維陣列 v
初始化 i := 0 時，當 i < nums 的大小，更新（將 i 增加 1），操作：
初始化 j := 0 時，當 j < nums[i] 的大小，更新（將 j 增加 1），操作：
將 { nums[i, j], i, j } 插入 v 的末尾
對陣列 v 進行排序
對 v 中的每個 it，操作：
將 it[0] 插入 ret 的末尾
返回 ret
示例
讓我們看看以下實現以獲得更好的理解：
#include <bits/stdc++.h>
using namespace std;
void print_vector(vector<int> v){
cout << "[";
for(int i = 0; i<v.size(); i++){
cout << v[i] << ", ";
}
cout << "]"<<endl;
}
class Solution {
public:
static bool cmp(vector <int>& a, vector <int>& b ){
int sum1 = a[1] + a[2];
int sum2 = b[1] + b[2];
return sum1 == sum2 ? a[1] > b[1] : sum1 < sum2;
}
vector<int> findDiagonalOrder(vector& nums) {
vector<int> ret;
vector<vector<int> > v;
for (int i = 0; i < nums.size(); i++) {
for (int j = 0; j < nums[i].size(); j++) {
v.push_back({ nums[i][j], i, j });
}
}
sort(v.begin(), v.end(), cmp);
for (auto& it : v)
ret.push_back(it[0]);
return ret;
}
};
main(){
Solution ob;
vector<vector<int>> v = {{1,2,3,4,5},{6,7},{8},{9,10,11},{12,13,14,15,16}};
print_vector(ob.findDiagonalOrder(v));
}輸入
{{1,2,3,4,5},{6,7},{8},{9,10,11},{12,13,14,15,16}}輸出
[1, 6, 2, 8, 7, 3, 9, 4, 12, 10, 5, 13, 11, 14, 15, 16, ]
更新於：2020 年 11 月 17 日
119 次瀏覽
廣告