[코딩테스트] 백준1049 기타줄 - C++,Python

✔문제

시간 제한 메모리 제한 제출 정답 맞은 사람 정답 비율
2 초 128 MB 14016 4475 3798 32.293%
문제
Day Of Mourning의 기타리스트 강토가 사용하는 기타에서 N개의 줄이 끊어졌다. 따라서 새로운 줄을 사거나 교체해야 한다. 강토는 되도록이면 돈을 적게 쓰려고 한다. 6줄 패키지를 살 수도 있고, 1개 또는 그 이상의 줄을 낱개로 살 수도 있다.

끊어진 기타줄의 개수 N과 기타줄 브랜드 M개가 주어지고, 각각의 브랜드에서 파는 기타줄 6개가 들어있는 패키지의 가격, 낱개로 살 때의 가격이 주어질 때, 적어도 N개를 사기 위해 필요한 돈의 수를 최소로 하는 프로그램을 작성하시오.

✔입력

첫째 줄에 N과 M이 주어진다. N은 100보다 작거나 같은 자연수이고, M은 50보다 작거나 같은 자연수이다. 둘째 줄부터 M개의 줄에는 각 브랜드의 패키지 가격과 낱개의 가격이 공백으로 구분하여 주어진다. 가격은 0보다 크거나 같고, 1,000보다 작거나 같은 정수이다.

✔출력

첫째 줄에 기타줄을 적어도 N개 사기 위해 필요한 돈의 최솟값을 출력한다.

💖문제 풀이

기타줄을 구매한 돈의 최솟값을 구하는 것이기 때문에 정렬을 이용하여 패키지의 최솟값과 낱개 가격의 최솟값을 구하였다. 패키지를 구매하는 것보다 낱개를 구매하는 것이 이득인 경우가 있기 때문에 이를 처리해주는 것이 중요하다.
i) 6개 구매했을 때 패키지 가격이 비싼 경우가 있는지 확인
ii) 패키지를 구매하고 남은 개수 만큼 패키지를 구매하는 것이 나은지 판단

💖C++ 코드

#include<iostream>
#include<vector>
#include<algorithm>

using namespace std;

int main() {

	int m, n; 
	cin >> n >> m; 
	vector <int> package_price; 
	vector <int> piece_price; 

	int buy_price = 0 ; 
	int package_num = 0 ; 

	//input 값 처리 
	for (int i = 0; i < m; i++) {
		int temp1, temp2;
		cin >> temp1 >> temp2; 

		package_price.push_back(temp1);
		piece_price.push_back(temp2);
	}

	//sort vector하여 min 값 구하기 
	sort(package_price.begin(), package_price.end());
	sort(piece_price.begin(), piece_price.end());

	if (package_price[0] < 6 * piece_price[0]) { // 6개 구매시 패키지 가격이 이득인 경우 
		package_num = n / 6; // 구매할 패키지의 수 
		n = n - package_num * 6; // 패키지 구매 기타줄 n에서 제외 
		buy_price = package_num * package_price[0]; // 패키지 구매 가격 계산 
	}

	
	if (n > 6) { 
		buy_price += n * piece_price[0];
	}
	else {
		if (package_price[0] < n * piece_price[0])
			buy_price += package_price[0];
		else
			buy_price += n * piece_price[0];
	}
	

	cout << buy_price << "\n";

	return 0;
}

Python 코드

n, m = input().split()
n = int(n)
m = int(m)

package = []
piece = []  
buy_price = 0 

for i in range(m): 
   a,b = input().split()
   a = int(a)
   b = int(b)
   package.append(a)
   piece.append(b)

package.sort()
piece.sort()

#패키지 구매가 이득인 경우 
if (package[0] < piece[0]*6): 
   package_num = int( n / 6 ) # 패키지 구매 가능 개수
   buy_price = package_num*package[0] # 패키지 구매 가격 계산
   n = n - package_num * 6 # 개수 갱신 

if(n > 6): #패키지 구매하지 않은 경우 모두 낱개로 
   buy_price = n * piece[0] 
else: # 패키지 구매로 6개 이하로 남은 경우 
   if(package[0]< n*piece[0]): 
       buy_price += package[0]
   else: 
       buy_price += n*piece[0]

print(buy_price)