BZOJ-3732: Network (kruskal+LCA)

3732: Network

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 2092  Solved: 998
[Submit][Status][Discuss]

Description

给你N个点的无向图 (1 <= N <= 15,000)，记为：1…N。 
图中有M条边 (1 <= M <= 30,000) ，第j条边的长度为： d_j ( 1 < = d_j < = 1,000,000,000).

现在有 K个询问 (1 < = K < = 20,000)。 
每个询问的格式是：A B，表示询问从A点走到B点的所有路径中，最长的边最小值是多少？

Input

第一行： N, M, K。 
第2..M+1行: 三个正整数：X, Y, and D (1 <= X <=N; 1 <= Y <= N). 表示X与Y之间有一条长度为D的边。 
第M+2..M+K+1行: 每行两个整数A B,表示询问从A点走到B点的所有路径中，最长的边最小值是多少？

Output

 对每个询问，输出最长的边最小值是多少。

Sample Input

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

Sample Output

5
5
5
4
4
7
4
5

HINT

 

1 <= N <= 15,000 

1 <= M <= 30,000 

1 <= d_j <= 1,000,000,000 

1 <= K <= 15,000

 

Source

很显然最后的图为一棵最小生成树，然后求一条路径上的最大值即可，一开始想着求最大值用暴力跑一遍，觉得实在是卡不过去，算了吧，这里的最大值可以在LCA的过程中求出，用gg[x][i]表示从第x个点开始到向上(1<<i)个点的路径中的最大值，注意预处理的时候gg[x][i]=max(gg[x][i-1],gg[ fa[x][i-1] ][i-1]); max中第二个gg里面第一维是fa[x][i-1]！！！一开始laj把搞成gg[x][i-1]了 _(:зゝ∠)_  不过可喜的是交上去的时候一遍过了ovo

#include "bits/stdc++.h"
using namespace std;
typedef long long LL;
const int MAX1=15005;
const int MAX2=30005;
int n,m,K,f[MAX1];
int tot,head[MAX1],adj[MAX2<<1],wei[MAX2<<1],next[MAX2<<1];
int fa[MAX1][21],gg[MAX1][21],deep[MAX1];
struct Edge{
    int u,v,w;
    bool operator < (const Edge &tt) const {
        return w<tt.w;
    }
}edge[MAX2];
inline int read(){
    int an=0,x=1;char c=getchar();
    while (c<'0' || c>'9') {if (c=='-') x=-1;c=getchar();}
    while (c>='0' && c<='9') {an=(an<<3)+(an<<1)+c-'0';c=getchar();}
    return an*x;
}
inline int getfather(int x){return f[x]==x?x:f[x]=getfather(f[x]);}
void addedge(int u,int v,int w){
    tot++,adj[tot]=v,wei[tot]=w,next[tot]=head[u],head[u]=tot;
}
void dfs(int x,int ff){
    int i,j;
    for (i=1;i<=20;i++){
        if (deep[x]<(1<<i)) break;
        fa[x][i]=fa[ fa[x][i-1] ][i-1];
        gg[x][i]=max(gg[x][i-1],gg[ fa[x][i-1] ][i-1]);
    }
    for (i=head[x];i;i=next[i]){
        if (adj[i]!=ff){
            deep[adj[i]]=deep[x]+1;
            fa[adj[i]][0]=x;gg[adj[i]][0]=wei[i];
            dfs(adj[i],x);
        }
    }
}
int lca(int x,int y){
    int an=0,i,j;
    if (deep[x]<deep[y]) swap(x,y);
    int dd=deep[x]-deep[y];
    for (i=20;i>=0;i--)
        if (dd&(1<<i))
            an=max(an,gg[x][i]),x=fa[x][i];
    for (i=20;i>=0;i--){
        if (fa[x][i]!=fa[y][i]){
            an=max(an,max(gg[x][i],gg[y][i]));
            x=fa[x][i],y=fa[y][i];
        }
    }
    if (x!=y) an=max(an,max(gg[x][0],gg[y][0])),x=fa[x][0];
    return an;
}
int main(){
    freopen ("network.in","r",stdin);freopen ("network.out","w",stdout);
    int i,j,u,v;tot=1;
    n=read();m=read();K=read();
    for (i=1;i<=m;i++) edge[i].u=read(),edge[i].v=read(),edge[i].w=read();
    sort(edge+1,edge+m+1);
    for (i=1;i<=n;i++) f[i]=i;
    for (i=1;i<=m;i++){
        int tx=getfather(edge[i].u);
        int ty=getfather(edge[i].v);
        if (tx!=ty){
            f[tx]=ty;
            addedge(edge[i].u,edge[i].v,edge[i].w);addedge(edge[i].v,edge[i].u,edge[i].w);
        }
    }
    dfs(1,0);
    for (i=1;i<=K;i++){
        u=read(),v=read();
        printf("%d\n",lca(u,v));
    }
    return 0;
}