首页 > 代码库 > 2017CCPC中南地区赛 H题(最长路)

2017CCPC中南地区赛 H题(最长路)

题目地址:202.197.224.59/OnlineJudge2/

来自湘潭大学OJ,题号:1267。

这里用到了一个树的直径(树中的最长边)的结论:当你找到一棵树的最长边后,这个树中所有点的最长边必定和这条边的两个端点相连。下面给出证明:

设这条最长边的两个端点分别为B和E;

1.当选择的任意点M在这条最长边上时:如果此时还存在另一个点T,使得MT > max{MB,ME}。则:MT + min{MB,ME} > max{MB,ME} + min{MB,ME} = BE这与题目假设相矛盾。

2.当选择的任意点M不在这条最长边上时:

Α.与它相连的最长边与BE有交点时,假设交于X,则:M的最长边 = MX + X的最长边,而X在BE上,所以M的最长边 = MX + max{XB,XE}即它的最长边终止于BE中的一个点。

B.若无交点,假设M的最长边为MN,则:取BE上一点X,连接MX,有:MN > max{XB,XE} + XM,MN + MX + max{XB,XE} > 2MX + 2max{XB,XE} > BE与题设矛盾

 

由此,本题思路即为:先找到最长边,然后将其余的n - 2个点到最长边两个端点的距离算出,不断地挑选这n-2个点到两个端点的更长的那个路径,最后加上这条最长路就是所求结果。

下面的代码用C++11提交能过,而用G++则会WA

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#define maxn 100005
#define F 0x3f
using namespace std;
struct edge
{
    int len,over;
    edge(int a = 0,int b = 0)
    {
        len = a,over = b;
    }
};
long long dist[3][maxn];
vector<edge> graph[maxn];
inline void init()
{
    for(int i = 1;i < maxn;++i)
        graph[i].clear();
}
queue<int> q;
void bfs(int use,int start)
{
    while(q.size()) q.pop();
    q.push(start);
    dist[use][start] = 0;
    while(q.size()){
        int temp = q.front();
        q.pop();
        for(int i = 0;i < graph[temp].size();++i){
            int length = graph[temp][i].len,nextone = graph[temp][i].over;
            if(dist[use][nextone] == -1){
                dist[use][nextone] = dist[use][temp] + length;
                q.push(nextone);
            }
        }
    }
    return;
}
int main()
{
    int n;
    while(scanf("%d",&n) == 1){
        int start,over,len;
        init();
        for(int i = 0;i < n - 1;++i){
            scanf("%d%d%d",&start,&over,&len);
            graph[start].push_back(edge(len,over));
            graph[over].push_back(edge(len,start));
        }
        int point1,point2;
        memset(dist,-1,sizeof(dist));
        //printf("now dist is %lld\n",dist[0][0]);
        bfs(0,1);
        long long maxone = 0;
        for(int i = 2;i <= n;++i){
            if(dist[0][i] > maxone) maxone = dist[0][i],point1 = i;
        }
        maxone = 0;
        bfs(1,point1);
        for(int i = 1;i <= n;++i){
            if(dist[1][i] > maxone) maxone = dist[1][i],point2 = i;
        }
        bfs(2,point2);
        //printf("point1 is %d and point2 is %d\n",point1,point2);
        long long answer = 0;
        answer += dist[1][point2];
        for(int i = 1;i <= n;++i){
            if(i != point1 && i != point2){
                answer += max(dist[1][i],dist[2][i]);
            }
        }
        if(n == 1) answer = 0;
        printf("%lld\n",answer);
    }
    return 0;
} 

 

2017CCPC中南地区赛 H题(最长路)