码迷,mamicode.com
首页 > 其他好文 > 详细

SGU195 New Year Bonus Grant

时间:2016-09-10 20:40:11      阅读:140      评论:0      收藏:0      [点我收藏+]

标签:

题解:

树的独立集问题。。。

树的独立集问题在dp上有两种写法,一种是搜索,一种是刷表。

这题是刷表

题意:给出一棵树,根节点是1,要求根据以下要求选择最多的节点:

          ①不能选择1

          ②若选择当前节点,那么该节点的父节点和儿子都不能选择。

          ③若某节点的某一个儿子节点被选择,那么该节点的其他儿子不能被选择。(中文题意链接:http://blog.csdn.net/qian99/article/details/16970341)

关键是对于第3种情况。

dp[u][0]表示当前点不选,dp[u][1]表示当前点选

分为2种第一种是所有儿子都不选择,第二种选择一个最优儿子,那么等价于先全部不选择儿子。然后一定要选儿子的话,选dp[v][1]-dp[v][0]最大的就行了。具体看代码

代码:

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<cstdio>
using namespace std;
#define pb push_back
#define mp make_pair
#define se second
#define fs first
#define LL long long
#define CLR(x) memset(x,0,sizeof x)
#define MC(x,y) memcpy(x,y,sizeof(x))  
#define SZ(x) ((int)(x).size())
#define FOR(it,c) for(__typeof((c).begin()) it=(c).begin();it!=(c).end();it++)
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1 
typedef pair<int,int> P;
const double eps=1e-9;
const int N=5e5+10;
const int M=1e3+10;
const LL mod=1000000007;
const int INF=1e9+10;

int n,cnt,num;
int head[N],dp[N][2],ans[N],to[N];

struct Edge{
    int v,nxt;
}edge[N*2];

void Init(){
    cnt=0;
    num=0;
    memset(head,-1,sizeof(head));
}

void addEdge(int u,int v){
    edge[cnt].v=v;
    edge[cnt].nxt=head[u];
    head[u]=cnt++;
}

void dfs(int u){
    dp[u][0]=0,dp[u][1]=1000;
    int tmp=-INF;
    for(int i=head[u];i!=-1;i=edge[i].nxt){
        int v=edge[i].v;
        dfs(v);
        dp[u][1]+=dp[v][0]; 
        dp[u][0]+=dp[v][0];
        if(dp[v][1]-dp[v][0]>tmp){
            tmp=dp[v][1]-dp[v][0];
            to[u]=v;
        }
    }
    if(tmp>0) dp[u][0]+=tmp;
    else      to[u]=0;
}

void Find_path(int u,int k){
    int Best=-1;
    if(k==0) Best=to[u];
    if(k==1) ans[num++]=u;
    for(int i=head[u];i!=-1;i=edge[i].nxt){
        int v=edge[i].v;
        if(k==1) Find_path(v,0);
        if(k==0){
            if(v==Best) Find_path(v,1);
            else        Find_path(v,0);
        }
    }
}


int main(){
    Init();
    scanf("%d",&n);
    for(int i=2;i<=n;i++){
        int x;
        scanf("%d",&x);
        addEdge(x,i);
    }
    dfs(1);
    printf("%d\n",dp[1][0]);
    Find_path(1,0);
    sort(ans,ans+num);
    for(int i=0;i<num;i++) printf("%d%c",ans[i],(i==num-1?\n: ));
    return 0;
}

 

SGU195 New Year Bonus Grant

标签:

原文地址:http://www.cnblogs.com/byene/p/5860069.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!