poj2378_树型DP

题目大意：给定一棵树的n个节点和n-1条边，对于每个节点，要求去掉它之后所有子树的最大节点数，并输出最大节点数小于n/2的节点编号

算法：用树型dp实现，对于每个节点，保留包含该点的子树的节点数son，并且保留去掉改点后的最大子树的节点数val

结构定义：

struct Node { int val; //存放去掉该点剩余的树中最大节点数 int son; //存放包含该点的子树的节点数 vector<int> child; //存放子节点id };

dp过程：

void dps(int p) { visit[p]=true; tree[p].val=0; tree[p].son=1; int i; for (i=0;i<tree[p].child.size();i++) { if(!visit[tree[p].child[i]]) { dps(tree[p].child[i]); tree[p].son+=tree[tree[p].child[i]].son; if (tree[tree[p].child[i]].son>tree[p].val) { tree[p].val=tree[tree[p].child[i]].son; } } } if (tree[p].val<n-tree[p].son) { tree[p].val=n-tree[p].son; } }

源代码：

#include <iostream> #include <vector> #include <string.h> using namespace std; struct Node { int val; //存放去掉该点剩余的树中最大节点数 int son; //存放包含该点的子树的节点数 vector<int> child;//存放子节点id }; Node tree[10005]; bool visit[10005]; int n; void dps(int p) { visit[p]=true; tree[p].val=0; tree[p].son=1; int i; for (i=0;i<tree[p].child.size();i++) { if(!visit[tree[p].child[i]]) { dps(tree[p].child[i]); tree[p].son+=tree[tree[p].child[i]].son; if (tree[tree[p].child[i]].son>tree[p].val) { tree[p].val=tree[tree[p].child[i]].son; } } } if (tree[p].val<n-tree[p].son) { tree[p].val=n-tree[p].son; } } int main() { int i; int a,b; scanf("%d",&n); memset(visit,false,sizeof(visit)); for (i=1;i<=n-1;i++) { scanf("%d%d",&a,&b); tree[a].child.push_back(b); tree[b].child.push_back(a); } dps(1); for (i=1;i<=n;i++) { if (2*tree[i].val<=n) { printf("%d/n",i); } } return 0; }