fork download
copy 
  1.  
  2. import java.util.*;
  3. import java.lang.*;
  4. import java.io.*;
  5.  
  6. /* Name of the class has to be "Main" only if the class is public. */
  7. class Ideone
  8. {
  9. 		public static void main (String[] args) throws java.lang.Exception
  10. 	{
  11. 		Scanner sc = new Scanner(System.in);
  12. 		int n = sc.nextInt();
  13. 		int m = sc.nextInt();
  14. 		List<List<Integer>> adj = new ArrayList<>();
  15. 		for(int i=0; i<=n;i++){
  16. 			adj.add(new ArrayList<>());
  17. 		}
  18.  
  19. 		int value[] = new int[n+1];
  20.  
  21. 		for(int i=1;i<=n;i++){
  22. 			value[i] = sc.nextInt();
  23. 		}
  24.  
  25. 		for(int i=0; i<m; i++){
  26. 			int u = sc.nextInt();
  27. 			int v = sc.nextInt();
  28. 			adj.get(u).add(v);
  29. 			adj.get(v).add(u);
  30. 		}
  31. 		int parent[] = new int[n+1];
  32. 		int visited[] = new int[n+1];
  33. 		int sum[] = new int[n+1];
  34. 		maxSumOfSubtree(1,parent,visited,value,sum,adj);
  35. 		int max =0;
  36. 		for(int x: sum){
  37. 			max = Math.max(max,x);
  38. 		}
  39. 		System.out.println(Arrays.toString(sum));
  40. 		System.out.println(max);
  41. 	}
  42.  
  43. 	public static void maxSumOfSubtree(int node,int []par,int []vis,int[] value,int []sum,List<List<Integer>> adj){
  44.  
  45. 		vis[node] = 1;
  46. 		for(int x : adj.get(node)){
  47. 			if(vis[x] == 0){
  48. 				par[x] = node;
  49. 				maxSumOfSubtree(x,par,vis,value,sum,adj);
  50. 			}
  51. 		}
  52.  
  53. 		int s=0;
  54. 		for(int child : adj.get(node)){
  55. 			if(child != par[node]){
  56. 				// this is a child node;
  57. 				s+=sum[child];
  58. 			}
  59. 		}
  60.  
  61. 		sum[node] = value[node] + s;
  62.  
  63. 	}
  64. }
Success #stdin #stdout 0.14s 56568KB
comments (?)
stdin
copy 
7
6
-5
4
6
3
-3
2
-9
1 2
1 3
2 4
2 5
3 6
3 7
stdout
copy 
[0, -2, 4, -1, 3, -3, 2, -9]
4
https://ideone.com/YSMEDW
language:
Java (HotSpot 12)
created:
2 days ago
visibility:
public

Share or Embed source code

Discover > Sphere Engine API

The brand new service which powers Ideone!

Discover > IDE Widget

Widget for compiling and running the source code in a web browser!