当前位置:编程学习 > C/C++ >>

zoj 3659 Conquer a New Region

题目大意:
有一棵树,每条边有权值,从树上一点到另一点的路径权值为该条路径上的最小权边,选一个点使得该点到其余所有点的路径权值和最大.

题目思路:
看一下规模,可能是贪心,dp,线段树之类的,线段树明显没想法,dp嘛,状态太多了,那就想想贪心,不过不管是什么方法,最终的目的就是要得到一个有根树,这里我们可以用并查集+路径压缩来做.
贪心我们只要考虑构树是按边的权值降序还是升序,如果选用升序的话,当前边要到前面加入的边的限制,所以就麻烦了,而降序呢,前面加入的边是受当前边的限制,所以相对容易多了.
对与每个父亲还要保存到其子孙的路径权值和以及子孙数量,这在合并两棵树是要用到.
1)两个点均无父亲,任选一个作为父亲.
2)其中一个无父亲,把有父亲的父亲作为无父亲的父亲(很绕口= =...将就一下).
3)均有父亲,那么就要判断一下到底谁做父亲可以使构成的树路径权值更大,在这边就要用到路径权值和以及子孙数量了(我是用到了,大牛可能不需要...).

代码:
[cpp]
#include <stdlib.h> 
#include <string.h> 
#include <stdio.h> 
#include <ctype.h> 
#include <math.h> 
#include <stack> 
#include <queue> 
#include <map> 
#include <set> 
#include <vector> 
#include <string> 
#include <iostream> 
#include <algorithm> 
using namespace std; 
 
#define ll long long 
#define ls rt<<1 
#define rs ls|1 
#define lson l,mid,ls 
#define rson mid+1,r,rs 
#define middle (l+r)>>1 
#define eps (1e-9) 
#define type int 
#define clr_all(x,c) memset(x,c,sizeof(x)) 
#define clr(x,c,n) memset(x,c,sizeof(x[0])*(n+1)) 
#define MOD 1000000007 
#define inf 0x3f3f3f3f 
#define pi acos(-1.0) 
#define M 200000 +5 
 
template <class T> void _swap(T &x,T &y){T t=x;x=y;y=t;} 
template <class T> T _max(T x,T y){return x>y? x:y;} 
template <class T> T _min(T x,T y){return x<y? x:y;} 
int test,cas; 
 
int n,m; 
struct node{ 
    int u,v; 
    ll c; 
    void read(){ 
        scanf("%d%d%lld",&u,&v,&c); 
    } 
    bool operator < (const node &t) const{ 
        return c > t.c; 
    } 
}p[M]; 
ll sum[M],cnt[M]; 
int fa[M]; 
 
int Find(int x){ 
    for(;x!=-1 && x!=fa[x];x=fa[x]); 
    return x; 

 
void Union(int u,int v,ll val){ 
    int uf=Find(u),vf=Find(v); 
    if(uf==-1 && vf==-1) 
        fa[u]=v,sum[v]+=val,cnt[v]++,fa[v]=v; 
    else if(uf==-1) 
        fa[u]=vf,sum[vf]+=val,cnt[vf]++; 
    else if(vf==-1) 
        fa[v]=uf,sum[uf]+=val,cnt[uf]++; 
    else{ 
        ll tu=sum[uf]+(cnt[vf]+1)*val; 
        ll tv=sum[vf]+(cnt[uf]+1)*val; 
        if(tu > tv) fa[vf]=uf,sum[uf]=tu,cnt[uf]+=cnt[vf]+1; 
        else fa[uf]=vf,sum[vf]=tv,cnt[vf]+=cnt[uf]+1; 
    } 

 
void run(){ 
    int i,j; 
    clr(sum,0,n),clr(cnt,0,n); 
    clr(fa,-1,n); 
    m=n-1; 
    for(i=0;i<m;i++) 
        p[i].read(); 
    sort(p,p+m); 
    for(i=0;i<m;i++) 
        Union(p[i].u,p[i].v,p[i].c); 
    for(i=1;i<=n;i++) if(cnt[i]+1==n){ 
        printf("%lld\n",sum[i]); 
        return; 
    } 

 
void preSof(){ 

 
int main(){ 
    //freopen("1.in","r",stdin); 
    //freopen("1.out","w",stdout); 
    preSof(); 
    //run(); 
    while(~scanf("%d",&n)) run(); 
    //for(scanf("%d",&test),cas=1;cas<=test;cas++) run(); 
    return 0; 

补充:软件开发 , C++ ,
CopyRight © 2022 站长资源库 编程知识问答 zzzyk.com All Rights Reserved
部分文章来自网络,