Submission #584298

#	Time	Username	Problem	Language	Result	Execution time	Memory
584298		amunduzbaev	Olympiads (BOI19_olympiads)	C++17	13 / 100	1817 ms	6668 KiB

olympiads

#include "bits/stdc++.h"
using namespace std;
 
#define ar array
typedef int64_t ll;
//~ #define int ll

const int K = 7;
vector<ar<int, K>> dp[2][K];

signed main(){
	ios::sync_with_stdio(0); cin.tie(0);
	
	int n, k, c; cin>>n>>k>>c;
	dp[0][0].push_back({});
	
	auto add = [&](auto& a, auto& b){
		ar<int, K> res {};
		for(int j=1;j<K;j++){
			res[j] = max(a[j], b[j]);
			res[0] += res[j];
		} return res;
	};
	
	for(int i=1;i<=n;i++){
		ar<int, K> a {};
		for(int j=1;j<=k;j++){
			cin>>a[j];
		}
		
		int t = i & 1, p = !t;
		for(int j=0;j<=k;j++){
			dp[t][j].clear();
			dp[t][j].insert(dp[t][j].end(), dp[p][j].begin(), dp[p][j].end());
			if(j){
				for(auto x : dp[p][j - 1]){
					dp[t][j].push_back(add(x, a));
				}
				sort(dp[t][j].begin(), dp[t][j].end(), [&](auto& a, auto& b){
					return a[0] > b[0];
				});
				while((int)dp[t][j].size() > c * 5) dp[t][j].pop_back();
			}
		}
	}
	
	assert((int)dp[n & 1][k].size() >= c);
	cout<<dp[n & 1][k][c - 1][0]<<"\n";
}

• Subtask 1: 13 / 13
• Subtask 2: 0 / 31
• Subtask 3: 0 / 24
• Subtask 4: 0 / 32