HDU 2196 求树上所有点能到达的最远距离

其实我不是想做这道题的...只是今天考试考了一道类似的题...然后我挂了...

但是乱搞一下还是有80分....可惜没想到正解啊！

所以今天的考试题是：

巡访 (path.pas/c/cpp)

　　Chanxer终于当上了“中华农民联盟”的盟主，他举目四望，决定四处走走，巡视自己的农土。

　　“中华农民联盟”的成员有个村庄，在“村村通”计划中，村庄们被条道路联通了起来，Chanxer计划从某个村庄出发，访问所有的村庄。

可是Chanxer出行有一个特殊的要求，那就是必须以农车代步，现在我们知道哪些村庄配备有农车，也就是说，只有配备有农车的村庄才能够被作为出发点。

Chanxer有点懒，他想知道访问全部的村庄所要走的路程长度最小是多少。

树的节点数 n<=10^5

 

题目大意：已知一棵树，求一条满足一个端点为给定的端点的最长链。

 

其实就是求出每个点能到达的最远距离  [这就是我们的标题所给题目要求的]。

 

所以应该怎么做呢？

方法一：

　　经过证明，从树上任意一个点出发到达的最远的点一定是这棵树的直径的一个端点。

　　反过来，从直径的端点出发到达每个点的距离也一定是最远距离...  [我怎么就没想到...囧]

　　所以先找到两个端点[先从随意一个点出发，然后这个点一定是一个端点，端点的最远点就是另一个端点]，然后再跑一遍dfs就可以了[ 因为找第二个端点的时候已经跑过一遍了 ]...

　　

 

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<ctime>
#include<queue>
#include<algorithm>

using namespace std;

inline int in(){
     int x=0;char ch=getchar();
     while(ch>'9' || ch<'0') ch=getchar();
     while(ch>='0' && ch<='9') x=x*10+ch-'0',ch=getchar();
     return x;
}

const int maxn=100010;
const int INF=0x3f3f3f3f;

struct Node{
    int data,next,weight;
}node[maxn<<1];

#define now node[point].data
#define then node[point].next
#define www node[point].weight

int n,cnt;
long long M_dis,M_sit=1,M_dis1;
int star[maxn];
int head[maxn];
bool vis[maxn];
long long ans,Sum;
long long f[maxn];

inline void add(int u,int v,int w){
    node[cnt].data=v;node[cnt].next=head[u];node[cnt].weight=w;head[u]=cnt++;
    node[cnt].data=u;node[cnt].next=head[v];node[cnt].weight=w;head[v]=cnt++;
}

int dfs(int x,long long sum){
    vis[x]=true;
    if(sum>M_dis)
        M_dis=sum,M_sit=x;
    for(int point=head[x];point!=-1;point=then)
        if(!vis[now])
            dfs(now,sum+www);
    vis[x]=false;
}

int dfs2(int x,long long sum){
    vis[x]=true;f[x]=max(f[x],sum);
    if(sum>M_dis)
        M_dis=sum,M_sit=x;
    for(int point=head[x];point!=-1;point=then)
        if(!vis[now])
            dfs2(now,sum+www);
    vis[x]=false;
}

int main(){
    freopen("path.in","r",stdin);
    freopen("path.out","w",stdout);
    
    int u,v,w,cot=0;
    
    n=in();
    for(int i=1;i<=n;i++) head[i]=-1;
    for(int i=1;i<n;i++){
        u=in(),v=in(),w=in(),add(u,v,w);
        Sum+=w<<1;
    }
    for(int i=1;i<=n;i++) star[i]=in(),cot+=star[i];
    
    dfs(1,0);
    int t=M_sit;
    dfs2(t,0);
    dfs2(M_sit,0);
    
    for(int i=1;i<=n;i++)
        if(star[i])
            ans=max(ans,f[i]);
    
    ans=Sum-ans;
    
    printf("%lld",ans);
    
    return 0;
}

 

 

方法二：

　　

详情可以见代码

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

using namespace std;

inline int in(){
    int x=0;char ch=getchar();
    while(ch>'9' || ch<'0') ch=getchar();
    while(ch>='0' && ch<='9') x=x*10+ch-'0',ch=getchar();
    return x;
}

const int maxn=100010;

struct Node{
    int data,next,weight;
}node[maxn<<1];

#define now node[point].data
#define then node[point].next
#define www node[point].weight

int n,cnt,ans,Sum;
int head[maxn];
int Max[maxn],Maxv[maxn];
int Smax[maxn],Smaxv[maxn];

inline void add(int u,int v,int w){
    node[cnt].data=v;node[cnt].next=head[u];node[cnt].weight=w;head[u]=cnt++;
    node[cnt].data=u;node[cnt].next=head[v];node[cnt].weight=w;head[v]=cnt++;
}

void dfs(int x,int p){
    for(int point=head[x];point!=-1;point=then){
        if(now==p) continue;
        dfs(now,x);
        if(Smax[x]<Max[now]+www){
            Smax[x]=Max[now]+www,Smaxv[x]=now;
            if(Smax[x]>Max[x]){
                swap(Smax[x],Max[x]);
                swap(Smaxv[x],Maxv[x]);
            }
        }
    }
}

void dfs2(int x,int p){
    for(int point=head[x];point!=-1;point=then){
        if(now==p) continue;
        if(now==Maxv[x]){
            if(Smax[now]<Smax[x]+www){
                Smax[now]=Smax[x]+www;Smaxv[now]=x;
                if(Smax[now]>Max[now]){
                    swap(Smax[now],Max[now]);
                    swap(Smaxv[now],Maxv[now]);
                }
            }
        }
        else{
            if(Smax[now]<Max[x]+www){
                Smax[now]=Max[x]+www;Smaxv[now]=x;
                if(Smax[now]>Max[now]){
                    swap(Smax[now],Max[now]);
                    swap(Smaxv[now],Maxv[now]);
                }
            }
        }
        dfs2(now,x);
    }
}

int main(){
    freopen("path.in","r",stdin);
    freopen("path.out","w",stdout);
    
    int u,v,w;
    
    n=in();
    for(int i=1;i<=n;i++) head[i]=-1;
    for(int i=1;i<n;i++)
        u=in(),v=in(),w=in(),add(u,v,w),Sum+=(w<<1);
    
    dfs(1,-1);
    dfs2(1,-1);
    
    for(int i=1;i<=n;i++)
        if(u=in()) ans=max(ans,Max[i]);
            
    printf("%d",Sum-ans);
    
    return 0;
}

另附标题的AC通道：

http://acm.hdu.edu.cn/showproblem.php?pid=2196