【POJ】1655 balance act

有人说是树形dp，单我感觉不是dp，就只是单纯的树的遍历而已。因为每个节点都访问了一次而已。

用图的邻接矩阵表示树，题目给出根节点为1号节点，从根节点开始遍历树即可。

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;

#define max_edge 21000

struct Node{
  int vertex;
  int next;
} ad_edge[max_edge * 2];//邻接矩阵表

int vertex_fe[max_edge];//每个点的第一条边，可能是连接父节点的边。
bool vertex_flag[max_edge];//记录点是否被访问过
const int balance_max = 9999999;//结果初始化用到的饿值
const int balance_ver_max = 9999999;//结果初始化用到的值，记录的是banlance相同时最小的点的序号
int balance, balance_ver;//结果值
int vertex_num, edge_num;//点总数，边总数

int balance_act(int idx){
  vertex_flag[idx] = true;//已访问，即防止子节点递归访问父节点
  int max_branch_node_num = 0;//记录去掉该节点后，最大的分支的节点数量
  int p_edge = vertex_fe[idx];//点的邻接表中的第一条边
  int node_num = 0;
  while (p_edge != -1){//遍历访问每一条边对应的节点
    if (!vertex_flag[ad_edge[p_edge].vertex]){//该点没有被访问过，即不是父节点
      int t = balance_act(ad_edge[p_edge].vertex);//递归求解子树内的节点数量
      max_branch_node_num = max(max_branch_node_num, t);
      node_num += t;//记录子树的节点总数
    }
    p_edge = ad_edge[p_edge].next;
  }
  max_branch_node_num = max(max_branch_node_num, vertex_num - 1 - node_num);
  if (balance > max_branch_node_num){//结果的更新
    balance = max_branch_node_num;
    balance_ver = idx;
  }
  if (balance == max_branch_node_num && balance_ver < idx){//banlance相同时结果的更新
    balance_ver = idx;
  }
  return node_num + 1;//返回子树的节点总数+自己本身，即以idx为根的子树的节点数
}

int main()
{
  int n;
  cin >> n;
  while (n --){
    //初始化点的第一条边和点访问记录
    memset(vertex_fe, -1, sizeof(vertex_fe));
    memset(vertex_flag, 0, sizeof(vertex_flag));
    cin >> vertex_num;
    edge_num = vertex_num - 1;
    //读入并构建邻接矩阵表
    for (int i = 0; i < edge_num; i ++){
      int v1, v2;
      scanf("%d%d", &v1, &v2);
      ad_edge[2*i].vertex = v2;
      ad_edge[2*i].next = vertex_fe[v1];
      vertex_fe[v1] = 2*i;
      ad_edge[2*i + 1].vertex = v1;
      ad_edge[2*i + 1].next = vertex_fe[v2];
      vertex_fe[v2] = 2*i + 1;
    }
    //初始化结果值
    balance = balance_max;
    balance_ver = balance_ver_max;
    balance_act(1);
    cout << balance_ver << ' ' << balance << endl;
  }
  return 0;
}