C++의 대각선 횡단 II

nums라는 목록 목록이 있다고 가정하고 nums의 모든 요소를 ​​대각선 순서로 표시해야 합니다.

따라서 입력이 다음과 같으면

그러면 출력은 [1,6,2,8,7,3,9,4,12,10,5,13,11,14,15,16]

이 문제를 해결하기 위해 다음 단계를 따릅니다. −

• ret 배열 정의

• 하나의 2D 배열 정의 v

• initialize i :=0의 경우, i <숫자의 크기일 때 업데이트(i를 1만큼 증가), 수행 -

  • j 초기화의 경우:=0, j

    • v

      끝에 { nums[i, j], i, j } 삽입

• 배열 v

  정렬

• v, do,

  의 각각에 대해

  • ret

    끝에 [0] 삽입

• 리턴 렛

예시

이해를 돕기 위해 다음 구현을 살펴보겠습니다. −

#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, ]