最大独立子集模板！！！（树形dp）

关键在于建树！！！void dfs(int u,int fa)//建树fa为父亲节点

 

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

/*********************************/
* 可以动态变化的邻接矩阵
* G[i]表示顶点i的邻接点
/*********************************/

const int MAXN=100;
vector<int> G[MAXN]; //无根树
int l[MAXN]; //结点层次
int p[MAXN]; //根树
int dp[MAXN]; //dp数组
int sumS[MAXN]; //孩子DP和
int sumgS[MAXN]; //孙子DP和
int maxL; //最大层次
int n;

/*********************************/
* 读入无根树，n顶点,n-1边
/*********************************/

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);
    }
}

/*********************************/
* 以无根树u顶点为根，构造有根树
* 主函数调用dfs(u,-1);
* 测试数据：
    8
    0 1
    1 4
    0 2
    0 3
    1 5
    5 6
    5 7
/*********************************/

void dfs(int u,int fa)//建树fa为父亲节点
{
    int d=G[u].size();//和点u相连的点的个数
     l[u]= (fa==-1)? 0: (l[fa]+1);//更新u的层次
     if(l[u]>maxL)//l[u]结点层次
     {
         maxL=l[u];//更新maxL最大层次
     }
    for(int i=0;i<d;++i)
    {
        int v=G[u][i];
        if(v!=fa)
        {
            dfs(v,p[v]=u);//把u设为v的根
        }
    }
}

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)//n个顶点
        {
            if(l[j]==i)//判断j的层次是否为i；
            {
                dp[j]=max(sumgS[j]+1,sumS[j]);//状态转移方程；见刘茹佳书p172；
                if(i-1>=0)//判断是否大于第0层
                {
                    sumS[p[j]]+=dp[j];
                }
                if(i-2>=0)
                {
                    sumgS[p[p[j]]]+=dp[j];
                }
            }
        }
    }
    return dp[u];
}

int main()
{
    readTree();
    int best=-1;
    //分别以每个顶点为根
    for(int i=0;i<n;++i)
    {
        memset(sumgS,0,sizeof(sumgS));
        memset(sumS,0,sizeof(sumS));
        int tmp;
        if((tmp=rootDp(i))>best)
        {
            best=tmp;
        }
    }
    //打印结果看看
    cout<<best<<endl;
    return 0;
}