(C++) - LeetCode (easy) 1337. The K Weakest Rows in a Matrix

https://leetcode.com/problems/the-k-weakest-rows-in-a-matrix/description/

LeetCode - The World's Leading Online Programming Learning Platform

정렬을 사용해본 문제였습니다.

📕 풀이방법

📔 입력 및 초기화

first는 행의 병사 수, second는 index를 저장할 vecotr pair soldiersCount를 선언해줍니다.

📔 풀이과정

1. mat의 각 행별 병사 수와 index를 구해 soldiersCount에 저장해줍니다.

2. 해당 vector를 sort해줍니다. first, second 순으로 우선순위로 두어 정렬됩니다.

3. 정답변수 ans에 soldiersCount k-1까지의 원소에서 second값을 저장해줍니다.

📔 정답 출력 | 반환

ans를 반환합니다.

---

📕 Code

📔 C++

using pii = pair<int,int>;
class Solution {
public:
    vector<int> kWeakestRows(vector<vector<int>>& mat, int k) {
        vector <pii> soldiersCount;
        for(int i = 0; i < mat.size(); i++) {
            int cnt = 0;
            for(int j = 0; j < mat[0].size(); j++) {
                if(mat[i][j]) cnt++;
            }
            soldiersCount.push_back({cnt, i});
        }
        sort(soldiersCount.begin(), soldiersCount.end());
        vector <int> ans;
        for (int i = 0; i < k; i++) {
            ans.push_back(soldiersCount[i].second);
        }
        return ans;
    }
};

---

*더 나은 내용을 위한 지적, 조언은 언제나 환영합니다.