[알고리즘]_그리디 알고리즘_로프 (백준 No.2217)

yerim·2023년 2월 18일
post-thumbnail

문제

N(1 ≤ N ≤ 100,000)개의 로프가 있다. 이 로프를 이용하여 이런 저런 물체를 들어올릴 수 있다. 각각의 로프는 그 굵기나 길이가 다르기 때문에 들 수 있는 물체의 중량이 서로 다를 수도 있다.

하지만 여러 개의 로프를 병렬로 연결하면 각각의 로프에 걸리는 중량을 나눌 수 있다. k개의 로프를 사용하여 중량이 w인 물체를 들어올릴 때, 각각의 로프에는 모두 고르게 w/k 만큼의 중량이 걸리게 된다.

각 로프들에 대한 정보가 주어졌을 때, 이 로프들을 이용하여 들어올릴 수 있는 물체의 최대 중량을 구해내는 프로그램을 작성하시오. 모든 로프를 사용해야 할 필요는 없으며, 임의로 몇 개의 로프를 골라서 사용해도 된다.

입력
첫째 줄에 정수 N이 주어진다.
다음 N개의 줄에는 각 로프가 버틸 수 있는 최대 중량이 주어진다.
이 값은 10,000을 넘지 않는 자연수이다.

출력
첫째 줄에 답을 출력한다.

코드

import java.util.ArrayList;
import java.util.Collections;
import java.util.Scanner;

public class Main {
	public static void main(String[] args) {
		Scanner sc=new Scanner(System.in);

		int N=sc.nextInt();
		ArrayList<Integer> rope=new ArrayList<>();
		for(int i=0;i<N;i++){
			rope.add(sc.nextInt());
		}
		Collections.sort(rope);

		int count=N;
		int max = rope.get(N-1);

		for(int i=0;i<N;i++) {
			if (rope.get(i)*count>max) {
				max=rope.get(i)*count;
				
			}
			count--;
		}
		System.out.println(max);


	}
}

💡 풀이

  • 먼저 count 변수에는 로프의 개수인 N을 대입한다.

  • max에는 가장 무거운 밧줄인 rope.get(N-1)을 대입한다.

  • 0부터 차례로(가장 가벼운 밧줄부터) rope.get(N-1) (밧줄 무게) * count(남은 밧줄의 개수)를 계산해 max보다 크면 max의 값을 바꾼다.

  • count--로 남은 밧줄의 개수를 조절한다.

  • 이렇게 모든 밧줄의 경우의 수를 계산해 가장 큰 수가 답이다.

🧐 마무리

처음에 문제내용이 좀 이해가 안가서 헷갈렸었다. 껄껄
그담에 나는 경우의 수가 가장 가벼운 밧줄 * 밧줄개수, 가장 무거운 밧줄 이렇게만 생각하고 코드를 짰었고 테스트 코드도 통과해서 제출했는데 자꾸 오류나서 다시 보니까 중간무게 이런 경우를 빼먹었어서 다시 코드짰었다...
그래도 아무것도 안보고 혼자 푼 것에 대해 뿌듯해 하며 ~

0개의 댓글