POJ 2342 树的最大独立集

题意：在树的最大独立集的基础上，加上权值。求最大。

分析：

采用刷表的方式写记忆化，考虑一个点选和不选，返回方式pair 型。

首先，无根树转有根树，dp(root)。

注意的是：u不选，那么他的子节点，可以选，或者不选。WA了无数次。

#include <cstdio>

#include <vector>

#include <cmath>

using namespace std;

const int maxn = +;

vector<int> G[maxn];

int a[maxn];

int p[maxn];

void dfs(int u,int fa)

{

    int d = G[u].size();

    for(int i=; i<d; i++)

    {

        int v = G[u][i];

        if(v!=fa)

            dfs(v,p[v]=u);

    }

}

int d[maxn][];

bool vis[maxn];

void dp(int u)

{

    if(vis[u])

        return;

    vis[u] = ;

    for(int i=; i<G[u].size(); i++)

    {

        int v = G[u][i];

        if(v!=p[u])

        {

            dp(v);

            d[u][] +=d[v][];

            d[u][] +=max(d[v][],d[v][]);

        }

    }

}

int main()

{

    //freopen("in.txt","r",stdin);

    int n;

    scanf("%d",&n);

    for(int i=; i<=n; i++) {

        scanf("%d",&a[i]);

        d[i][] = a[i];

    }

    int u,v;

    while(scanf("%d%d",&u,&v),u+v)

    {

        G[u].push_back(v);

        G[v].push_back(u);

    }

    p[] = -;

    dfs(,-);

    dp();

    printf("%d\n",max(d[][],d[][]));

    return ;

}