[코딩테스트 Lv.0] 약수 구하기

문제 설명

정수 n이 매개변수로 주어질 때, n의 약수를 오름차순으로 담은 배열을 return하도록 solution 함수를 완성해주세요.

제한 사항

1 ≤ n ≤ 10,000

입출력 예

n	result
24	[1, 2, 3, 4, 6, 8, 12, 24]
29	[1, 29]

입출력 예 설명

입출력 예 #1
24의 약수를 오름차순으로 담은 배열 [1, 2, 3, 4, 6, 8, 12, 24]를 return합니다.

입출력 예 #2
29의 약수를 오름차순으로 담은 배열 [1, 29]를 return합니다.

---

내 풀이

Java

class Solution {
    public int[] solution(int n) {
        int cnt = 0;
        for(int i = 1; i <= n; i++){
            if(n % i == 0) cnt++;      
        }
        
        int[] answer = new int[cnt];
        
        cnt = 0;
        for(int i = 1; i <= n; i++){
            if(n % i == 0) answer[cnt++] = i;
        }
        
        return answer;
    }
}

C

#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>

int* solution(int n) {
    // return 값은 malloc 등 동적 할당을 사용해주세요. 할당 길이는 상황에 맞게 변경해주세요.
    int cnt = 0;
    for(int i = 1; i <= n; i++){
        if(n % i == 0) cnt++;
    }
    
    int* answer = (int*)malloc(sizeof(int)*cnt);
    
    cnt = 0;
    for(int i = 1; i <= n; i++){
        if(n % i == 0) answer[cnt++] = i;
    }
    
    return answer;
}

Python

def solution(n):
    answer = []
    
    for i in range(1, n + 1):
        if n % i == 0:
            answer.append(i)
    
    return answer

---

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

프로그래머스