ssl2779 Heatwave 最小生成树+倍增

题目大意

给你N个点的无向连通图，图中有M条边，第j条边的长度为： d_j.
现在有 K个询问。
每个询问的格式是：A B，表示询问从A点走到B点的所有路径中，最长的边最小值是多少？

分析

要求u,v两点中的最长边的最小值，设该值为ans，那么，把权小于等于ans的边组成的图中，u,v一定是连通的。也就是说，对于一个图，求任意两点的答案，它的路径，一定是在这个图的最小生成树上。然后，考虑每两个点，用树上倍增法，预处理出max[I,j]为从i往根走2^j步的最大值，类似lca的方法求答案即可。

ps：学习了如何树上倍增，我都已经忘光了。。。

code

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<queue>

using namespace std;

struct arr{
    int x,y,w,next;
}edge[300000];
int ls[200000];
int n,m,k;
int edge_m;

bool cmp(arr a,arr b) 
{
  return a.w<b.w; 
}

int f[200000]; 

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

void merge(int x,int y)
{
    int x1,y1;
    x1=find(x);
    y1=find(y);
    f[x1]=y1;
}

void add(int x,int y,int w)
{
    edge_m++;
    edge[edge_m]=(arr){x,y,w};
}

struct arr1{
    int x,y;
    int w;
    int next;
}edge1[2000000]; 
int ls1[2000000]; 
int nn;

void add1(int x,int y,int w) 
{
    nn++;
    edge1[nn].x=x;edge1[nn].y=y;edge1[nn].w=w;edge1[nn].next=ls1[x];
    ls1[x]=nn;

    nn++;
    edge1[nn].x=y;edge1[nn].y=x;edge1[nn].w=w;edge1[nn].next=ls1[y];
    ls1[y]=nn;
}

int fa[25000][20];
int d[25000][20];
int dep[25000];
int maxn;

int dfs(int x)
{
    for (int i=1;i<=maxn;i++)
        if (fa[x][i-1])
            fa[x][i]=fa[fa[x][i-1]][i-1],d[x][i]=max(d[fa[x][i-1]][i-1],d[x][i-1]);
    for (int i=ls1[x];i!=0;i=edge1[i].next)
    {
        if (edge1[i].y==fa[x][0])
            continue;
        fa[edge1[i].y][0]=x;
        d[edge1[i].y][0]=edge1[i].w;
        dep[edge1[i].y]=dep[x]+1;
        dfs(edge1[i].y);
    }
}

int dp(int x,int y)
{
    int ans=0;
    if (dep[x]<dep[y]) swap(x,y);
    int delta=dep[x]-dep[y];
    for (int i=0;i<=maxn;i++)
        if ((1<<i)&delta)
            ans=max(ans,d[x][i]),x=fa[x][i];
    if (x==y) return ans;
    for (int i=maxn;i>=0;i--)
        if (fa[x][i]!=fa[y][i])
        {
            ans=max(max(ans,d[x][i]),d[y][i]);
            x=fa[x][i];
            y=fa[y][i];
        }
    return max(max(ans,d[x][0]),d[y][0]);
}
int main()
{
    scanf("%d%d%d",&n,&m,&k);
    for (int i=1;i<=m;i++)
    {
        int x,y,w;
        scanf("%d%d%d",&x,&y,&w);
        add(x,y,w);
    }
    sort(edge+1,edge+m+1,cmp);
    for (int i=1;i<=n;i++) f[i]=i;
    int ans=0;
    int num=0;
    int i=1; nn=0;
    while ((num<n-1)&&(i<=m))
    {
        if (find(edge[i].x)!=find(edge[i].y))
        {
            merge(edge[i].x,edge[i].y);
            add1(edge[i].x,edge[i].y,edge[i].w);
            num++;
        }
        i++;
    }
    maxn=trunc(log2(n)); 
    dfs(1);
    for (int i=1;i<=k;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        printf("%d\n",dp(x,y));
    }
}