मान लीजिए हमारे पास nums नामक लिस्ट ऑफ लिस्ट है, हमें nums के सभी एलिमेंट्स को डायगोनल क्रम में दिखाना है।
इसलिए, यदि इनपुट है
तो आउटपुट होगा [1,6,2,8,7,3,9,4,12,10,5,13,11,14,15,16]
इसे हल करने के लिए, हम निम्न चरणों का पालन करेंगे:
एक ऐरे ret परिभाषित करें
एक 2D ऐरे v परिभाषित करें
i := 0 से शुरू करें, जब i < nums का आकार, i बढ़ाते हुए, करें -
j := 0 से शुरू करें, जब j < nums[i] का आकार, j बढ़ाते हुए, करें -
{ 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, ]
