JZOJ4899. 【NOIP2016提高A组集训第17场11.16】雪之国度

Description

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

Input

每一组数据第一行有3个整数，依次为N,M,Q，表示城市个数，道路个数，和所需要进行的调查次数。
之后一行，有N个整数，依次为每一个城市被赋予的能量Wi。
之后M行，每一行有2个整数，表示对应编号的两个城市之间有一条道路。
之后Q行，每一行有2个整数，表示一组调查的城市目标。
（保证图联通！）

Output

输出一共有Q行，依次对应Q次调查的结果。其中第j行给出了第j次调查的结果，即雪之女王需要保持的最少能量值。如果永远也无法做到，输出”infinitely”。

Sample Input

7 8 4
3 2 4 1 3 5 9
1 2
1 3
2 4
2 5
3 6
6 7
4 6
5 6
4 5
4 6
5 6
2 7

Sample Output

4
4
2
infinitely

Data Constraint

对于20％的数据来说，3<=N<=10, 3<=M<=20, 1<=Q<=10
对于另30％的数据来说，Wi=0
对于100％的数据来说，3<=N<=100000, 3<=M<=500000, 1<=Q<=100000, 每一座城市的能量Wi满足0<=Wi<=200000.

题解

很显然，答案就是在最小生成树上面，
对于每个联通块就行单独处理。

先按照边长做一棵最小生成树，
那些剩下的边，也应该考虑进去，
最后用倍增求lca。

code

#include<queue>
#include<cstdio>
#include<iostream>
#include<algorithm>
#include <cstring>
#include <string.h>
#include <cmath>
#include <math.h>
#include <time.h> 
#define ll long long
#define N 1000003
#define M 103
#define db double
#define P putchar
#define G getchar
#define inf 998244353
using namespace std;
char ch;
void read(int &n)
{
    n=0;
    ch=G();
    while((ch<'0' || ch>'9') && ch!='-')ch=G();
    ll w=1;
    if(ch=='-')w=-1,ch=G();
    while('0'<=ch && ch<='9')n=(n<<3)+(n<<1)+ch-'0',ch=G();
    n*=w;
}

int max(int a,int b){return a>b?a:b;}
int min(int a,int b){return a<b?a:b;}
int abs(int x){return x<0?-x:x;}
ll sqr(ll x){return x*x;}
void write(ll x){if(x>9) write(x/10);P(x%10+'0');}

struct node
{
    int x,y,z;
}a[N*5];

bool cmp(node a,node b){return a.z<b.z;}

int n,m,q,v[N*2],nxt[N*2],to[N*2],val[N],last[N*2];
int f[17][N],g[17][N],deep[N],x,y,fa[N],tot,t;
bool p[N*5];

int lca(int x,int y)
{
    int mx=0;
    if(deep[y]>deep[x])swap(x,y);
    for(int i=16;i>=0;i--)
        if(deep[f[i][x]]>=deep[y])mx=max(mx,g[i][x]),x=f[i][x];

    if(x==y)return mx;

    for(int i=16;i>=0;i--)
        if(f[i][x]!=f[i][y])mx=max(mx,g[i][x]),x=f[i][x],mx=max(mx,g[i][y]),y=f[i][y];

    mx=max(mx,g[0][x]);mx=max(mx,g[0][y]);
    return mx;
}

int get(int x){return fa[x]=(fa[x]==x?x:get(fa[x]));}

void ins(int x,int y,int z)
{
    nxt[++tot]=last[x];
    to[tot]=y;
    v[tot]=z;
    last[x]=tot;
}

void dfs(int x)
{
    for(int i=last[x];i;i=nxt[i])
        if(f[0][x]!=to[i])f[0][to[i]]=x,g[0][to[i]]=v[i],
            deep[to[i]]=deep[x]+1,dfs(to[i]);
}

void pre()
{
    for(int j=1;j<17;j++)
        for(int i=1;i<=n;i++)
            f[j][i]=f[j-1][f[j-1][i]],g[j][i]=max(g[j-1][i],g[j-1][f[j-1][i]]);
}

int main()
{
    freopen("city11.in","r",stdin);
    freopen("city.out","w",stdout);
    read(n);read(m);read(q);
    for(int i=1;i<=n;i++)
        read(val[i]),fa[i]=i;

    for(int i=1;i<=m;i++)
        read(a[i].x),read(a[i].y),a[i].z=abs(val[a[i].x]-val[a[i].y]);

    sort(a+1,a+1+m,cmp);
    memset(p,0,sizeof(p));

    for(int i=1;i<=m;i++)
    {
        x=get(a[i].x);y=get(a[i].y);
        if(x!=y)
        {
            ins(a[i].x,a[i].y,a[i].z);
            ins(a[i].y,a[i].x,a[i].z);
            fa[x]=y;
            p[i]=1;
        }
    }

    deep[1]=1;
    dfs(1);

    tot=0;memset(last,0,sizeof(last));
    for(int i=1;i<=n;i++)fa[i]=i;

    for(int i=1;i<=m;i++)
        if(!p[i])
        {
            t=x=get(a[i].x);
            y=get(a[i].y);
            for(;t!=y;t=get(f[0][t]))
                if(deep[t]<deep[y])swap(t,y);

            y=get(a[i].y);
            for(;x!=y;)
            {
                if(deep[x]<deep[y])swap(x,y);
                ins(t,x,a[i].z);
                fa[x]=t;
                x=get(f[0][x]);
            }
        }

    memset(deep,0,sizeof(deep));
    for(int i=1;i<=n;i++)
        if(!deep[i])deep[i]=1,dfs(i);
    pre();

    for(int i=1;i<=q;i++)
    {
        read(x);read(y);
        if(get(x)!=get(y))printf("infinitely");
            else write(lca(x,y));
        P('\n');
    }
}