Submission #52116664

---

Source Code Expand

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;

int n;
int arr[100010];
vector<int> v[100010];
long long sum[100010];
int depth[100010];
long long total;
long long ans;

void dfs(int cur, int prv)
{
	int i;

	sum[cur] = arr[cur];
	for (i = 0; i < v[cur].size(); i++)
	{
		if (v[cur][i] == prv) continue;

		depth[v[cur][i]] = depth[cur] + 1;

		dfs(v[cur][i], cur);

		sum[cur] += sum[v[cur][i]];
	}
}

void dfs2(int cur, int prv, long long val)
{
	int i;

	ans = min(ans, val);

	for (i = 0; i < v[cur].size(); i++)
	{
		if (v[cur][i] == prv) continue;

		dfs2(v[cur][i], cur, val - sum[v[cur][i]] + total - sum[v[cur][i]]);
	}
}

int main()
{
	int i;
	int a, b;
	long long x;

	cin >> n;
	for (i = 0; i < n - 1; i++)
	{
		cin >> a >> b;

		v[a].push_back(b);
		v[b].push_back(a);
	}
	for (i = 1; i <= n; i++)
	{
		cin >> arr[i];

		total += arr[i];
	}

	dfs(1, 0);

	x = 0;
	for (i = 1; i <= n; i++)
	{
		x += 1LL * depth[i] * arr[i];
	}

	ans = x;
	dfs2(1, 0, x);

	cout << ans;
}

Submission Info

Compile Error

Main.cpp: In function ‘void dfs(int, int)’:
Main.cpp:19:23: warning: comparison of integer expressions of different signedness: ‘int’ and ‘std::vector<int>::size_type’ {aka ‘long unsigned int’} [-Wsign-compare]
   19 |         for (i = 0; i < v[cur].size(); i++)
      |                     ~~^~~~~~~~~~~~~~~
Main.cpp: In function ‘void dfs2(int, int, long long int)’:
Main.cpp:37:23: warning: comparison of integer expressions of different signedness: ‘int’ and ‘std::vector<int>::size_type’ {aka ‘long unsigned int’} [-Wsign-compare]
   37 |         for (i = 0; i < v[cur].size(); i++)
      |                     ~~^~~~~~~~~~~~~~~

Judge Result