Submission #746010

#	Time	Username	Problem	Language	Result	Execution time	Memory
746010		vjudge1	Kitchen (BOI19_kitchen)	C++17	29 / 100	282 ms	108332 KiB

kitchen

#include <bits/stdc++.h>
using namespace std;

const int MAXN = 301;
const int MAXS = MAXN*MAXN+1000;

int dp[MAXN][MAXS], n, m, k;

void solve() {
	vector<int> a(n+1), b(n+1);
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
	}
	for (int j = 1; j <= m; j++) {
		cin >> b[j];
	}
	
	int ans = INT_MAX;
	for (int bits = 0; bits < (1 << m); bits++) {
		int x = bits<<1;
		priority_queue<int> pq;
		for (int i = 1; i <= n; i++) {
			pq.push(a[i]);
		}
		multiset<int> chefs;
		for (int i = 1; i <= m; i++) {
			if ((x>>i)&1) {
				chefs.insert(b[i]);
			}
		}

		bool possible = true;
		while (!pq.empty() && !chefs.empty()) {
			bool good = false;
			int left = pq.top();
			pq.pop();
			if (chefs.size() < k || left < k) {
				possible = false;
				break;
			}
			while (left && !chefs.empty()) {
				vector<int> used;
				for (int u : chefs) {
					if (left) {
						left--;
						used.push_back(u);
					}
				}
				for (int u : used) {
					chefs.erase(chefs.find(u));
					if (u != 1) chefs.insert(u-1);
				}
			}
			if (left) {
				possible = false;
				break;
			}
		}
		if (possible && pq.empty()) {
			int res = 0;
			for (int u : chefs) res += u;
			ans = min(ans, res);
		}

	}
	if (ans == INT_MAX)
		cout << "Impossible\n";
	else
		cout << ans << "\n";
}

int main() {
	ios::sync_with_stdio(0); cin.tie(0);
	cin >> n >> m >> k;
	if (k != 1) {
		solve();
		return 0;
	}
	vector<int> a(n+1), b(m+1);
	int sum = 0;
	for (int i = 1; i <= n; i++) {
		cin >> a[i]; sum += a[i];
	}
	for (int j = 1; j <= m; j++) {
		cin >> b[j];
	}
	
	dp[0][0] = true;
	for (int i = 1; i <= m; i++) {
		for (int j = 0; j < MAXS; j++) {
			dp[i][j] = dp[i-1][j];
		}
		for (int j = b[i]; j < MAXS; j++) {
			if (dp[i-1][j-b[i]]) {
				dp[i][j] = true;
			}
		}
	}
	for (int i = 0; i < MAXS; i++) {
		if (dp[m][i]) cerr << i << " ";
	}
	for (int i = sum; i < MAXS; i++) {
		if (dp[m][i]) {
			cout << i-sum << "\n";
			return 0;
		}
	}
	cout << "Impossible\n";
	return 0;
}

Compilation message (stderr)

kitchen.cpp: In function 'void solve()':
kitchen.cpp:37:21: warning: comparison of integer expressions of different signedness: 'std::multiset<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   37 |    if (chefs.size() < k || left < k) {
      |        ~~~~~~~~~~~~~^~~
kitchen.cpp:34:9: warning: unused variable 'good' [-Wunused-variable]
   34 |    bool good = false;
      |         ^~~~

• Subtask 1: 9 / 9
• Subtask 2: 0 / 22
• Subtask 3: 20 / 20
• Subtask 4: 0 / 21
• Subtask 5: 0 / 28