망공부/코딩테스트

[프로그래머스] 주사위 게임2 (Java)

moonmm 2023. 11. 21. 16:26

문제설명

 

조건문 사용하는 문제입니다.

 

 

풀이

 

1.   

class Solution {
    public int solution(int a, int b, int c) {
        int answer = 0;
        if(a==b && b==c){  //다 같을때 
            answer = (a+b+c)*(a*a+b*b+c*c)*(a*a*a+b*b*b+c*c*c);
        }else if(a==b || a==c ||b==c){ //두개씩만 같을때
            answer = (a + b + c) * (a * a + b * b + c * c);
        }else { //다 다를때
            answer = a+b+c;
        }
        return answer;
    }
    
}

 

 

 

2.

class Solution {
    public int solution(int a, int b, int c) {
    	int answer = 1;
        int count = 1;
        if(a == b || a == c || b == c) { //두개 같을때
        	count++;
        }

        if(a == b && b == c) { //다 같을때 
             count++;
        }
        for(int i = 1; i <= count; i++) {
             answer *= (pow(a,i)+pow(b,i)+pow(c,i));
        }
        return answer;
     }
    private int pow(int base, int exponent) {  //거듭제곱 재귀함수 사용
         if(exponent == 0) return 1;
         return base * pow(base, exponent -1);
    }
    
}

재귀함수를 사용하였습니다!

반복문안에서 재귀함수 사용하기

 if(exponent == 0) return 1;    // 재귀함수의 무한루프를 방지하기 위해 종료조건이 필요합니다

 

 

 

 

재귀함수에 대해 자세한 설명은

https://wildeveloperetrain.tistory.com/116

 

 

 

시간복잡도

1번 코드는 상수시간인 O(1)

 

 

2번 코드는 O(count^2)

재귀함수 pow에 영향을 받습니다.

루프의 반복횟수 count이고 각 반복에서 pow함수가 호출됩니다

pow()의 시간복잡도는 O(i)  여기서 i는 현재 반복의 인덱스입니다.

따라서 전체 루프의 시간 복잡도는 O(count * max_i)가 됩니다.

 

재귀함수는 스택오버플로우를 발생시킬 수 있으며 메모리를 많이 차지하여 느리다고 하지만

이문제는 간단해서  그런지 속도차이는 거의 없더라고요!