[SWEA] D2 - 21425 번 | +=

🔍 문제 보러 가기

주어진 정수 A, B, N에 대해, 변수 x와 y가 각각 A와 B로 초기화된 상태에서 x += y 또는 y += x 연산을 반복적으로 수행하여, x나 y 중 하나의 값이 N을 초과할 때까지 최소 몇 번의 연산이 필요한지 계산하는 프로그램을 작성

내 코드

package D2;

// SWEA D2 21425번 "+=" 문제 풀이 

import java.util.*;

public class N21425 {
	public static void main(String[] args) {
		Scanner scan = new Scanner(System.in);
		
		int T = scan.nextInt();
		
		Sol sol = new Sol();
		
		for (int i = 0; i < T; i++) {
			int A = scan.nextInt();
			int B = scan.nextInt();
			int N = scan.nextInt();
			
			System.out.println(sol.solution(A, B, N));
		}
		
	}
}

class Sol {
		
	public int solution(int A, int B, int N) {
		
		int cnt = 0;
		
		while (true) {
			if (A >= B) {
				B = A + B;
				cnt++;
				if (B > N) {
					return cnt;
				}
				
			}else if(A < B) {
				A = A + B;
				cnt++;
				if (A > N) {
					return cnt;
				}
			}
		}		
	}
}

접근 방법

A와 B 중에서 더해지는 값을 선택할때 더 큰 값으로 선택해 더해야 += 연산을 최소로 수행해 N의 값을 초과시킬수 있을 것이라고 생각했다.

1. A와 B중에서 큰 값을 선택해 현재의 총합에 더한다.
2. 더하는 횟수를 저장한다.
3. 총합이 N보다 크다면 횟수를 리턴한다.

개선 코드

package D2;

//SWEA D2 21425번 "+=" 문제 풀이 개선  

import java.util.*;

public class N21425_b {
	public static void main(String[] args) {
		Scanner scan = new Scanner(System.in);
		
		int T = scan.nextInt();
		
		StringBuilder result = new StringBuilder();
		
		for(int i = 0; i < T; i++) {
			int A = scan.nextInt();
			int B = scan.nextInt();
			int N = scan.nextInt();
			
			result.append(solution(A, B, N)).append("\n");
		}
		
		System.out.print(result);
		scan.close();
	}
	
	public static int solution(int A, int B, int N) {
		int cnt = 0;
		
		while(Math.max(A, B) <= N) {
			if (A < B) {
				A += B;
			}else {
				B += A;
			}
			cnt++;
		}
		return cnt;
	}
}

Point

1. StringBuilder 사용

   결과를 StringBuilder에 누적해 System.out.print로 한번에 출력한다.
   이는 테스트케이스가 많을 때 입출력 속도를 개선해준다.

2. Math.max 사용

   이전 코드는 while(true) 코드로 반복문 내부에 조건문을 넣어서 반복 진행 여부를 결정했었는데 매 반복시에 Math.max를 사용해 A와 B 중 더 큰 값이 N을 넘는지 확인한다.
   이로써, 반복문 내부 조건문이 간결해졌고, 불필요한 else if 절을 제거할 수 있다.

3. 스캐너 자원 해제
   scan.close를 통해 스캐너 객체를 닫아준다.

4. 메모리 사용 최적화
   매번 새로운 객체를 생성하지 않고 solution 메서드를 static으로 선언해 사용해 메모리 관리를 더 쉽게 한다.

• scan.colse()를 해주는 이유