[프로그래머스] Java 코딩테스트 - 개미 군단 (그리디 알고리즘 문제)

yihyun·2025년 4월 5일

코딩테스트

목록 보기
46/105

개미 군단 (그리디 알고리즘 문제)

✅ 문제 설명

개미 군단이 사냥을 나가려고 합니다. 개미군단은 사냥감의 체력에 딱 맞는 병력을 데리고 나가려고 합니다. 장군개미는 5의 공격력을, 병정개미는 3의 공격력을 일개미는 1의 공격력을 가지고 있습니다. 예를 들어 체력 23의 여치를 사냥하려고 할 때, 일개미 23마리를 데리고 가도 되지만, 장군개미 네 마리와 병정개미 한 마리를 데리고 간다면 더 적은 병력으로 사냥할 수 있습니다. 사냥감의 체력 hp가 매개변수로 주어질 때, 사냥감의 체력에 딱 맞게 최소한의 병력을 구성하려면 몇 마리의 개미가 필요한지를 return하도록 solution 함수를 완성해주세요.

✅ 제한사항

hp는 자연수입니다.
0 ≤ hp ≤ 1000

🔽 소스코드 1

💻 풀이

  • 가장 큰 값을 먼저 나눠준 후 그 나머지 값을 hp에 저장해 점차 작은 값으로 나눠주는 방식으로 풀이

시간 0.01ms ~ 0.02ms

	public int solution(int hp) {
			
		int count = hp / 5;
		hp %= 5;
		
		count += hp / 3;
		hp %= 3;
		
		count += hp;
		
		return count;
	}

🔽 소스코드 2 (배열과 반복문으로 풀이)

💻 풀이

  • 배열에 수를 담아준 후 배열을 순회해준다.
  • 가장 큰 값부터 배열에 담아주어야 하며, 나눈 값과 나머지 값을 통해 결과를 return 해주었다.

시간 0.01ms ~ 0.02ms

	public int solution1(int hp) {
		
		int[] power = {5,3,1};
		int count = 0;
		
		
		for(int num : power) {
			count += hp / num;
			hp %= num;
		}
		
		
		return count;
	}
profile
개발자가 되어보자

0개의 댓글