Submission #388266

#	Time	Username	Problem	Language	Result	Execution time	Memory
388266		leaked	Olympiads (BOI19_olympiads)	C++14	31 / 100	128 ms	262148 KiB

olympiads

#include <bits/stdc++.h>
//#pra
#define sz(x) (int)x.size()
#define vec vector
#define pb push_back
#define rall(x) x.rbegin(),x.rend()
#define all(x) x.begin(),x.end()
#define pii pair<int,int>
#define f first
#define s second
using namespace std;
const int N=3e2+3;
typedef unsigned long long ull;
auto rnd=bind(uniform_int_distribution<int>(1,1e9),mt19937(time(0)));
ull r[N];

signed main(){
    ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    int n,k,c;
    cin>>n>>k>>c;
    vec<vec<int>>a(n,vec<int>(k));
    vec<vec<int>>p(k,vec<int>(n)),obr(n,vec<int>(k));
    for(int i=0;i<n;i++){
        for(int j=0;j<k;j++){
            cin>>a[i][j];
        }
    }
    for(int i=0;i<k;i++){
        auto cmp=[&](int x,int y){
            return a[x][i]>a[y][i];
        };
        iota(all(p[i]),0);
        sort(all(p[i]),cmp);
        for(int j=0;j<n;j++){
            obr[p[i][j]][i]=j;
        }
    }
    auto func=[&](array<int,6> &vc){
//        vec<int>vl(k,0);
        int s=0;
        for(int i=0;i<k;i++){
            int mx=0;
            for(int j=0;j<k;j++){
                mx=max(mx,a[vc[j]][i]);
            }
            s+=mx;
        }
        return s;
    };
    priority_queue<pair<int,array<int,6>>>pq;
    /// to push;

    {
        set<int>st;
        for(int i=0;i<k;i++) st.insert(p[i][0]);
        for(int i=0;i<n;i++){
            if(sz(st)<k) st.insert(i);
        }
        array<int,6>lel;
        for(int i=0;i<k;i++){
            lel[i]=*st.begin();
            st.erase(st.begin());
        }
        pq.push({func(lel),lel});
    }
    for(int i=0;i<n;i++) r[i]=rnd();
    map<ull,int>mp;
    while(!pq.empty()){
        auto v=pq.top();pq.pop();
        ull hs=0;
        for(int i=0;i<k;i++){
            hs+=r[v.s[i]];
        }
        if(mp.count(hs)) continue;
        mp[hs]=1;
        c--;
//        cerr<<v.f<<endl;
        if(c==0){
            cout<<v.f<<endl;
            return 0;
        }
        auto arr=v.s;
        for(int chng=0;chng<k;chng++){
            for(int w=0;w<k;w++){
                for(int j=0;j<k;j++){
                    if(obr[arr[w]][j]==n-1) continue;
                    int nw=p[j][obr[arr[w]][j]+1];
                    bool ok=1;
                    for(int t=0;t<k;t++){
                        if(t!=chng && arr[t]==nw) ok=0;
                    }
                    if(ok){
                        int prv=arr[chng];
                        arr[chng]=nw;
                        pq.push({func(arr),arr});
                        arr[chng]=prv;
                    }
                }
            }
        }
    }
    assert(false);
    return 0;
}

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