#include <bits/stdc++.h>
#define ve vector
#define vi vector<int>
#define vii vector<ii>
#define ii pair<int,int>
#define fi first
#define se second
#define ll long long
#define INF 1e9+7
#define pb push_back
#define optimise ios_base::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL)
using namespace std;
#include <ext/pb_ds/assoc_container.hpp>
using namespace __gnu_pbds;
template<class T>
using Tree = tree<T, null_type, less<T>, rb_tree_tag,
tree_order_statistics_node_update>;
const int MOD = 1e9+7;
const int nax = 2e5+5;
void readio(){
#ifndef ONLINE_JUDGE
freopen("input.txt", "r", stdin);
freopen("output.txt", "w", stdout);
#endif
}
vii adj[nax];
bool vis[nax];
int dfs(int u, int p){
int mx = 0;
for(auto x : adj[u]){
if(x.fi == p) continue;
if(!vis[x.fi]) mx = max(mx, dfs(x.fi,u)+x.se);
else mx = max(mx, dfs(x.fi, u));
}
//cout << u << " " << mx << endl;
return mx;
}
bool ass(int u, int p, int path, int mx){
if(path == mx) return 1;
for(auto x : adj[u]){
if(x.fi == p) continue;
int cur = path;
if(!vis[x.fi]) cur += x.se;
if(ass(x.fi,u,cur,mx)){
vis[x.fi] = 1;
return 1;
}
}
return 0;
}
int main()
{
optimise;
int n, k;
cin >> n >> k;
for(int i = 0; i < n-1; i++){
int a,b,c;
cin >> a >> b >> c;
a--,b--;
adj[a].pb({b,c});
adj[b].pb({a,c});
}
for (int i = 0; i < n; ++i)
{
int mx = 0;
memset(vis,0, sizeof vis);
for (int j = 0; j < k; ++j)
{
int cur = dfs(i,i);
mx += cur;
ass(i,i, 0, cur);
}
cout << mx << endl;
}
}
Compilation message
Main.cpp: In function 'void readio()':
Main.cpp:22:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
22 | freopen("input.txt", "r", stdin);
| ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~
Main.cpp:23:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
23 | freopen("output.txt", "w", stdout);
| ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
3 ms |
5112 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
3 ms |
5112 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
3 ms |
5112 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
3 ms |
5112 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
1064 ms |
10956 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
3 ms |
5112 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |