제출 #613773

#제출 시각아이디문제언어결과실행 시간메모리
613773AugustinasJucas도로 폐쇄 (APIO21_roads)C++14
7 / 100
57 ms10056 KiB
#include "roads.h"

#include <bits/stdc++.h>
using namespace std;
const int dydis = 1e5 + 10;
vector<pair<int, int> > gr[dydis];
int n, m;
vector<long long> minimum_closure_costs(int N, std::vector<int> U,
                                             std::vector<int> V,
                                             std::vector<int> W) {
    n = N; m = W.size();
    bool line = true, star = true;
    vector<int> mas;
    for(int i = 0; i < m; i++) {
        int a = U[i]; int b = V[i]; int w = W[i];
        if(a != b-1) line = false;
        if(a != 0) star = false;
        gr[a].push_back({b, w});
        gr[b].push_back({a, w});
        mas.push_back(w);
    }

    if(line) {
        /// Galiu praleisti max 1 is eiles. Kokia min suma?
        long long dp[n]; /// dp[i] - turiu visus iki i ir i-aji emiau. Kokia min suma?

        dp[0] = mas[0];
        dp[1] = mas[1];
        for(int i = 2; i < n; i++) {
            dp[i] = min(dp[i-1] + mas[i], dp[i-2] + mas[i]);
        }
        vector<long long> ret(n, 0);
        for(auto x : mas) ret[0] += x;
        ret[1] = min(dp[n-1], dp[n-2]);
        return ret;
    }
    if(star) {
        sort(mas.begin(), mas.end());
        reverse(mas.begin(), mas.end());
        vector<long long> ret(n, 0);
        for(auto x : mas) ret[0] += x;
        long long sum = 0;
        for(int i = n-1; i > 0; i--) {
            ret[i] = sum;
            sum += mas[i];
        }
        return ret;
    }
    return vector<long long>(N, 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...