망공부/코딩테스트
[프로그래머스] 주사위 게임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)가 됩니다.
재귀함수는 스택오버플로우를 발생시킬 수 있으며 메모리를 많이 차지하여 느리다고 하지만
이문제는 간단해서 그런지 속도차이는 거의 없더라고요!