树的最大独立集 (算法经典入门）

思考：看过算法经典入门再看看网上的这个程序就是一种享受啊！禁不住转载，我还在树上的动态规划入门阶段。。。加油！

#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