P1967 货车运输

P1967 货车运输
最大生成树+lca+并查集

#include<iostream>
#include<cstdio>
#include<queue>
#include<algorithm>
#include<cmath>
#include<ctime>
#include<set>
#include<map>
#include<stack>
#include<cstring>
#define inf 2147483647
#define For(i,a,b) for(register int i=a;i<=b;i++)
#define p(a) putchar(a)
#define g() getchar()
//by war
//2017.11.8
using namespace std;
int n,m,q;
int x,y,z;
int cnt;
int Min[10010][15];
int f[10010][15];
int d[10010];
int deep[10010];

struct kru
{
    int l,r,v;
    bool operator<(const kru&a)const
    {
        return v>a.v;
    }
}E[50010];

struct node
{
    int n;
    int v;
    node *next;
}*e[50010];

inline void in(register int &x)
{
    int y=1;
    char c=g();x=0;
    while(c<'0'||c>'9')
    {
    if(c=='-')
    y=-1;
    c=g();
    }
    while(c<='9'&&c>='0')x=(x<<1)+(x<<3)+c-'0',c=g();
    x*=y;
}
inline void o(register int x)
{
    if(x<0)
    {
        p('-');
        x=-x;
    }
    if(x>9)o(x/10);
    p(x%10+'0');
}

inline int find(register int x)
{
    if(d[x]==x)return x;
    d[x]=find(d[x]);
    return d[x];
}

inline void push(register int x,register int y,register int v)
{
    node *p;
    p=new node();
    p->n=y;
    p->v=v;
    if(e[x]==NULL)
    e[x]=p;
    else
    {
        p->next=e[x]->next;
        e[x]->next=p;
    }
}

inline void build(register int now)
{
    deep[now]=deep[f[now][0]]+1;
    for(register int i=1;(1<<i)<=n;i++)
    {
        Min[now][i]=min(Min[now][i-1],Min[f[now][i-1]][i-1]);
        f[now][i]=f[f[now][i-1]][i-1];
    }
    for(node *i=e[now];i!=NULL;i=i->next)
    {
        if(i->n!=f[now][0])
        {
            f[i->n][0]=now;
            Min[i->n][0]=i->v;
            build(i->n);
        }
    }
}

int query(int x,int y)
{
    int MIN=inf;
    if(deep[x]<deep[y])
    swap(x,y);
    int c=deep[x]-deep[y];
    for(register int i=0;(1<<i)<=c;i++)
    if((1<<i)&c)
    {
    MIN=min(MIN,Min[x][i]);
    x=f[x][i];    
    }
    if(x==y)
    return MIN;
    c=log2(deep[x]);
    for(register int i=c;i>=0;i--)
    {
        if(f[x][i]!=f[y][i])
        {
            MIN=min(MIN,Min[x][i]);
            MIN=min(MIN,Min[y][i]);
            x=f[x][i];
            y=f[y][i];
        }
    }
    return min(MIN,min(Min[y][0],Min[x][0]));
}

int main()
{
    in(n),in(m);
    For(i,1,n)
    d[i]=i;
    For(i,1,m)
    in(E[i].l),in(E[i].r),in(E[i].v);
    sort(E+1,E+m+1);
    For(i,1,m)
    {
        if(find(E[i].l)!=find(E[i].r))
        {
            d[find(E[i].l)]=find(E[i].r);
            cnt++;
            push(E[i].l,E[i].r,E[i].v);
            push(E[i].r,E[i].l,E[i].v);
        }
        if(cnt==n-1)
        break;
    }
    Min[1][0]=inf;
    build(1);
    in(q);
    For(i,1,q)
    {
        in(x),in(y);
        if(find(x)!=find(y))
        o(-1),p('
');
        else
        o(query(x,y)),p('
');
    }
     return 0;
}