1049_기타줄_그리디 알고리즘

https://www.acmicpc.net/problem/1049

1049번: 기타줄

문제

  Day Of Mourning의 기타리스트 강토가 사용하는 기타에서 N개의 줄이 끊어졌다. 따라서 새로운 줄을 사거나 교체해야 한다. 강토는 되도록이면 돈을 적게 쓰려고 한다. 6줄 패키지를 살 수도 있고, 1개 또는 그 이상의 줄을 낱개로 살 수도 있다.

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

입력

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

출력

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

---

문제 이해

  적어도 N개의 끊어진 줄을 사기 위해 필요한 돈의 최솟값을 구하는 문제이다. 여기서 포인트는 적어도 N개이므로 최솟값만 된다면 N+K개를 구매해도 상관없다는 점이다.

문제 해결

N < 6 일 경우

  N * K < 6개 패키지 보다 싸다면 낱개로 구매를 선택하면 된다.

N >= 6 일 경우

  일단 min(K / 6개만큼의 6개 패키지 vs 낱개로 K개로 N / 6 * 6개), 이 계산을 진행후 남은 기타줄을 패키지로 살지 아니면 낱개로 살지를 또한번 min연산을 진행해주면 된다.

 

그리고 구지 벡터로 할필요 없고 최솟값을 반복문 안에서 계속 갱신해주면 공간메모리의 효율도 늘릴 수 있다.

 

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

using namespace std;

int n, m, set, one, sum;
vector<pair<int, int>> v1, v2;

bool compareSet(pair<int, int> a, pair<int, int> b) {
	return a.first < b.first;
}

bool compareOne(pair<int, int> a, pair<int, int> b) {
	return a.second < b.second;
}

int main(void) {
	cin >> n >> m;
	for(int i = 0; i < m; i++) {
		int x, y;
		cin >> x >> y;
		v1.push_back(make_pair(x, y));
		v2.push_back(make_pair(x, y));
	}
	sort(v1.begin(), v1.end(), compareSet);
	sort(v2.begin(), v2.end(), compareOne);
	
	if(n >= 6) {
		sum = min(v1[0].first * (n / 6), (n / 6) * 6 * v2[0].second);
		n -= 6 * (n / 6);
		set = min(v1[0].first, n * v2[0].second);
		sum += set;
	}
	else if(n < 6) {
		one = min(v1[0].first, n * v2[0].second);
		sum = one;
	}
	cout << sum << '\n';
	return 0;
}