[백준] 2217번 : 로프 자바(Java)

[백준] 2217번 로프 : https://www.acmicpc.net/problem/2217

문제

N(1 ≤ N ≤ 100,000)개의 로프가 있다. 이 로프를 이용하여 이런 저런 물체를 들어올릴 수 있다. 각각의 로프는 그 굵기나 길이가 다르기 때문에 들 수 있는 물체의 중량이 서로 다를 수도 있다. 하지만 여러 개의 로프를 병렬로 연결하면 각각의 로프에 걸리는 중량을 나눌 수 있다. k개의 로프를 사용하여 중량이 w인 물체를 들어올릴 때, 각각의 로프에는 모두 고르게 w/k 만큼의 중량이 걸리게 된다. 각 로프들에 대한 정보가 주어졌을 때, 이 로프들을 이용하여 들어올릴 수 있는 물체의 최대 중량을 구해내는 프로그램을 작성하시오. 모든 로프를 사용해야 할 필요는 없으며, 임의로 몇 개의 로프를 골라서 사용해도 된다.

입력

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

출력

첫째 줄에 답을 출력한다.

접근방식 및 풀이

처음에는 "각각의 로프에는 모두 고르게 w/k 만큼의 중량이 걸리게 된다."라는 문구를 보고 중량의 최댓값과, 중량의 최솟값*N을 비교하여 푸는 문제인 줄 알았다. 예를들어 로프의 개수는 3개이며, 10,25,35의 중량을 버틸 수 있는 로프가 있다고 가정했을 때, 30과 35중에 더 큰 중량인 35이 정답이라고 생각헸다. 하지만 계속되는 오답에 문제를 자세히 보니 " 모든 로프를 사용해야 할 필요는 없으며, 임의로 몇 개의 로프를 골라서 사용해도 된다."라는 문구가 있었다. 즉, 10,25,35의 중량의 로프가 있다면 25,35 중량의 로프가 각 25중량만큼 들어서 합 50중량을 들 수 있다는 것이다. 문제를 이해하고 나니 아래의 간단한 최댓값 구하는 로직을 이용하여 문제를 풀이할 수 있었다.

max<arr[i]*(n-i)

전체 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        int n = Integer.parseInt(st.nextToken());
        int arr[] = new int[n];


        for (int i=0;i<n; i++){
            st = new StringTokenizer(br.readLine());
            arr[i] =Integer.parseInt(st.nextToken());
        }

        Arrays.sort(arr);//오름차순 정렬
        int max = arr[n-1];
        for (int i=0; i<n;i++){
            if (max<arr[i]*(n-i)){
                max = arr[i]*(n-i);
            }
        }
        

        System.out.println(max);

    }
    }