Submission #1323932

#TimeUsernameProblemLanguageResultExecution timeMemory
1323932AestivateMuseum (CEOI17_museum)C++20
100 / 100
404 ms745952 KiB
#include <bits/stdc++.h>
using namespace std;

// #define int long long
#define F first 
#define S second 
#define pb push_back

vector<pair<int, int>>node[10001];
int f[10001][10001][2];
int sz[10001];
int n, k, x;

void dfs(int v, int pre){
    sz[v]=1;
    for(auto kk: node[v]){
        int gg=kk.F;int c=kk.S;
        if(gg==pre) continue;
        dfs(gg, v);
        for(int i=sz[v];i>=1;i--){
            for(int j=1;j<=sz[gg];j++){
                if(i+j>k) break;
                f[v][i+j][0]=min(f[v][i+j][0], f[v][i][0]+f[gg][j][0]+c*2);
                f[v][i+j][1]=min({f[v][i+j][1], f[v][i][1]+c*2+f[gg][j][0], f[v][i][0]+c+f[gg][j][1]});
            }
        }
        sz[v]+=sz[gg];
    }
}

void solve(){
    cin>>n>>k>>x;
    
    for(int i=1;i<=n;i++) for(int j=2;j<=k;j++) for(int _=0;_<2;_++) f[i][j][_]=1e9;
    for(int i=1;i<n;i++){
        int g, h, c;
        cin>>g>>h>>c;
        node[g].pb({h, c});
        node[h].pb({g, c});
    }
    dfs(x, x);
    cout<<f[x][k][1]<<"\n";
}

signed main() {
    ios::sync_with_stdio(0);
    cin.tie(0);
    // int t;
    // cin>>t;
    // while(t--)
    solve();
    return 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...