Description
Input
Output
输出文件仅包含一行,一个整数,表示最少的被重建(即修改储存器容量)的米特储存器的数目。
Sample Input
5
4
3
2
1
1 2
1 3
2 4
2 5
Sample Output
HINT
【样例解释】
一个最优解是将A[1]改成8,A[3]改成4,A[5]改成2。这样,2和3运给1的量相等,4和5运给2的量相等,且每天晚上六点的时候,1,2满,3,4,5空,满足所有限制条件。
题目大意
给一棵树,每个点有一个权值,要求修改一些点的权值,使得:
①同一个父亲的儿子权值必须相同
②父亲的取值必须是所有儿子权值之和
题解
有这样一个结论,当这棵树的任何一个节点的权值确定之后,其余所有节点的权值便都可算出来。
例如下图:
若我们确定了 $5$ 号节点的权值为 $x$ ,那么 $7$ 号节点的权值 $y$ 可以算出 $y = \frac{3}{2} \cdot x$ 。
现在我们将每一条边定向,方向为从父亲指向儿子,对于每个节点,统计每个节点的出度,做一遍树上前缀积 $prod_u$ 。例如上图中 $prod_5 = 6$ , $prod_7 = 4$ ,特别地 $prod_1 = 1$ 。
我们假设 $u$ 节点的权值是不变的,那么必然有修改后的根节点的权为 $w = a_u \cdot prod_u$。
记 $f_u = prod_u*a_u$ ,我们将树上 $f_u$ 相同的点放在一组,现在问题就变成了求点数最多的一组的点的个数。
由于乘积过大,直接 $hash$ 。
1 //It is made by Awson on 2018.1.3 2 #include <set> 3 #include <map> 4 #include <cmath> 5 #include <ctime> 6 #include <queue> 7 #include <stack> 8 #include <cstdio> 9 #include <string> 10 #include <vector> 11 #include <cstdlib> 12 #include <cstring> 13 #include <iostream> 14 #include <algorithm> 15 #define LL long long 16 #define LD long double 17 #define Max(a, b) ((a) > (b) ? (a) : (b)) 18 #define Min(a, b) ((a) < (b) ? (a) : (b)) 19 using namespace std; 20 const int N = 500000; 21 const int MOD1 = 1e6+9; 22 const int MOD2 = 1e6+7; 23 const int MOD3 = 1e6-3; 24 25 int a[N+5], u, v, n; 26 struct tt { 27 int to, next; 28 }edge[(N<<1)+5]; 29 int path[N+5], top, degree[N+5]; 30 int hash1[MOD1+5], hash2[MOD2+5], hash3[MOD3+5]; 31 int ans1, ans2, ans3; 32 33 void add(int u, int v) { 34 edge[++top].to = v; 35 edge[top].next = path[u]; 36 path[u] = top; 37 } 38 void dfs(int u, int fa, int num1, int num2, int num3) { 39 int tmp, d = --degree[u]; 40 tmp = ++hash1[(LL)num1*a[u]%MOD1], ans1 = Max(ans1, tmp); 41 tmp = ++hash2[(LL)num2*a[u]%MOD2], ans2 = Max(ans2, tmp); 42 tmp = ++hash3[(LL)num3*a[u]%MOD3], ans3 = Max(ans3, tmp); 43 for (int i = path[u]; i; i = edge[i].next) 44 if (edge[i].to != fa) dfs(edge[i].to, u, (LL)num1*d%MOD1, (LL)num2*d%MOD2, (LL)num3*d%MOD3); 45 } 46 void work() { 47 scanf("%d", &n); 48 for (int i = 1; i <= n; i++) scanf("%d", &a[i]); 49 for (int i = 1; i < n; i++) { 50 scanf("%d%d", &u, &v); 51 add(u, v), add(v, u); ++degree[u], ++degree[v]; 52 } 53 ++degree[1]; 54 dfs(1, 0, 1, 1, 1); 55 printf("%d\n", n-Min(Min(ans1, ans2), ans3)); 56 } 57 int main() { 58 work(); 59 return 0; 60 }