51nod 1743雪之国度【最小生成树】【LCA】【并查集】

Description

雪之国度有N座城市，依次编号为1到N，又有M条道路连接了其中的城市，每一条道路都连接了不同的2个城市，任何两座不同的城市之间可能不止一条道路。
雪之女王赋予了每一座城市不同的能量，其中第i座城市被赋予的能量为Wi。
如果城市u和v之间有一条道路，那么只要此刻雪之女王的能量不小于|Wu-Wv|，这条道路就是安全的。
如果城市u和v之间存在两条没有重复道路的安全路径（其中每一段道路都是安全的），则认为这两座城市之间有着良好的贸易关系。
最近，雪之女王因为情感问题，她的能量产生巨大的波动。为了维持雪之国度的经济贸易，她希望你能帮忙对Q对城市进行调查。
对于第j对城市uj和vj，她希望知道在保证这两座城市之间有着良好贸易关系的前提之下，自己最少需要保持多少的能量。

题解

一看题目，感觉很像边双，做了半天发现不会。

先求最小生成树，然后将所有非树边从小到大添加进去，添加的时候只要把这条非树边与树形成的环上的节点缩成同一个边双，同时将经过的路径修正为当前非树边的权值，缩点的时候可以用并查集暴力维护，因为每个点只会被缩一次，所以复杂度是对的。最后答案查询的时候直接在树上查询路径最大值即可。

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define maxn 100006
#define maxe 500006
using namespace std;
inline char nc(){
    static char buf[100000],*i=buf,*j=buf;
    return i==j&&(j=(i=buf)+fread(buf,1,100000,stdin),i==j)?EOF:*i++;
}
inline int _read(){
    char ch=nc();int sum=0;
    while(!(ch>='0'&&ch<='9'))ch=nc();
    while(ch>='0'&&ch<='9')sum=sum*10+ch-48,ch=nc();
    return sum;
}
struct data{
    int x,y,w;
    bool operator <(const data&b)const{return w<b.w;}
}a[maxe];
int n,e,T,tot,w[maxn],lnk[maxn],son[maxn*2],nxt[maxn*2],
    fa[maxn],dep[maxn],p[maxn][18],f[maxn][18];
bool vis[maxn],use[maxe];
int get(int x){return fa[x]==x?x:fa[x]=get(fa[x]);}
void add(int x,int y){
    nxt[++tot]=lnk[x];son[tot]=y;lnk[x]=tot;
}
void dfs(int x){
    vis[x]=0;
    for(int j=lnk[x];j;j=nxt[j]) if(vis[son[j]])p[son[j]][0]=x,dep[son[j]]=dep[x]+1,dfs(son[j]);
}
inline int query(int x,int y){
    if(dep[x]<dep[y])swap(x,y);
    int sum=0;
    for(int j=log2(n);j>=0;j--) if(dep[x]-(1<<j)>=dep[y])sum=max(sum,f[x][j]),x=p[x][j];
    for(int j=log2(n);j>=0;j--) if(p[x][j]!=p[y][j])sum=max(sum,max(f[x][j],f[y][j])),x=p[x][j],y=p[y][j];
    if(x==y)return sum;
    return max(sum,max(f[x][0],f[y][0]));
}
int abs(int x){return x>0?x:-x;}
int main(){
    freopen("city.in","r",stdin);
    freopen("city.out","w",stdout);
    n=_read();e=_read();T=_read();
    for(int i=1;i<=n;i++)w[i]=_read();
    for(int i=1;i<=e;i++)a[i].x=_read(),a[i].y=_read(),a[i].w=abs(w[a[i].x]-w[a[i].y]);
    sort(a+1,a+1+e);
    for(int i=1;i<=n;i++)fa[i]=i;
    memset(use,1,sizeof(use));
    for(int i=1;i<=e;i++){
        int fx=get(a[i].x),fy=get(a[i].y);
        if(fx!=fy)use[i]=0,fa[fx]=fy,add(a[i].x,a[i].y),add(a[i].y,a[i].x);
    }
    memset(vis,1,sizeof(vis));
    dep[1]=1;dfs(1);
    for(int i=1;i<=n;i++)fa[i]=i;
    for(int i=1;i<=e;i++) if(use[i]){
        int x=get(a[i].x),y=get(a[i].y);
        while(x!=y){
            if(dep[x]<dep[y])swap(x,y);
            f[x][0]=a[i].w;
            fa[x]=p[x][0];x=get(x);
        }
    }
    for(int j=1;j<=log2(n);j++)
     for(int i=1;i<=n;i++)
      p[i][j]=p[p[i][j-1]][j-1],f[i][j]=max(f[i][j-1],f[p[i][j-1]][j-1]);
    while(T--){
        int x=_read(),y=_read();
        if(get(x)==get(y))printf("%d\n",query(x,y));else printf("infinitely\n");
    }
    return 0;
};