#include <bits/stdc++.h>
using namespace std;
int k,n;
int p[501][7];
struct Partition{
vector<int> best,choice;
long long cost;
Partition(vector<int> A):choice(A){
vector<int> tmp(k,0);
int cnt = 0;
for(int i = 0;i<n;i++){
if(choice[i]==1){
cnt++;
best.push_back(i);
for(int j = 0;j<k;j++){
tmp[j] = max(tmp[j],p[i][j]);
}
}
}
vector<int> tt = choice;
if(cnt>k){
cost = INT_MIN;
return ;
}
while(cnt<k){
int mx = -1 , id = 0;
for(int i = 0;i<n;i++){
if(tt[i]==0&&mx<p[i][cnt]){
mx = p[i][cnt];
id = i;
}
}
if(mx==-1){
cost = INT_MIN;
return ;
}
tt[id] = 1;
best.push_back(id);
for(int i = 0;i<k;i++){
tmp[i] = max(tmp[i],p[id][i]);
}
cnt++;
}
cost = accumulate(tmp.begin(),tmp.end(),0);
}
bool operator <(Partition r) const{
return cost < r.cost;
}
};
signed main(){
ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0);
int c;cin>>n>>k>>c;
for(int i = 0;i<n;i++){
for(int j = 0;j<k;j++){
cin>>p[i][j];
}
}
priority_queue<Partition> q;
q.push(Partition(vector<int>(n, 0)));
/*
5 4 4
7 0 4 9
3 0 8 4
1 1 3 7
5 1 3 4
4 2 2 9
*/
while(c--){
auto w = q.top();q.pop();
cout<<w.cost<<"\n";
if(c==0){
cout<<w.cost<<"\n";
return 0;
}
for(int i = 0;i<k;i++){
if(w.choice[w.best[i]]==0){
vector<int> choice = w.choice;
choice[w.best[i]] = -1;
for(int j = 0;j<i;j++){
choice[w.best[j]] = 1;
}
q.push(Partition(choice));
}
}
}
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
9 ms |
468 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
8 ms |
724 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
8 ms |
468 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
9 ms |
468 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |