[Algorithm] 나누어 떨어지는 숫자 배열

문제 설명

array의 각 element 중 divisor로 나누어 떨어지는 값을 오름차순으로 정렬한 배열을 반환하는 함수, solution을 작성해주세요. divisor로 나누어 떨어지는 element가 하나도 없다면 배열에 -1을 담아 반환하세요.

 

문제 유형

• 배열 필터링
• 정렬

 

의사코드

1. priorty queue 자료형인 변수(pq)를 선언한다.
2. [반복문] array.size()만큼 반복한다.
   1. 만약 array의 요소가 나누어 떨어진다면
      1. pq에 요소를 담는다.
3. 만약 pq의 사이즈가 0이라면 배열에 -1을 담는다.
4. 그렇지 않다면 pq의 요소를 배열에 담는다.
5. 배열을 반환한다.

 

코드 작성

vector<int> solution(vector<int> arr, int divisor) {
    vector<int> result;
    priority_queue<int, vector<int>, greater<int>> pq;
    
    for (int num : arr)
    {
        if (num % divisor == 0)
            pq.push(num);
    }
    
    if (pq.size())
    {
        while (!pq.empty())
        {
            int num = pq.top();
            pq.pop();
            result.push_back(num);
        }
    }
    else
    {
        result.push_back(-1);
    }
    
    return result;
}

 

 

다른 풀이 비교

vector<int> solution(vector<int> arr, int divisor) {
    vector<int> answer;
    sort(arr.begin(), arr.end());

    for (int i = 0; i < arr.size(); i++)
    {
        if (arr[i] % divisor == 0)
            answer.push_back(arr[i]);
    }

    return answer.empty() ? vector<int>(1, -1) : answer;
}

• 내 코드는 탐색 > 필터 > 저장(정렬) > 체크 > 다른 자료형으로 저장 > 반환
• 위 코드는 정렬 > 탐색 > 필터 > 저장 > 체크 > 반환