POJ 1192最优连通子集

最优连通子集

题目的描述太繁琐了。。。其实意思简单,就是给定若干个点,如果两个点之间的曼哈顿距离小于1,就连边。然后就构成一颗树,然后每个点都

有一个权值,然后选取若干个点,使得权值和最大,并且点与点之前都连通。简单的树形DP,dp[i]表示以i点为根的子树并且选取i点时的最大权值。

方程就很好写,dp[i]+=dp[u] ,其中u为i的儿子节点,并且dp[u]>0直接dfs一次,同时直接记录最大值就可以了。


#include 
#include 
#include 
using std::vector ;

#define maxn 1005
#define INF 0xc3c3c3c3

int n ;
int x[maxn] ;
int y[maxn] ;
int w[maxn] ;
vector G[maxn] ;
int dp[maxn] ;
int ans ;
void read(){
    scanf("%d" , &n) ;
    int p , q , s ;
    for(int i = 0 ; i < n ; i ++){
        scanf("%d%d%d" , &x[i] , &y[i] , &dp[i]) ;
    }
}
int ManHatten(int p , int q){
    int dist = 0 ;
    dist = (x[p] - x[q] > 0) ? (x[p] - x[q]) : (x[q] - x[p]) ;
    dist += (y[p] - y[q] > 0) ? (y[p] - y[q]) : (y[q] - y[p]) ;
    return dist ;
}
void build(){
    for(int i = 0 ; i < n ; i ++){
        for(int j = i ; j < n ; j ++){
            if(ManHatten(i , j) == 1){
                G[i].push_back(j) ;
                G[j].push_back(i) ;
            }
        }
    }
}
void dfs(int v , int f){
    for(int i = 0 ; i != G[v].size() ; i ++){
        int u = G[v][i] ;
        if(u == f)
            continue ;
        dfs(u , v) ;
        if(dp[u] > 0)
            dp[v] += dp[u] ;
    }
    ans = ans > dp[v] ? ans : dp[v] ;
}
void solve(){
    dfs(0 , -1) ;
    printf("%d\n" ,ans) ;
}
int main(){
    ans = INF  ;
    read() ;
    build();
    solve();
    return 0 ;
}

 
  
 
  
 
  
 
 

你可能感兴趣的:(POJ,树形DP)