生命之树

Description
在X森林里，上帝创建了生命之树。

他给每棵树的每个节点（叶子也称为一个节点）上，都标了一个整数，代表这个点的和谐值。
上帝要在这棵树内选出一个非空节点集S，使得对于S中的任意两个点a,b，都存在一个点列 {a, v1, v2, ..., vk, b} 使得这个点列中的每个点都是S里面的元素，且序列中相邻两个点间有一条边相连。

在这个前提下，上帝要使得S中的点所对应的整数的和尽量大。
这个最大的和就是上帝给生命之树的评分。

经过atm的努力，他已经知道了上帝给每棵树上每个节点上的整数。但是由于 atm 不擅长计算，他不知道怎样有效的求评分。他需要你为他写一个程序来计算一棵树的分数。

「输入格式」
第一行一个整数 n 表示这棵树有 n 个节点。
第二行 n 个整数，依次表示每个节点的评分。
接下来 n-1 行，每行 2 个整数 u, v，表示存在一条 u 到 v 的边。由于这是一棵树，所以是不存在环的。

「输出格式」
输出一行一个数，表示上帝给这棵树的分数。

「样例输入」
5
1 -2 -3 4 5
4 2
3 1
1 2
2 5

「样例输出」
8

Input
第一行一个整数 n 表示这棵树有 n 个节点。
第二行 n 个整数，依次表示每个节点的评分。
接下来 n-1 行，每行 2 个整数 u, v，表示存在一条 u 到 v 的边。由于这是一棵树，所以是不存在环的。
Output
输出一行一个数，表示上帝给这棵树的分数。
Sample Input
5
1 -2 -3 4 5
4 2
3 1
1 2
2 5
Sample Output

8

 

这题瞎搜，居然被我水过去了，严重怀疑学校的测试数据：

package 省赛2015;

import java.util.Scanner;

public class T9 {

  static boolean[][] data;
  static boolean[][] data1;
  static int[] fenzhi;
  static int max;
  static int n;
  static boolean[] vis; 
  
  public static void main(String[] args) {
    
    Scanner sc=new Scanner(System.in);
    n=sc.nextInt();
    
    fenzhi=new int[n+1];
    for (int i = 1; i <= n; i++) {
      fenzhi[i]=sc.nextInt();
    }
    
    data=new boolean[n+1][n+1];
    data1=new boolean[n+1][n+1];
    vis=new boolean[n+1];
    
    for (int i = 1; i < n; i++) {
      int a=sc.nextInt();
      int b=sc.nextInt();
      
      data1[a][b]=data1[b][a]=data[a][b]=data[b][a]=true;
      
    }
    
    //
    for (int i = 1; i < n; i++) {
      for (int j = i+1; j <= n; j++) {
        vis=new boolean[n+1];
        vis[i]=true;      
        dfs(i,j,fenzhi[i]);
        
        for (int a = 0; a <= n; a++) {
          for (int b = 0; b <= n; b++) {
            data[a][b]=data1[a][b];
          }
        }
      }
    }
    System.out.println(max);
  }
  private static void dfs(int begin,int end,int fen) {
    
    if(data[begin][end]){
      
      for (int i = 1; i <= n; i++) {
        if(vis[i]||i==end)continue;
        if(data[begin][i]){
          if(fenzhi[i]>0)
            fen+=fenzhi[i];
        }
      }
      
      for (int i = 1; i <= n; i++) {
        if(vis[i]||i==begin)continue;
        if(data[end][i]){
          if(fenzhi[i]>0)
            fen+=fenzhi[i];
        }
      }
      fen+=fenzhi[end];
      if(fen>max){
        max=fen;
      }
      return;
    }
    
    for (int i = 1; i <= n; i++) {
      if(vis[i])continue;
      
      if(data[end][i]){
        data[end][i]=false;
        vis[i]=true;
        fen+=fenzhi[i];
        
        dfs(end,i,fen);
        
        data[end][i]=true;
        vis[i]=false;
        fen-=fenzhi[i];
      }
    }
  }
}