洛谷P2195 HXY造公园

题目描述

现在有一个现成的公园，有n个休息点和m条双向边连接两个休息点。众所周知，HXY是一个SXBK的强迫症患者，所以她打算施展魔法来改造公园并即时了解改造情况。她可以进行以下两种操作：

1、对某个休息点x，查询公园中所有经过该休息点的路径中的最长路径。

2、对于两个休息点x、y，如果x，y已经可以互相到达则忽略此次操作。否则，在x可到达的所有休息点和y可到达的所有休息点（包括x，y自身）分别选择一个休息点，然后在这两个休息点之间连一条边，并且这个选择应该满足对于连接后的公园，x和y所在的区域（即x，y可达到的所有休息点和边组成的集合）中的最长路径的长度最小。

HXY打算进行q个操作，请你回答她的对于公园情况的询问（操作1）或者执行她的操作（操作2）。

注：所有边的长度皆为1。保证不存在环。最长路径定义为：对于点v1,v2……vk，如果对于其中任意的vi和vi+1(1<=i<=k-1)，都有边相连接，那么vj(1<=j<=k)所在区域的最长路径就是k-1。

输入输出格式

输入格式：
第一行，三个正整数，分别为n，m，q。

接下来的m行，每一行有两个正整数xi，yi，表示xi和yi有一条双向边相连。

再接下来的q行，每一行表示一个操作。

若该行第一个数为1，则表示操作1，之后还有一个正整数xi，表示要查询的休息点。

若该行第一个数为2，则表示操作2，之后还有两个正整数xi，yi，表示需要执行操作的两个休息点。

输出格式：
输出行数为操作1的个数。

每行输出对于操作1询问的回答。

输入输出样例

输入样例#1： 复制
6 0 6
2 1 2
2 3 4
2 5 6
2 3 2
2 5 3
1 1
输出样例#1： 复制
4

说明

数据范围：

对于10%的数据，只存在操作1。

对于30%的数据，1<=m

题解：

洛谷题解里面讲的很详细了，大家可以看看。
代码：

#include<bits/stdc++.h>
using namespace std;
const int N=300010;
int n,m,que,tot,head[N],path[N],d[N],flag[N],f[N],q[N];
struct node{
    int next,to;
}e[N*2];
inline void ins(int from,int to){
    e[++tot].next=head[from];
    e[tot].to=to; head[from]=tot;
}
int get(int x){
    return (x==f[x])? x : f[x]=get(f[x]);
}
int find(int x)  
{  
    d[x]=0;  
    flag[x]=x;  
    q[q[0]=1]=x;  
    int yuan=x;  
    for(int i=1;i<=q[0];i++)  
    {  
        int y=q[i];  
        for(int u=head[y];u;u=e[u].next)  
        if(flag[e[u].to]!=x)  
        {  
            flag[e[u].to]=x;  
            q[++q[0]]=e[u].to;  
            d[e[u].to]=d[y]+1;  
            if(d[e[u].to]>d[yuan])   
            yuan=e[u].to;  
        }  
    }  
    int y2=yuan;  
    d[yuan]=0;  
    q[q[0]=1]=yuan;  
    flag[yuan]=yuan;  
    for(int i=1;i<=q[0];i++)  
    {  
        int y=q[i];  
        for(int u=head[y];u;u=e[u].next)  
        if(flag[e[u].to]!=yuan)  
        {  
            flag[e[u].to]=yuan;  
            q[++q[0]]=e[u].to;  
            d[e[u].to]=d[y]+1;  
            if(d[e[u].to]>d[y2])  
            y2=e[u].to;  
        }  
    }  
    return d[y2];  
}  
int main(){
    scanf("%d%d%d",&n,&m,&que);
    for(int i=1;i<=n;++i) f[i]=i;
    for(int i=1;i<=m;++i){
        int x,y; scanf("%d%d",&x,&y);
        ins(x,y); ins(y,x);
        if(get(x)!=get(y))
        f[get(x)]=f[get(y)];
    }
    for(int i=1;i<=n;++i)
    if(i==get(i)) path[i]=find(i);
    for(int i=1;i<=que;++i){
        int pos; scanf("%d",&pos);
        if(pos==1){
            int x; scanf("%d",&x);
            printf("%d\n",path[get(x)]);
        }else {
            int x,y; scanf("%d%d",&x,&y);
            x=get(x); y=get(y);
            if(x==y) continue;
            int r1=path[x]; int r2=path[y];
            f[x]=y;
            path[y]=(r1+1)/2+(r2+1)/2+1;
            if(path[y]<r1) path[y]=r1;
            if(path[y]<r2) path[y]=r2;
        }
    }
    return 0;
}