개발/소프티어
softeer lv2. 금고털이
bsorry
2023. 12. 29. 13:44
softeer: https://softeer.ai/practice/info.do?idx=1&eid=395
문제
루팡은 배낭을 하나 메고 은행금고에 들어왔다. 금고 안에는 값비싼 금, 은, 백금 등의 귀금속 덩어리가 잔뜩 들어있다. 배낭은 W ㎏까지 담을 수 있다.
각 금속의 무게와 무게당 가격이 주어졌을 때 배낭을 채울 수 있는 가장 값비싼 가격은 얼마인가?
루팡은 전동톱을 가지고 있으며 귀금속은 톱으로 자르면 잘려진 부분의 무게만큼 가치를 가진다.
제약조건
1 ≤ N ≤ 106인 정수
1 ≤ W ≤ 104인 정수
1 ≤ Mi, Pi ≤ 104인 정수
입력형식
첫 번째 줄에 배낭의 무게 W와 귀금속의 종류 N이 주어진다. i + 1 (1 ≤ i ≤ N)번째 줄에는 i번째 금속의 무게 Mi와 무게당 가격 Pi가 주어진다.
출력형식
첫 번째 줄에 배낭에 담을 수 있는 가장 비싼 가격을 출력하라.
입력예제1
100 2 90 1 70 2
출력예제1
170
자바 코드
import java.util.*;
import java.io.*;
public class Main
{
public static void main(String args[]) throws Exception
{
//Scanner sc = new Scanner(System.in);
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer stBr = new StringTokenizer(br.readLine()," ");
String[] tokens = new String[2];
int stBrIndex = 0;
while(stBr.hasMoreTokens()) {
tokens[stBrIndex] = stBr.nextToken();
stBrIndex++;
}
int bagWeight = Integer.parseInt(tokens[0]);
int jKinds = Integer.parseInt(tokens[1]);
int[][] jews = new int[jKinds][2];
int highCost;
for (int jew[] : jews) {
StringTokenizer stBr2 = new StringTokenizer(br.readLine()," ");
String[] tokens2 = new String[2];
int stBrIndex2 = 0;
while(stBr2.hasMoreTokens()) {
tokens2[stBrIndex2] = stBr2.nextToken();
stBrIndex2++;
}
jew[1] = Integer.parseInt(tokens2[0]);
jew[0] = Integer.parseInt(tokens2[1]);
}
//무게 당 값이 많이 나가는 보석부터 내림차순 정렬
Arrays.sort(jews, (o1,o2)->(o2[0]-o1[0]));
int inBag = 0;
int totalPrice = 0;
int prevInBag;
for (int jew[] : jews) {
//보석 담기
inBag += jew[1];
totalPrice += jew[1]*jew[0];
//현재 보석을 담고 가방 무게가 넘칠 경우 보석 빼고 남은 무게 만큼 보석 커팅 후 다시 보석 담기
if (inBag > bagWeight) {
inBag -= jew[1]; //보석 빼기
totalPrice -= jew[1]*jew[0]; //가격 빼기
int space = bagWeight - inBag;
inBag += space;
totalPrice += space*jew[0];
}
}
System.out.println(totalPrice);
}
}
ps. scanner로 값을 받으니 시간초과되어 실패했다. BufferedReader를 쓰는 습관을 길러야 할 듯.