Submission #1089717

#TimeUsernameProblemLanguageResultExecution timeMemory
1089717underwaterkillerwhale도로 폐쇄 (APIO21_roads)C++17
24 / 100
2094 ms24780 KiB
#include <bits/stdc++.h>
#define ll              long long
#define pii             pair<int,int>
#define pll             pair<ll,ll>
#define rep(i,m,n)      for(int i=(m); i<=(n); i++)
#define reb(i,m,n)      for(int i=(m); i>=(n); i--)
#define iter(id, v)     for(auto id : v)
#define fs              first
#define se              second
#define MP              make_pair
#define pb              push_back
#define bit(msk, i)     ((msk >> i) & 1)
#define SZ(v)           (ll)v.size()
#define ALL(v)          v.begin(),v.end()

using namespace std;

mt19937_64 rd(chrono :: steady_clock :: now ().time_since_epoch().count());
ll Rand (ll l, ll r) { return uniform_int_distribution<ll> (l, r) (rd); }

const int N = 2e5 + 7;
const int Mod = 1e9 + 2022; ///loonf mod sai
const ll INF = 1e14;
const ll BASE = 137;
const int szBL = 350;

struct edges {
    int u, v, w;
};

int n;
vector<int> edges;
vector<pii> ke[N];

namespace sub4 {
    int Wp[N];
    ll dp[N][2];

    void dfs (int u, int p, int X) {
        vector<ll> vec;
        ll Tot = 0;
        iter (&id, ke[u]) {
            int v, w;
            tie(v, w) = id;
            if (v != p) {
                Wp[v] = w;
                dfs(v, u, X);
                Tot += dp[v][0];
                vec.pb({-dp[v][0] + dp[v][1]});
            }
        }
        int m = SZ(ke[u]);
        sort (ALL(vec));
        dp[u][0] = Tot;
        dp[u][1] = Tot + Wp[u];
        rep (i, 0, SZ(vec) - 1) {
            if (m - i - 1 >= X) dp[u][0] += vec[i];
            else dp[u][0] += min(0LL, vec[i]);
            if (m - i - 2 >= X) dp[u][1] += vec[i];
            else dp[u][1] += min(0LL, vec[i]);
        }
        if (m - SZ(vec) > X) dp[u][0] = INF;
    }

    vector<ll> solution() {
        vector<ll> Ans;
        rep (X, 0, n - 1) {
            dfs(1, 0, X);
            Ans.pb(dp[1][0]);
//            cout << dp[1][0] <<" ";
        }
        return Ans;
    }
}

vector<ll> minimum_closure_costs (int _n, vector<int> _u, vector<int> _v, vector<int> _w) {
    n = _n;
    rep (i, 1, n - 1) {
        int u = _u[i - 1], v = _v[i - 1], w = _w[i - 1];
        ++u;
        ++v;
        ke[u].pb({v, w});
        ke[v].pb({u, w});
    }
    if (n <= 2000) return sub4 :: solution();
}

void solution() {
    cin >> n;
    rep (i, 1, n - 1) {
        int u, v, w;
        cin >> u >> v >> w;
        ++u;
        ++v;
        ke[u].pb({v, w});
        ke[v].pb({u, w});
    }
    if (n <= 2000) sub4 :: solution();
}

//#define file(name) freopen(name".inp","r",stdin); \
//freopen(name".out","w",stdout);
//int main () {
////    file("noel");
//    ios_base :: sync_with_stdio(false); cin.tie(0); cout.tie(0);
//    int num_Test = 1;
////    cin >> num_Test;
//    while (num_Test--)
//        solution();
//}
/*
no bug +8
chu y break hay return se lam hong logic
xet transition cua i va i + 1
construct ket qua
5
0 1 1
0 2 4
0 3 3
2 4 2


*/

Compilation message (stderr)

roads.cpp:101:1: warning: multi-line comment [-Wcomment]
  101 | //#define file(name) freopen(name".inp","r",stdin); \
      | ^
roads.cpp: In function 'std::vector<long long int> minimum_closure_costs(int, std::vector<int>, std::vector<int>, std::vector<int>)':
roads.cpp:86:1: warning: control reaches end of non-void function [-Wreturn-type]
   86 | }
      | ^
#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...