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

문제 설명

array의 각 element 중 divisor로 나누어 떨어지는 값을 오름차순으로 정렬한 배열을 반환하는 함수, solution을 작성해주세요.

divisor로 나누어 떨어지는 element가 하나도 없다면 배열에서 -1을 담아 반환하세요.

 

 

제한사항

• arr은 자연수를 담은 배열입니다.
• 정수 i, j에 대해 i!=j 이면 arr[i]!=arr[j] 입니다.
• divisor는 자연수입니다.
• array는 길이 1 이상인 배열입니다.

 

 

입출력 예

arr			divisor	return	설명
[5, 9, 7, 10]			5	[5, 10]	arr의 원소 중 5로 나누어 떨어지는 원소는 5와 10입니다. 따라서 [5, 10]을 리턴합니다.
[2, 36, 1, 3]			1	[1, 2, 3, 36]	arr의 모든 원소는 1으로 나누어 떨어집니다. 원소를 오름차순으로 정렬해 [1, 2, 3, 36]을 리턴합니다.
[3, 2, 6]			10	[-1]	3, 2, 6은 10으로 나누어 떨어지지 않습니다. 나누어 떨어지는 원소가 없으므로 [-1]을 리턴합니다.

 

 

문제 풀이

풀이 언어 : C++

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

vector<int> solution(vector<int> arr, int divisor) {
    vector<int> answer;
    
    // 나누어 떨어지는 수 저장하기
    for(int i=0; i<arr.size(); i++)
    {
        if(arr[i]%divisor==0) // 나누어 떨어진다면
        {
            answer.push_back(arr[i]); // answer 배열에 저장
        }
    }
    
    // 나누어 떨어지는 수가 없다면 -1 담기
    if(answer.size()==0)
    {
        answer.push_back(-1); // -1 요소 추가
    }
    else // 나누어 떨어지는 수가 있다면 정렬하기
    {
        sort(answer.begin(), answer.end()); // 오름차순 정렬
    }
    
    return answer;
}

for문을 이용하여 arr 배열을 쭉 살펴보며 divisior로 나누어 떨어지는 수만 조건을 걸어 answer에 저장한다. arr 배열의 크기만큼 반복문을 돌리고 반복문을 빠져나오면 answer에 들어온 값이 있는지 확인한다. answer 배열의 크기를 확인하여 크기가 0이라면 divisor로 나누어 떨어지는 element가 하나도 없다는 뜻이니 answer 배열에 -1을 추가한다. 아니라면 sort 함수를 이용해 answer 배열을 오름차순으로 정렬한다.

 

 

오답 노트

answer은 배열의 크기가 정해지지 않은 vector인데 -1을 추가하려고 answer[0]=-1;을 적어 오류가 발생했다. 벡터에서는 push_back을 사용하도록 하자.

 

 

문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/12910

프로그래머스