poj 1849 Two 树形dp好题

题意：

可以转换为于在一棵树中，找m个叶子节点，使他们连线（重复部分只计一次）的距离和最大。

分析：

m==2时求树的直径倒是简单，m>2时就要用树形dp了，有n*m^2的树形dp.对于树形dp,不仅仅每一棵子树可以看做一个子问题，每一个作为子树的子问题也可以看成是由一些更小的子问题构成，代码中的这种思想用滚动数组实现了。

代码：

//poj 1849
//sep9
#include <iostream>
#include <vector>
using namespace std;
const int MAXN=100024;
vector<int> g[MAXN],val[MAXN];
int dp[MAXN][3];

void dfs(int u,int p)
{
for(int i=g[u].size()-1;i>=0;--i){
int v=g[u][i];
if(p==v) continue;
dfs(v,u);
for(int j=2;j>=0;--j){
dp[u][j]=dp[u][j]+dp[v][0]+2*val[u][i];
for(int s=1;s<=j;++s)
dp[u][j]=min(dp[u][j],dp[u][j-s]+dp[v][s]+s*val[u][i]);
}
}
}

int main()
{
int n,s;
scanf("%d%d",&n,&s);
for(int i=1;i<=n;++i)
g[i].clear();
for(int i=1;i<n;++i){
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
g[u].push_back(v),g[v].push_back(u);
val[u].push_back(w),val[v].push_back(w);
}
memset(dp,0,sizeof(dp));
dfs(s,-1);
printf("%d",dp[s][2]);
return 0;
}