树剖

今天做了一道题目：提高组2013年Day1T3。

这题有一种LCA的做法，我之前写过，最近新学了树链剖分，心血来潮写了一下，发现要把边权转换成点权（因为树剖好像只能解决点权的问题，反正我是这么理解的）

发现有好多好多的坑点

套路和注意点都在代码旁边的批注上

代码如下：

#include<iostream>
#include<cstring>
#include<string>
#include<algorithm>
#include<cstdio>
#include<cmath>
#define N 50005
using namespace std;
int head[N],x[N],y[N],dd[N],d[N],b[N],c[N],a[N],top[N],id[N],father[N],depth[N],son[N],heavyson[N],aa[N],real[N];
int n,m,Q,xx,yy;
struct Edge{
    int next,to,step;
}e[50005];
int kkk=0;
void add(int x,int y,int z)
{
    kkk++;
    e[kkk].next=head[x];
    e[kkk].step=z;
    e[kkk].to=y;
    head[x]=kkk;
}
void qsort(int l,int r)
{
    int i=l; int j=r;
    int mid=dd[(l+r)/2];
    while (i<=j){
        while (dd[i]<mid) i++;
        while (dd[j]>mid) j--;
        if (i<=j){
            int p=x[i]; x[i]=x[j]; x[j]=p;
            p=y[i]; y[i]=y[j]; y[j]=p;
            p=dd[i]; dd[i]=dd[j]; dd[j]=p;
            i++; j--;
        }
    }
    if (l<j) qsort(l,j);
    if (i<r) qsort(i,r);
}
int find(int x)
{
    if (x==a[x]) return x;
    a[x]=find(a[x]);
    x=a[x];
}
void make(int x,int y)
{
    a[find(x)]=find(y);
}
void Dfs1(int now,int fa)
{
    father[now]=fa;
    depth[now]=depth[fa]+1;
    son[now]=1;
    for (int i=head[now];i;i=e[i].next){
        int v=e[i].to;
        if (v==fa) continue;
        aa[v]=e[i].step;//套路，一般都把边权赋给v（到达的节点） 
        Dfs1(v,now);
        son[now]=son[now]+son[v];
        if (!heavyson[now]||son[v]>son[heavyson[now]]) heavyson[now]=v;
    }
}
int temp=0;
void Dfs2(int now,int first)
{
    temp++;
    top[now]=first;
    real[temp]=now;
    id[now]=temp;
    if (!heavyson[now]) return;
    Dfs2(heavyson[now],first);
    for (int i=head[now];i;i=e[i].next){
        int v=e[i].to;
        if (v!=heavyson[now]&&v!=father[now]) Dfs2(v,v);
    }
}
void build(int k,int x,int y)
{
    b[k]=x; c[k]=y;
    if (x==y){
        d[k]=aa[real[x]];
        return;
    }
    int mid=(x+y)/2;
    build(k*2,x,mid);
    build(k*2+1,mid+1,y);
    d[k]=min(d[k*2],d[k*2+1]);
}
int sum(int k,int x,int y)
{
    if (x<=b[k]&&y>=c[k]) return d[k];
    int mid=(b[k]+c[k])/2;
    if (y<=mid) return sum(k*2,x,y);
    if (x>mid) return sum(k*2+1,x,y);
    if (x<=mid&&y>mid) return min(sum(k*2,x,mid),sum(k*2+1,mid+1,y)); 
}
int find1(int u,int v)
{
    int ans=1000000;
    int tou=top[u];
    int tov=top[v];
    while (tou!=tov){
        if (depth[tou]<depth[tov]) swap(tou,tov),swap(u,v);
        ans=min(ans,sum(1,id[tou],id[u]));
        u=father[tou];
        tou=top[u];
    }
    if (depth[u]>depth[v]) swap(u,v);
    if (id[u]==id[v]) return ans;//id可能会相等，然后就gg，这个调了一个上午 
	else ans=min(ans,sum(1,id[u]+1,id[v]));//这个就是套路，一定要注意
    return ans;
}
int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=m;i++) scanf("%d%d%d",&x[i],&y[i],&dd[i]);
    qsort(1,m);
    for (int i=1;i<=n;i++) a[i]=i;
    int kk=0;
    for (int i=m;i>=1;i--){
        if (find(x[i])!=find(y[i])){
            kk++;
            make(x[i],y[i]);
            add(y[i],x[i],dd[i]);
            add(x[i],y[i],dd[i]);
        }
        if (kk==n-1) break;
    }
    memset(depth,0,sizeof(depth));
    for (int i=1;i<=n;i++){
        if (!depth[i]){
            depth[i]=1;
            Dfs1(i,0);
            Dfs2(i,i);
        }
    }
    build(1,1,n);
    scanf("%d",&Q);
    while (Q--){
        scanf("%d%d",&xx,&yy);
        if (find(xx)!=find(yy)) printf("%d\n",-1);
        else printf("%d\n",find1(xx,yy));
    }
    return 0;
}