[프로그래머스] Summer/Winter Coding(~2018) 소수 만들기

 

🔥DAY +6

 

https://programmers.co.kr/learn/courses/30/lessons/12977?language=java 

코딩테스트 연습 - 소수 만들기

 

> 문제를 보고 싶다면 더보기 클릭!

문제 설명

주어진 숫자 중 3개의 수를 더했을 때 소수가 되는 경우의 개수를 구하려고 합니다. 숫자들이 들어있는 배열 nums가 매개변수로 주어질 때, nums에 있는 숫자들 중 서로 다른 3개를 골라 더했을 때 소수가 되는 경우의 개수를 return 하도록 solution 함수를 완성해주세요.

제한사항

• nums에 들어있는 숫자의 개수는 3개 이상 50개 이하입니다.
• nums의 각 원소는 1 이상 1,000 이하의 자연수이며, 중복된 숫자가 들어있지 않습니다.

---

입출력 예numsresult

[1,2,3,4]	1
[1,2,7,6,4]	4

입출력 예 설명

입출력 예 #1
[1,2,4]를 이용해서 7을 만들 수 있습니다.

입출력 예 #2
[1,2,4]를 이용해서 7을 만들 수 있습니다.
[1,4,6]을 이용해서 11을 만들 수 있습니다.
[2,4,7]을 이용해서 13을 만들 수 있습니다.
[4,6,7]을 이용해서 17을 만들 수 있습니다.

 

---

 

🔥 [로직]

for문을 돌려서 num[0], num[1], num[2]부터 시작해서 각각 하나씩 증가하는 방식으로 모든 케이스의 소수 여부를 확인하는 방식이다.

n이라는 숫자의 소수의 판별 방법은 2부터 n-1까지중 나누어 떨어지는 수가 없는 경우를 의미한다.

따라서 나머지값을 확인하는 용도의 for문을 한번 더 돌려서 나머지가 0이 아닌경우 더이상 확인할 필요가 없기 때문에 for문을 나오고 정답 카운트를 증가해준다!

 

🔥 [최종 코드]

class Solution {
    public int solution(int[] nums) {
        int answer = 0;

        for(int i = 0; i < nums.length; i++) {
            for(int j = i+1; j < nums.length; j++) {
                for(int k = j+1; k < nums.length; k++) {
                    int sum = nums[i]+nums[j]+nums[k];
                    boolean flag = true;
                    for(int p = 2; p < sum; p++) {
                        if((sum % p) == 0){
                            flag = false;
                            break;
                        }
                    }
                    if(flag)
                        answer++;
                }
            }
        }

        return answer;
    }
}

 

🔥 [소감]

for문이 4번이나 쓰여서 시간이 테스트케이스에서 시간초과가 발생할까봐 조마조마했는데..

다른사람들의 답안에서도 대체적으로 for문을 4번 쓰는 것 같다.

 

아무래도 라인이 짧아서 가독성이 그래도 나쁘진 않은 것 같은데(내기준엔 그렇다..) 그래도 소수 판별 부분을 따로 함수로 뺐으면 가독성 측면에서 더욱 좋지 않았을까 하는 아쉬움이 있다! 

---

 

코테 준비를 본격적으로 시작한지 얼마 되지 않아 코드에 미흡한 점이 많을 수 있습니다. 
부족한 점 피드백 주시면 앞으로의 포스팅에 반영하겠습니다! 봐주셔서 정말 감사합니다 :)
-budtree