P4897 【模板】最小割树（Gomory-Hu Tree）

据说这题数据有误，给定的真实点数是n+1个，编号为0–n。
实际上我做的时候，没有考虑0号节点（可能是查询里面没有0号节点，只是图出了问题）。
主要学习一下怎么建立最小割树。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
#include<vector>
#include<map>
#include<set>
#include<cmath>
#include<queue>
#include<bitset>
#define ll long long
#define llu unsigned ll
#define pr make_pair
#define pb push_back
//#define lc (cnt<<1)
//#define rc (cnt<<1|1)
using namespace std;
const int inf=0x3f3f3f3f;
const ll lnf=0x3f3f3f3f3f3f3f3f;
const double dnf=1e15;
const int mod=1e9+7;
const double eps=1e-8;
const int maxn=550;
const int maxx=3010;
int n,m;

struct Flow
{
    int head[maxn],ver[maxx],edge[maxx],nt[maxx];
    int d[maxn];
    int s,t,tot=1;
    queue<int>q;


    void add(int x,int y,int z)
    {
        ver[++tot]=y,edge[tot]=z,nt[tot]=head[x],head[x]=tot;
        ver[++tot]=x,edge[tot]=z,nt[tot]=head[y],head[y]=tot;
    }

    void init(void)
    {
        for(int i=2;i<tot;i+=2)
        {
            edge[i]=(edge[i]+edge[i^1])/2;
            edge[i^1]=edge[i];
        }
    }

    bool bfs(void)
    {
        memset(d,0,sizeof(d));
        while(q.size()) q.pop();

        q.push(s);
        d[s]=1;

        while(q.size())
        {
            int x=q.front();
            q.pop();

            for(int i=head[x];i;i=nt[i])
            {
                int y=ver[i];
                if(edge[i]&&!d[y])
                {
                    q.push(y);
                    d[y]=d[x]+1;
                    if(y==t) return true;
                }
            }
        }
        return false;
    }

    int dinic(int x,int flow)
    {
        if(x==t) return flow;
        int rest=flow,k;

        for(int i=head[x];i&&rest;i=nt[i])
        {
            int y=ver[i];
            if(edge[i]&&d[y]==d[x]+1)
            {
                k=dinic(y,min(rest,edge[i]));
                if(!k) d[y]=0;

                edge[i]-=k;
                edge[i^1]+=k;
                rest-=k;
            }
        }
        return flow-rest;
    }

    int get_maxx(void)
    {
        init();
        int flow=0,maxflow=0;
        while(bfs())
            while(flow = dinic(s,inf))
                maxflow += flow;
        return maxflow;
    }
}fl;

struct Tree
{
    int head[maxn],ver[maxn<<1],edge[maxn<<1],nt[maxn<<1];
    int f[maxn][12],minn[maxn][12],d[maxn],tot=1,t;
    int a[maxn],tmp1[maxn],tmp2[maxn];

    void init(void)
    {
        t=log2(n)+1;
        d[0]=-1;
        for(int i=1;i<=n;i++)
            a[i]=i;
    }

    void add(int x,int y,int z)
    {
        ver[++tot]=y,edge[tot]=z,nt[tot]=head[x],head[x]=tot;
        ver[++tot]=x,edge[tot]=z,nt[tot]=head[y],head[y]=tot;
    }

    void build(int l,int r)
    {
        if(l==r) return ;
        fl.s=a[l],fl.t=a[l+1];
        add(a[l],a[l+1],fl.get_maxx());

        int cnt1=0,cnt2=0;
        //考虑dinic算法中的最后一次bfs,因为现在残量网络上s到达不了t,
        //所以bfs访问到的点就是s所在的点集，它们的deep不0
        for(int i=l;i<=r;i++)
        {
            if(fl.d[a[i]]) tmp1[++cnt1]=a[i];
            else tmp2[++cnt2]=a[i];
        }
        for(int i=l;i<=l+cnt1-1;i++) a[i]=tmp1[i-l+1];
        for(int i=l+cnt1;i<=r;i++) a[i]=tmp2[i-cnt1-l+1];
        build(l,l+cnt1-1);
        build(l+cnt1,r);
    }

    void dfs(int x,int fa)
    {
        for(int i=head[x];i;i=nt[i])
        {
            int y=ver[i],z=edge[i];
            if(y==fa) continue;
            d[y]=d[x]+1;
            f[y][0]=x;
            minn[y][0]=z;
            for(int j=1;j<=t;j++)
            {
                f[y][j]=f[f[y][j-1]][j-1];
                minn[y][j]=min(minn[y][j-1],minn[f[y][j-1]][j-1]);
            }
            dfs(y,x);
        }
    }

    int ask_min(int x,int y)
    {
        int ans=inf;
        if(d[x]>d[y]) swap(x,y);
        for(int i=t;i>=0;i--)
        {
            if(d[f[y][i]]>=d[x])
            {
                ans=min(ans,minn[y][i]);
                y=f[y][i];
            }
        }
        if(x==y) return ans;
        for(int i=t;i>=0;i--)
        {
            if(f[x][i]!=f[y][i])
            {
                ans=min(ans,min(minn[x][i],minn[y][i]));
                x=f[x][i],y=f[y][i];
            }
        }
        return min(ans,min(minn[x][0],minn[y][0]));
    }
}tr;

int main(void)
{
    scanf("%d%d",&n,&m);
    int x,y,z;
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&x,&y,&z);
        fl.add(x,y,z);
    }
    tr.init();
    tr.build(1,n);
    tr.dfs(1,0);

    int q;
    scanf("%d",&q);
    for(int i=1;i<=q;i++)
    {
        scanf("%d%d",&x,&y);
        printf("%d\n",tr.ask_min(x,y));
    }
    return 0;
}