Submission #1056031

#	Time	Username	Problem	Language	Result	Execution time	Memory
1056031		Muhammet	Road Closures (APIO21_roads)	C++17	5 / 100	2091 ms	10176 KiB

roads

#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;
    for(int i = 0; i < n-1; i++){
        v.push_back({w[i],{u[i],u1[i]}});
        s += w[i];
    }
    vector <ll> p(n), p1(n), v1;
    for(int i = 0; i < n; i++){
        p1[u[i]]++;
        p1[u1[i]]++;
    }
    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;
    }
    for(int i = 0; i < n; i++){
        // p = p1;
        p.clear();
        ll s1 = s;
        for(auto j : v){
            if(p[j.ss.ff] < i and p[j.ss.ss] < i){
                s1 -= j.ff;
                p[j.ss.ff]++;
                p[j.ss.ss]++;
            }
        }

        v1.push_back(s1);
    }
    return v1;
}

• Subtask 1: 5 / 5
• Subtask 2: 0 / 7
• Subtask 3: 0 / 14
• Subtask 4: 0 / 10
• Subtask 5: 0 / 17
• Subtask 6: 0 / 25
• Subtask 7: 0 / 22