Advertisement
Guest User

Untitled

a guest
Feb 9th, 2016
50
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 2.42 KB | None | 0 0
  1. import java.io.BufferedOutputStream;
  2. import java.io.BufferedReader;
  3. import java.io.IOException;
  4. import java.io.InputStreamReader;
  5. import java.io.PrintWriter;
  6. import java.util.HashMap;
  7. import java.util.HashSet;
  8. import java.util.Random;
  9. import java.util.StringTokenizer;
  10.  
  11.  
  12. public class CutTreeFull {
  13.  
  14. static HashMap<Integer, HashSet<Integer>> G = new HashMap<>();
  15. static HashMap<Integer, Long> vertexCost = new HashMap<>();
  16. static HashMap<Integer, Long> vertexSum = new HashMap<>();
  17. static HashMap<Integer, Integer> visitedOrder = new HashMap<>();
  18. static HashSet<Integer> visited = new HashSet<>();
  19. static int n;
  20. static int[] fr;
  21. static int[] to;
  22. static long totalSum = 0;
  23. static int orderIndex = 0;
  24.  
  25. static void addDirectedEdge(int u, int v) {
  26. HashSet<Integer> ady = G.getOrDefault(u, null);
  27. if (ady == null) {
  28. ady = new HashSet<>();
  29. }
  30. ady.add(v);
  31. G.put(u, ady);
  32. }
  33.  
  34. static void addUndirectedEdge(int u, int v) {
  35. addDirectedEdge(u, v);
  36. addDirectedEdge(v, u);
  37. }
  38.  
  39. static long DFS(int root) {
  40. long sum = vertexCost.get(root);
  41. visitedOrder.put(root, orderIndex ++);
  42. visited.add(root);
  43. for (int v : G.get(root)) {
  44. if (!visited.contains(v)) {
  45. sum += DFS(v);
  46. }
  47. }
  48. vertexSum.put(root, sum);
  49. return sum;
  50. }
  51.  
  52. static void solve(PrintWriter out) {
  53. long sol = Long.MAX_VALUE;
  54. Random r = new Random();
  55. int root = r.nextInt(n) + 1;
  56. DFS(root);
  57. for (int i = 0; i < n-1; ++ i) {
  58. int u = fr[i];
  59. int v = to[i];
  60. if (visitedOrder.get(v) < visitedOrder.get(u)) {
  61. int temp = u;
  62. u = v;
  63. v = temp;
  64. }
  65. long vSum = vertexSum.get(v);
  66. long rest = totalSum - vSum;
  67. sol = Math.min(sol, Math.abs(rest - vSum));
  68. }
  69. out.println(sol);
  70. out.close();
  71. }
  72.  
  73. public static void main(String[] args) throws IOException {
  74.  
  75. BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
  76. PrintWriter out = new PrintWriter(new BufferedOutputStream(System.out));
  77.  
  78. n = Integer.parseInt(in.readLine());
  79.  
  80. fr = new int[n-1];
  81. to = new int[n-1];
  82.  
  83. StringTokenizer st = new StringTokenizer(in.readLine());
  84. for (int i = 0; i < n; ++ i) {
  85. long c = Long.parseLong(st.nextToken());
  86. vertexCost.put(i+1, c);
  87. totalSum += c;
  88. }
  89.  
  90. for (int i = 0; i < n-1; ++ i) {
  91. st = new StringTokenizer(in.readLine());
  92. int u = Integer.parseInt(st.nextToken());
  93. int v = Integer.parseInt(st.nextToken());
  94. fr[i] = u;
  95. to[i] = v;
  96. addUndirectedEdge(u, v);
  97. }
  98.  
  99. solve(out);
  100.  
  101. }
  102.  
  103. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement