树形dp ( vector dfs 01背包 ) hdu 1011 Starship Troopers

来源:互联网 发布:matlab矩阵添加一行 编辑:程序博客网 时间:2024/05/29 15:28
#include <algorithm>#include <cstring>#include <cstdio>#include <vector>using namespace std;int n, m;int dp[101][101];int used[101];int need[101];int get[101];vector<int>v[101];void dfs( int x ){int i, j, k;    used[x]=1;    int cost=( need[x]+19 )/20;    for( i=cost;i<=m;i++ )        dp[x][i]=get[x];    for( i=0;i<( int )v[x].size();i++ )    {        int num=v[x][i];        if( used[num] )            continue;        dfs( num );        for( j=m;j>=cost;j-- )   //  01背包        for( k=1;j-k>=cost;k++ )            dp[x][j]=max( dp[x][j], dp[x][j-k]+dp[num][k] );//dp[num][k]就是代表 和x连着的下一个房间num房间分配k个人所获得最大量    }}int main(){    int i, j, k;    int x, y;    while( scanf( "%d%d", &n, &m ), n+m>0 )    {        for( i=1;i<=n;i++ )            scanf( "%d%d", &need[i], &get[i] );        for( i=1;i<=n;i++ )            v[i].clear();        for( i=1;i<n;i++ )        {            scanf( "%d%d", &x, &y );            v[x].push_back( y );            v[y].push_back( x );        }        if( m==0 )            printf( "0\n" );        else        {             memset( used, 0, sizeof( used ) );             memset( dp, 0, sizeof( dp ) );             dfs( 1 );             printf( "%d\n", dp[1][m] );        }    }    return 0;}
</pre><pre name="code" class="cpp">树形dp学习   http://www.cnblogs.com/gq-ouyang/archive/2013/02/26/2933431.html


蓝桥杯题 

#include <algorithm>
#include <cstring>
#include <cstdio>
#include <vector>


using namespace std;


int n;
int f[100001];
int g[100001];
int data[100001];
vector<int>v[100001];


void dfs( int root, int father  )// 深艘的停止,叶节点处。
{
    for( int i=0;i<v[root].size();i++ )
    {
        if( v[root][i]!=father )// 深搜的停止,叶节点处。
        {
            dfs( v[root][i], root );
            f[root]+=g[v[root][i]];
            g[root]+=max( f[v[root][i]], g[v[root][i]] );
        }


    }
    f[root]+=data[root];
}
int main()
{
    int i, j;


    while( scanf( "%d", &n )!=EOF )
    {
        for( i=1;i<=n;i++ )
            scanf("%d", &data[i] );


        for( i=1;i<n;i++ )
        {
            int x, y;
            scanf( "%d%d", &x, &y );
            v[x].push_back( y );
            v[y].push_back( x );
        }


        dfs( 1, 1 );


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




    }


    return 0;
}


0 0
原创粉丝点击