Submission #920978

#	Time	Username	Problem	Language	Result	Execution time	Memory
920978		shoryu386	Kitchen (BOI19_kitchen)	C++17	29 / 100	95 ms	348 KiB

kitchen

#include <bits/stdc++.h>
using namespace std;
 
#define int long long
 
main(){
	ios_base::sync_with_stdio(0); cin.tie(0);
	
	int n, m, k; cin >> n >> m >> k;
	
	int buckets[n], balls[m];
	
	for (int x = 0; x < n; x++) cin >> buckets[x];
	for (int x = 0; x < m; x++) cin >> balls[x];
	
	#define IMP cout << "Impossible"; return 0;
	
	/*
	if (m == 1){
		//st1 pt 1
		if (k > 1){
			IMP;
		}
		
		int bucketsum = 0;
		for (int x = 0; x < n; x++){
			bucketsum += buckets[x];
		}
		
		if (bucketsum > balls[0]){
			IMP;
		}
		
		cout << balls[0] - bucketsum;
	}
	else if (m == 2){
		//st1 pt2
		if (k > 2){
			IMP;
		}
		else if (k == 2){
			
		}
		else{
			
		}
	}
	*/
	
	bool dead = false;
	for (int x = 0; x < n; x++){
		if (buckets[x] < k){
			dead = true;
		}
	}

	if (dead) {IMP;}
	
	if (m <= 15){
		
		int ans = LLONG_MAX/20;
		sort(balls, balls+m); 
		
		for (int bm = 1; bm < (1LL<<m); bm++){
			
			
			//if (__builtin_popcountll(bm) < k){
			//	continue;
			//}
			
			vector<int> chefs;
			
			for (int x = 0; x < m; x++){
				if (bm & (1<<x)){
					chefs.push_back(balls[x]);
				}
			}
			
			
			int chefsum = 0, bucketSum = 0;
			for (int x = 0; x < chefs.size(); x++) chefsum += chefs[x];
			for (int x = 0; x < n; x++) bucketSum += buckets[x]; 
			
			//can we support this?
			int needed[n]; for (int x = 0; x < n; x++) needed[x] = k;
			
			for (int x = ((int)chefs.size()-1); x > -1; x--){
				//for each chef, try and distribute
				for (int y = 0; y < n; y++){
					if (chefs[x] == 0){
						break;
					}
					if (needed[y] > 0) needed[y]--, chefs[x]--;
				}
			}
			
			bool works = true;
			for (int x = 0; x < n; x++){
				if (needed[x] != 0){
					works = false;
				}
			}
			
			if (works && chefsum >= bucketSum){
				ans = min(ans, chefsum - bucketSum);
			}
		}
		
		if (ans == LLONG_MAX/20) {IMP;}
		else cout << ans;
	}
	else if (k == 1){
		//subtask 3
		#define BSMAX 100000
		//300^2
		
		bitset<BSMAX> bs;
		bs[0] = 1;
		
		for (int x = 0; x < m; x++){
			bs |= (bs << balls[x]);
		}
		
		int bucketsum = 0;
		for (int x = 0; x < n; x++){
			bucketsum += buckets[x];
		}
		
		int ans = -1;
		for (int x = bucketsum; x < BSMAX; x++){
			if (bs[x]){
				ans = x-bucketsum;
				break;
			}
		}
		
		if (ans == -1) cout << "Impossible";
		else cout << ans;
	}
}

Compilation message (stderr)

kitchen.cpp:6:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
    6 | main(){
      | ^~~~
kitchen.cpp: In function 'int main()':
kitchen.cpp:81:22: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   81 |    for (int x = 0; x < chefs.size(); x++) chefsum += chefs[x];
      |                    ~~^~~~~~~~~~~~~~

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