제출 #485384

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
485384		MOUF_MAHMALAT	도로 폐쇄 (APIO21_roads)	C++14	0 / 100	265 ms	22852 KiB

roads

#include "roads.h"
#include<bits/stdc++.h>
#define F first
#define S second
#define all(v) v.begin(),v.end()
using namespace std;
typedef long long ll;
ll n,dp[2009][2],k;
vector<vector<pair<ll,ll> > >v;
vector<ll>ans;
void dfs(ll d,ll p)
{
    dp[d][0]=dp[d][1]=0;
    vector<pair<ll,ll> >op;
    for(auto z:v[d])
        if(z.F!=p)
        {
            dfs(z.F,d);
            dp[z.F][0]+=z.S;
            op.push_back({dp[z.F][0]-dp[z.F][1],z.F});
        }
    sort(all(op));

    ll o=max(0LL,(ll)op.size()-k);
    for(ll i=0; i<o; i++)
        dp[d][0]+=dp[op[i].S][0];
    for(ll i=o; i<(ll)op.size(); i++)
        dp[d][0]+=dp[op[i].S][1];

    o=max(0LL,(ll)op.size()-k+1);
    for(ll i=0; i<o; i++)
        dp[d][1]+=dp[op[i].S][0];
    for(ll i=o; i<(ll)op.size(); i++)
        dp[d][1]+=dp[op[i].S][1];
}
vector<ll>minimum_closure_costs(int N,vector<int> U,vector<int> V,vector<int> W)
{
    n=N;
    v.resize(n);
    ans.resize(n);
    for(ll i=0; i<n-1; i++)
    {
        v[U[i]].push_back({V[i],W[i]});
        v[V[i]].push_back({U[i],W[i]});
        ans[0]+=W[i];
    }
    for(k=1; k<n; k++)
    {
        dfs(0,0);
        ans[k]=dp[0][0];
    }
    return ans;
}

• Subtask 1: 0 / 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