제출 #1213618

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1213618		Sir_Ahmed_Imran	Paths (RMI21_paths)	C++17	19 / 100	890 ms	7752 KiB

Main

            //    01001100 01001111 01010100 01000001    \\
            //                                           \\
            //                ╦  ╔═╗╔╦╗╔═╗               \\
            //                ║  ║ ║ ║ ╠═╣               \\
            //                ╩═╝╚═╝ ╩ ╩ ╩               \\
            //                                           \\
            //    01001100 01001111 01010100 01000001    \\

#include <bits/stdc++.h>
using namespace std;
#define N 2001
#define nl '\n'
#define ff first
#define ss second
#define add insert
#define ll long long
#define ld long double
#define terminator main
#define pll pair<ll,ll>
#define append push_back
#define pii pair<int,int>
#define all(x) (x).begin(),(x).end()
#define L0TA ios_base::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL)

int n, k;
ll dp[N][N];
vector<pll> a[N];

void dfs(int v, int u){
    vector<ll> x; 
    for(auto & [i, j] : a[v]){
        if(i == u) continue;
        dfs(i, v);
        x.append(dp[i][1] + j);
        for(int l = 1; l < k; l++)
            x.append(dp[i][l + 1] - dp[i][l]);
    }
    sort(all(x));
    reverse(all(x));
    for(int i = 0; i < k; i++){
        dp[v][i + 1] = dp[v][i];
        if(!x.empty()) dp[v][i + 1] += x[i];
    }
}

void solve(){
    int v, u, w;
    cin >> n >> k;
    for(int i = 1; i < n; i++){
        cin >> v >> u >> w;
        a[v].append({u, w});
        a[u].append({v, w});
    }
    for(int r = 1; r <= n; r++){
        dfs(r, 0);
        cout << dp[r][k] << nl;
    }
}
int terminator(){
    L0TA;
    solve();
    return 0;
}

• Subtask 1: 8 / 8
• Subtask 2: 11 / 11
• Subtask 3: 0 / 17
• Subtask 4: 0 / 20
• Subtask 5: 0 / 12
• Subtask 6: 0 / 32