CF1098A Sum in the tree 题解树+贪心

最新推荐文章于 2024-05-20 19:43:04 发布

BestMonkey

最新推荐文章于 2024-05-20 19:43:04 发布

阅读量1k

点赞数 28

分类专栏：题解文章标签： c++ c语言算法

本文链接：https://blog.csdn.net/bestmonkey/article/details/136204063

版权

题解专栏收录该内容

63 篇文章 0 订阅

订阅专栏

Sum in the tree

传送门

Mitya has a rooted tree with $n$ vertices indexed from $1$ to $n$ , where the root has index $1$ . Each vertex $v$ initially had an integer number $a_v \ge 0$ written on it. For every vertex $v$ Mitya has computed $s_v$ : the sum of all values written on the vertices on the path from vertex $v$ to the root, as well as $h_v$ — the depth of vertex $v$ , which denotes the number of vertices on the path from vertex $v$ to the root. Clearly, $s_1=a_1$ and $h_1=1$ .

Then Mitya erased all numbers $a_v$ , and by accident he also erased all values $s_v$ for vertices with even depth (vertices with even $h_v$ ). Your task is to restore the values $a_v$ for every vertex, or determine that Mitya made a mistake. In case there are multiple ways to restore the values, you’re required to find one which minimizes the total sum of values $a_v$ for all vertices in the tree.

Input

The first line contains one integer $n$ — the number of vertices in the tree ( $\le n \le 10^5$ ). The following line contains integers $p_2$ , $p_3$ , … $p_n$ , where $p_i$ stands for the parent of vertex with index $i$ in the tree ( $\le p_i < i$ ). The last line contains integer values $s_1$ , $s_2$ , …, $s_n$ ( $\le s_v \le 10^9$ ), where erased values are replaced by $- 1$ .

Output

Output one integer — the minimum total sum of all values $a_v$ in the original tree, or $- 1$ if such tree does not exist.

Examples

input #1

5
1 1 1 1
1 -1 -1 -1 -1

output #1

input #2

5
1 2 3 1
1 -1 2 -1 -1

output #2

input #3

3
1 2
2 -1 1

output #3

-1

题目翻译

米迪亚有一棵有根的树，树上有 $n$ 个顶点，索引从 $1$ 到 $n$ ，其中根的索引为 $1$ 。每个顶点 $v$ 最初都写有一个整数 $a_v \ge 0$ 。对于每个顶点 $v$ 米迪亚计算了 $s_v$ ：从顶点 $v$ 到根的路径上写在顶点上的所有值的总和，以及 $h_v$ - 顶点 $v$ 的深度，即从顶点 $v$ 到根的路径上的顶点数。显然， $s_1=a_1$ 和 $h_1=1$ 。

然后米迪亚删除了所有数字 $a_v$ ，并且意外地删除了偶数深度顶点（偶数顶点 $h_v$ ）的所有数值 $s_v$ 。你的任务是恢复每个顶点的值 $a_v$ ，或者确定 Mitya 犯了一个错误。如果有多种还原数值的方法，你需要找到一种，使树中所有顶点的数值总和 $a_v$ 最小。

输入格式

第一行包含一个整数 $n$ - 树的顶点数（ $\le n \le 10^5$ ）。下面一行包含整数 $p_2$ , $p_3$ , … $p_n$ ，其中 $p_i$ 代表树中索引为 $i$ 的顶点的父顶点（ $\le p_i < i$ ）。最后一行包含整数值 $s_1$ 、 $s_2$ 、…、 $s_n$ （ $\le s_v \le 10^9$ ），其中已删除的值由 $- 1$ 代替。

输出格式

输出一个整数 - 原始树中所有值 $a_v$ 的最小总和，如果不存在这样的树，则输出 $- 1$ 。

注明

$以上来自 C o d e F orces ，翻译： Dee p L$

解题思路

简要解释一下题意：给出一棵以 $1$ 为根节点的树，大小为 $n$ 。记根节点深度为 $1$ 。树上每个节点都有一个非负权值 $a_i$ （不给出），记 $s_i$ 表示从 $i$ 号节点到根节点的路径上所有点的权值和，包括根节点及其本身。
现在给你每个深度为奇数节点的 $s_i$ 。深度为偶数的节点的 $s_i$ 不给出，值为 $- 1$ 。请你还原树的每个节点的权值。使得其满足所有给出的值不为 $- 1$ 的 $s_i$ 。并输出最小的 $\Sigma a_i$ 。若无解，则输出 −1。

想一想就能明白，让一个 $a_i$ 尽可能的去给更多的 $s_i$ 做贡献，即可使 $\Sigma a_i$ 最小。所以对每个 $- 1$ 的 $s_i$ 搜索出其孩子中的最小 $s_i$ （因为要求每个子节点 $s_i$ 总会大于父节点 $s_i$ ）,最后判断如果有子节点 $s_i$ 小于父节点 $s_i$ ，则输出 $- 1$ 。

AC Code

#include<bits/stdc++.h>
using namespace std;
char buf[1048576], *p1, *p2;
template<typename T>inline void Super_Quick_Read(T &x) {
	bool f = 1;
	x = 0;
	char ch = (p1 == p2 && (p2 = (p1 = buf) + fread(buf, 1, 1 << 20, stdin), p1 == p2) ? 0 : *p1++);
	while (ch < '0' || ch > '9') {
		if (ch == '-') f = !f;
		ch = (p1 == p2 && (p2 = (p1 = buf) + fread(buf, 1, 1 << 20, stdin), p1 == p2) ? 0 : *p1++);
	}
	while (ch >= '0' && ch <= '9')x = (x << 1) + (x << 3) + (ch ^ 48), ch = (p1 == p2 && (p2 = (p1 = buf) + fread(buf, 1, 1 << 20, stdin), p1 == p2) ? 0 : *p1++);
	x = (f ? x : -x);
	return;
}
template<typename T>inline void Quick_Write(T x) {
	if (x < 0) putchar('-'), x = -x;
	if (x > 9) Quick_Write(x / 10);
	putchar(x % 10 + '0');
	return;
}
int n;
long long s[200005];
int U[200005], V[200005], W[200005], Fst[200005], Next[200005], Father[200005], Length;
long long Answer = 0;
inline void Add(int a, int b) {
	U[++Length] = a, V[Length] = b, Next[Length] = Fst[a], Fst[a] = Length;
}
inline long long Dfs(int n) {
	if (s[n] != -1)	return s[n];
	else {
		int k = Fst[n];
		long long _min = 1e9 + 5;
		while (k != -1)_min = min(Dfs(V[k]), _min), k = Next[k];
		s[n] = _min;
		return s[n];
	}
}
signed main() {
	Super_Quick_Read(n);
	for (register int i = 0; i <= n + 1; i++)Fst[i] = -1;
	for (register int i = 2, t; i <= n; ++i)Super_Quick_Read(t), Add(t, i), Father[i] = t;
	for (register int i = 1; i <= n; ++i)Super_Quick_Read(s[i]);
	int flag = 0;
	for (register int i = 1, t1, t2; i <= n; ++i) {
		t1 = Dfs(i), t2 = Dfs(Father[i]);
		if (t1 != 1e9 + 5) {
			if (t2 > t1)flag = 1;
			Answer += (t1 - t2);
		}
	}
	if (flag)puts("-1");
	else Quick_Write(Answer);
	return 0;
}