COdevs1700 施工方案第二季

题目描述 Description
c国边防军在边境某处的阵地是由n个地堡组成的。工兵连受命来到阵地要进行两期施工。

第一期的任务是挖掘暗道让所有地堡互联互通。现已勘测设计了m条互不相交的暗道挖掘方案，如果这m条暗道都实施挖掘，肯定能达到互联互通的目的。事实上，适当选择其中n-1个方案挖掘，就能实现互联互通，即从每个地堡出发都能到达其他任何一个地堡（允许经过别的地堡）。

连长精心谋算，在m个设计规划中选取了挖掘总距离最短且能保证互联互通的若干个暗道规划实施了挖掘，完成了第一期的施工任务后又接受了第二期的施工任务，要求选择一个地堡进行扩建改造，使其能向每个地堡提供弹药。为了让弹药供应更及时、更快捷，从改扩建的地堡到最远地堡的距离（称为最远输送距离）应当尽量小。

你的任务是先求出第一期施工挖掘的总距离，再求改扩建地堡最远输送距离的最小值。

输入描述 Input Description
其中第一行是n和m，m>=n
下面的m行每行3个数xi、yi、zi,表示xi到yi的距离是zi
zi<1000000且m个距离互不相等

输出描述 Output Description
共包含两行，每行一个整数，
第一行是第一期的挖掘总距离，第二行是最远输送距离的最小值。

样例输入 Sample Input
4 5
1 2 1
2 3 2
3 4 3
4 1 4
3 1 5

样例输出 Sample Output
6
3

数据范围及提示 Data Size & Hint
【样例说明】
第一期挖掘1到2、2到3和3到4的3条暗道，第二期选择3号地堡进行改扩建，最远输送距离是3
【数据规模】
60%的数据 n<10且m<20
80%的数据 n<1000且m<2000
100%的数据 n<100000且m<200000

这题。。mdzz。。。居然坑在ll上。。我的心好累。。。
对于第一问。Kruskal完美解决。
第二问。我们有这么一个结论。任意一点搜到最远点i。从i开始搜，再次搜到最远点，这个路径就是树的直径。那么我们根据树的重心的性质可以得出。如果要建弹药库，一定会建在重心上，而重心就在直径上，所以很容易求出重心。

#include<cstdio>
#include<algorithm>
#include<vector>
#include<cstring>
#include<cmath>

using namespace std;
struct bian {
    int x,y,z;
};
struct dian {
    int x,z;
};

int comp(bian a,bian b)
{
    return a.z<b.z;
}

long long f[100005],n,m,t,ans=0,mark[100005],location,location1,da=-1,sum,zhi;
bian s[200005];
vector <dian> tree[100005];

int find (int a)
{
  if(f[a]==a)return a;
  else return f[a]=find(f[a]);
}

int search (int x,int step) 
{
   mark[x]=1;
   for(int i=0;i<tree[x].size();i++){
        if(!mark[tree[x][i].x])search(tree[x][i].x,step+tree[x][i].z);
     }
     if(step>da){da=step;location=x;}
}

int zhao (int x)
{
   mark[x]=1;
   if(x==location)return 1;
   int possible=0;
   for(int i=0;i<tree[x].size();i++){
        if(!mark[tree[x][i].x])possible=zhao(tree[x][i].x);
        if(possible==1){mark[x]=2;sum+=tree[x][i].z;return 1;}
     }     
}

int zuizhong(int x,int step)
{
   for(int i=0;i<tree[x].size();i++)
   {
        if(mark[tree[x][i].x]==2)
        {
            mark[tree[x][i].x]=3;
        if(fabs(sum/2-step)<fabs(sum/2-step-tree[x][i].z))
        {
            zhi=step;
            return 0;
        }
        zuizhong(tree[x][i].x,step+tree[x][i].z);
        }
     }
}

int main(){
   scanf("%lld %lld",&n,&m);
   for(int i=0;i<m;i++)
   {
       scanf("%d %d %d",&s[i].x,&s[i].y,&s[i].z);
         if(s[i].x>s[i].y)
         {
            t=s[i].x;
            s[i].x=s[i].y;
            s[i].y=t;
         }
     }
     sort(s,s+m,comp);
     dian now;
     for(int i=1;i<=n;i++)f[i]=i;
     for(int i=0;i<m;i++)
     {
        if(find(s[i].x)!=find(s[i].y))
        {
           f[find(s[i].x)]=find(s[i].y);
           ans+=s[i].z;
           now.z=s[i].z;
           now.x=s[i].x;
           tree[s[i].y].push_back(now);
           now.x=s[i].y;
           tree[s[i].x].push_back(now);
            }
     }
     printf("%lld\n",ans);
     search(1,0);
     da=-1;
     location1=location;
     for(int i=1;i<=n;i++)mark[i]=0;
     search(location,0);
     for(int i=1;i<=n;i++)mark[i]=0;
     zhao(location1);
     zuizhong(location1,0);
     if(zhi<sum/2)zhi=sum-zhi;
     printf("%lld",zhi);
}