백준 24954번 물약 구매 C++

출처 : https://www.acmicpc.net/problem/24954

 

백트래킹을 이용해서 물약을 구매하는 모든 순서를 구해서 순서에 맞춰서 가격을 줄여가면서 나온 동전 중에 가장 적게 나온 동전의 값을 구했다.

 

#include <iostream>
#include <string>
#include <vector>
#include <math.h>
#include <algorithm>
#include <utility>
#include <stack>
#include <queue>
#include <math.h>
#include <set>
#include <map>
#include <unordered_map>
#include <unordered_set>
#include <iomanip>

using namespace std;
using ll = long long;

int n;
vector<int> potions(11);
vector<int> products;
vector<bool> visited(11, false);
vector<vector<pair<int, int>>> discounts(11, vector<pair<int, int>>());

int ans = 987654321;

void BT() {
	if (products.size() >= n) {
		int coin = 0;
		vector<int> prices;
		prices.assign(potions.begin(), potions.end());
		for (auto a : products) {
			coin += prices[a];
			for (auto b : discounts[a]) {
				prices[b.first] = max(1, prices[b.first] - b.second);
			}
		}
		ans = min(ans, coin);
		return;
	}
	for (int i = 1; i <= n; i++) {
		if (visited[i]) continue;
		visited[i] = true;
		products.push_back(i);
		BT();
		visited[i] = false;
		products.pop_back();
	}
}

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	cin >> n;
	for (int i = 1; i <= n; i++) {
		cin >> potions[i];
	}
	for (int i = 1; i <= n; i++) {
		int p;
		cin >> p;
		for (int j = 0; j < p; j++) {
			int a, d;
			cin >> a >> d;
			discounts[i].push_back({ a,d });
		}
	}
	BT();
	cout << ans;
};