思考:看过算法经典入门再看看网上的这个程序就是一种享受啊!禁不住转载,我还在树上的动态规划入门阶段。。。加油!
#include <iostream>
#include <vector>
#include <algorithm>
#include <cstring>
using namespace std;
const int MAXN=100;
vector<int> G[MAXN]; //无根树
int l[MAXN]; //结点层次
int p[MAXN]; //根树
int dp[MAXN]; //dp数组
int sumC[MAXN]; //孩子DP和
int sumS[MAXN]; //孙子DP和
int maxL; //最大层次
int n;
void readTree()
{
int u,v;
cin >> n;
for(int i=0;i<n-1;++i)
{
cin >> u >> v;
G[u].push_back(v);
G[v].push_back(u);
}
}
void dfs(int u,int fa)
{
int d = G[u].size();
l[u] = (fa==-1) ? 0: (l[fa]+1);
if(l[u] > maxL)
{
maxL=l[u];
}
for(int i=0; i<d; ++i)
{
int v = G[u][i];
if(v != fa)
{
dfs(v, p[v] = u);
}
}
}
int rootDp(int u)
{
//构造u根树
p[u] = -1;
maxL = -1;
dfs(u,p[u]);
for(int i=maxL; i>=0; --i)
{
for(int j=0; j<n; ++j)
{
if(l[j] == i)
{
dp[j] = max(sumS[j]+1, sumC[j]);
if(i-1 >= 0)
{
sumC[p[j]] += dp[j];
}
if(i-2 >= 0)
{
sumS[p[p[j]]] += dp[j];
}
}
}
}
return dp[u];
}
int main()
{
readTree();
int best=-1;
//分别以每个顶点为根
for(int i=0;i<n;++i)
{
memset(sumS,0,sizeof(sumS));
memset(sumC,0,sizeof(sumC));
int tmp;
if((tmp=rootDp(i))>best)
{
best=tmp;
}
}
//打印结果看看
cout<<best<<endl;
return 0;
}
/*********************************
* 以无根树u顶点为根,构造有根树
* 主函数调用dfs(u,-1);
* 测试数据:
8
0 1
1 4
0 2
0 3
1 5
5 6
5 7
*********************************/
转载:http://blog.csdn.net/qq120848369/article/details/5673804