P3329 [ZJOI2011]最小割

【题意】

多次询问最小割小于等于x的点对数

【分析】

最小割树建好后暴力统计即可

【代码】

#include<bits/stdc++.h>
using namespace std;
#define mp make_pair
#define fi first
#define se second
#define lson now<<1
#define rson now<<1|1
typedef long long ll;
const int maxn=155;
const int maxm=3005;
int n,m,q;
struct edge
{
    int to,nxt,cap,fl;
};
int node[maxn],color[maxn];
int ans[maxn][maxn];
const int inf=0x3f3f3f3f;
namespace Dinic
{
    int head[maxn],tot=1,S,T,cur[maxn],dep[maxn];
    edge e[maxm<<1];
    void add(int x,int y,int z)
    {
        e[++tot].to=y; e[tot].nxt=head[x]; head[x]=tot; e[tot].cap=z;
        e[++tot].to=x; e[tot].nxt=head[y]; head[y]=tot; e[tot].cap=z;
    }
    bool bfs()
    {
        memset(dep,-1,sizeof(dep));
        queue <int> q;
        dep[S]=0; q.push(S);
        while(!q.empty())
        {
            int u=q.front(); q.pop();
            cur[u]=head[u];
            for(int i=head[u];i;i=e[i].nxt)
            {
                int to=e[i].to;
                if(dep[to]==-1 && e[i].cap>e[i].fl)
                {
                    dep[to]=dep[u]+1;
                    q.push(to);
                }
            }
        }
        return (dep[T]!=-1);
    }
    int dfs(int u,int flow)
    {
        if(u==T) return flow;
        int res=0;
        for(int &i=cur[u];i;i=e[i].nxt)
        {
            int to=e[i].to;
            if(dep[to]==dep[u]+1 && e[i].cap>e[i].fl)
            {
                int tmp=dfs(to,min(flow,e[i].cap-e[i].fl));
                if(tmp)
                {
                    flow-=tmp; res+=tmp;
                    e[i].fl+=tmp; e[i^1].fl-=tmp;
                    if(!flow) break;
                }
            }
        }
        if(!flow) dep[u]=-1;
        return res;
    }
    int dinic()
    {
        int res=0;
        for(int i=1;i<=tot;i++) e[i].fl=0;
        while(bfs()) 
            res+=dfs(S,inf);
        return res;
    }
    void getcolor(int now,int c)
    {
        color[now]=c;
        for(int i=head[now];i;i=e[i].nxt)
        {
            int to=e[i].to;
            if(color[to]!=c && e[i].cap>e[i].fl)
                getcolor(to,c); 
        }
    }
};

namespace GHT
{
    int head[maxn],tot=0,colnum=0,tmp[maxn];
    edge e[maxn<<1];
    int fa[maxn][10],mn[maxn][10],dep[maxn];
    void add(int x,int y,int z)
    {
        e[++tot].to=y; e[tot].nxt=head[x]; e[tot].cap=z; head[x]=tot;
        e[++tot].to=x; e[tot].nxt=head[y]; e[tot].cap=z; head[y]=tot;
    }
    void build(int l,int r)
    {
        if(l==r) return;
        Dinic::S=node[l];
        Dinic::T=node[l+1];
        int mincut=Dinic::dinic();
        add(node[l],node[l+1],mincut);
        Dinic::getcolor(node[l],++colnum);
        int L=l,R=r;
        for(int i=l;i<=r;i++)
            if(color[node[i]]==colnum) tmp[L++]=node[i];
            else tmp[R--]=node[i];
        for(int i=l;i<=r;i++) node[i]=tmp[i];
        build(l,L-1); build(R+1,r);
    }
    void dfs(int u)
    {
        for(int i=1;i<=9;i++)
        {
            fa[u][i]=fa[fa[u][i-1]][i-1];
            mn[u][i]=min(mn[u][i-1],mn[fa[u][i-1]][i-1]);
        }
        for(int i=head[u];i;i=e[i].nxt)
        {
            int to=e[i].to;
            if(fa[u][0]!=to)
            {
                fa[to][0]=u; dep[to]=dep[u]+1;
                mn[to][0]=e[i].cap;
                dfs(to);
            }
        }
    }
    int getcut(int x,int y)
    {
        int res=inf;
        if(dep[x]<dep[y]) swap(x,y);
        for(int i=9;i>=0;i--)
            if(dep[fa[x][i]]>=dep[y])
                res=min(res,mn[x][i]),x=fa[x][i];
        if(x==y) return res;
        for(int i=9;i>=0;i--)
            if(fa[x][i]!=fa[y][i])
            {
                res=min(res,mn[x][i]);
                res=min(res,mn[y][i]);
                x=fa[x][i]; y=fa[y][i];
            }
        res=min(res,mn[x][0]); res=min(res,mn[y][0]);
        return res;
    }
}
void init()
{
    GHT::tot=0; GHT::colnum=0;
    memset(GHT::head,0,sizeof(GHT::head));
    Dinic::tot=1;
    memset(Dinic::head,0,sizeof(Dinic::head));
    for(int i=0;i<=maxn;i++)
    {
        color[i]=0;
        for(int j=0;j<=9;j++)
            GHT::mn[i][j]=0,GHT::fa[i][j]=0;
    }
}
int main()
{
    // freopen("a.in","r",stdin);
    // freopen("a.out","w",stdout);
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        init();
        int x,y,z;
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d%d",&x,&y,&z);
            Dinic::add(x,y,z);
        }
        for(int i=1;i<=n;i++) node[i]=i;
        GHT::build(1,n);
        GHT::dep[1]=1;
        GHT::dfs(1);
        int res=inf;
        for(int i=1;i<=n;i++)
            for(int j=i+1;j<=n;j++)
                    ans[i][j]=GHT::getcut(i,j);
        scanf("%d",&q);
        for(int i=1;i<=q;i++)
        {
            scanf("%d",&x);
            int res=0;
            for(int j=1;j<=n;j++)
                for(int k=j+1;k<=n;k++)
                    if(ans[j][k]<=x)
                        res++;
            printf("%d\n",res);
        }
        printf("\n");
    }
    return 0;
}