Submission #1056296

#TimeUsernameProblemLanguageResultExecution timeMemory
1056296MuhammetRoad Closures (APIO21_roads)C++17
5 / 100
44 ms7916 KiB
#include <bits/stdc++.h>
#include "roads.h"

using namespace std;

#define ll long long
#define ff first
#define ss second

vector<ll> minimum_closure_costs(int n, vector<int> u, vector<int> u1, vector<int> w) {
    vector <pair<ll,pair<ll,ll>>> v;
    ll s = 0;
    bool tr = 0;
    for(int i = 0; i < n-1; i++){
        v.push_back({w[i],{u[i],u1[i]}});
        if(u1[i] != u[i]+1) tr = 1;
        s += w[i];
    }
    vector <ll> v1;
    sort(v.rbegin(), v.rend());
    if(*max_element(u.begin(), u.end()) == 0){
        ll s2 = s;
        v1.push_back(s);
        for(int i = 0; i < n-1; i++){
            s2 -= (v[i].ff);
            v1.push_back(s2);
        }
        return v1;
    }
    if(tr == 0){
        v1.push_back(s);
        vector <pair<int,int>> v2;
        for(int i = 0; i < n-1; i++){
            v2.push_back({u[i],w[i]});
        }
        sort(v2.begin(), v2.end());
        vector <ll> p1(n+1,0), p2(n+1,0);
        for(int i = 0; i < n-1; i++){
            if(i > 0) p1[i] = p1[i-1], p2[i] = p2[i-1];
            if(i % 2 == 0) p2[i] += v2[i].ss;
            else p1[i] += v2[i].ss;
        }
        ll mx = max(p1[n-2],p2[n-2]);
        for(int i = 0; i < n-2; i++){
            mx = max(mx,p1[i]+p2[n-2]-p2[min(i+2,n-2)]);
            mx = max(mx,p2[i]+p1[n-2]-p1[min(i+2,n-2)]);
        }
        v1.push_back(s-mx);
        for(int i = 2; i < n; i++) v1.push_back(0);
        return v1;
    }
    return vector<ll>(n-1,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...