제출 #1334288

#제출 시각아이디문제언어결과실행 시간메모리
1334288gohchingjaykIMO (EGOI25_imo)C++20
28 / 100
110 ms17172 KiB
#pragma GCC optimize("O3,inline")
#include <bits/stdc++.h>
 
using namespace std;
 
using ll = long long;
#define int ll

using ii = pair<int, int>;
using iii = pair<int, ii>;
using pvii = pair<vector<int>, ii>;
 
constexpr int MAXN = 20000 + 5;
constexpr int INF = 1e18 + 5;
constexpr int LOG = 20;

signed main() {
	ios_base::sync_with_stdio(false); cin.tie(nullptr); cout.tie(nullptr);

	int N, M, K; cin >> N >> M >> K;
	int ans = N * M;
	
	vector<pvii> inp(N);
	
	for (int i = 0; i < N; ++i) {
		vector<int> lol(M); for (int j = 0; j < M; ++j) cin >> lol[j];
		sort(lol.begin(), lol.end());
		int s = 0; for (int j = 0; j < M; ++j) s += lol[j];
		inp[i] = pvii{lol, ii{s, i}};
	}
	
	sort(inp.begin(), inp.end(), [&](const pvii &a, const pvii &b) {
		return a.second.first < b.second.first || a.second.first == b.second.first && a.second.second > b.second.second;
	});

	int prevMin = INF;
	for (int i = N-1; i >= 0; --i) {
		int maxRem = (!i ? inp[i].second.first : inp[i].second.first - inp[i-1].second.first);
		if (i && inp[i].second.second > inp[i-1].second.second) maxRem--;
		if (i != N-1 && inp[i+1].second.second > inp[i].second.second) prevMin--;
		
		vector<int> &v = inp[i].first;
		int s = inp[i].second.first;
		
		vector<bitset<105>> dp_prev(maxRem + 1);
		vector<bitset<105>> dp_curr(maxRem + 1);

		dp_curr[0].set(0);
		
		for (int j = 0; j < M; ++j) {
			dp_prev = dp_curr;

			for (int k = v[j]; k <= maxRem; ++k) {
				dp_curr[k] |= dp_prev[k - v[j]] << 1;
			}
		}
		
		[&]() {
			for (int f = M; f >= 0; --f) {
				for (int k = maxRem; k >= 0; --k) {
					if (dp_curr[k][f] && s - k + f * K <= prevMin) {
						ans -= f;
						prevMin = s - k;
						return;
					}
				}
			}
			
			prevMin = s;
		}();
	}
	
	cout << ans;
}

/*
2 2 5
3 3
0 0
*/
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...