Submission #2668102

---

Source Code Expand

#include <bits/stdc++.h>
using namespace std;

#define fi first
#define se second
long long n, u, v, w, res[100010], par[100010], lazy[100010], ans = 1e18;
vector <pair<int, pair <int, int> > > E;
vector <int> ccnt[100010];

int main() {
	scanf("%lld", &n);
	for(int i = 1; i < n; i++) {
		scanf("%lld%lld%lld", &u, &v, &w);
		E.push_back({w, {u, v}});
	}
	for(int i = 1; i <= n; i++) {
		par[i] = i, lazy[i] = res[i] = 0;
		ccnt[i].push_back(i);
	}
	sort(E.begin(), E.end(), greater<pair<int, pair<int, int> > >());
	for(int i = 0; i < n-1; i++) {
		int u = E[i].se.fi, v = E[i].se.se, w = E[i].fi, p1 = par[u], p2 = par[v];
		if(ccnt[p1] < ccnt[p2]) swap(p1, p2);
		int s1 = ccnt[p1].size(), s2 = ccnt[p2].size(); lazy[p1] += s2 * w;
		for(int j = 0; j < s2; j++) {
			v = ccnt[p2][j], par[v] = p1, res[v] += s1 * w + lazy[p2] - lazy[p1];
			ccnt[p1].push_back(v);
		}
	}
	for(int i = 1; i <= n; i++) printf("%lld\n", res[i] + lazy[par[i]]);
}

Submission Info

Compile Error

./Main.cpp: In function ‘int main()’:
./Main.cpp:11:19: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
  scanf("%lld", &n);
                   ^
./Main.cpp:13:36: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
   scanf("%lld%lld%lld", &u, &v, &w);
                                    ^

Judge Result