洛谷3329 [ZJOI2011]最小割

Luogu

分析

最小割树模板题。

建出最小割树,查询直接暴力枚举点对就好了。

可以在建完最小割树后暴力算出所有点对的最小割。

代码

// ===================================
//   author: M_sea
//   website: http://m-sea-blog.com/
// ===================================
#include <algorithm>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <queue>
#define re register
using namespace std;

inline int read() {
    int X=0,w=1; char c=getchar();
    while (c<'0'||c>'9') { if (c=='-') w=-1; c=getchar(); }
    while (c>='0'&&c<='9') X=X*10+c-'0',c=getchar();
    return X*w;
}

const int N=150+10,M=3000+10;

int n,m,Q,cut[N][N];

namespace GHT {
    struct edge { int v,w,nxt; } e[N<<1];
    int head[N],cnt=0;
    inline void addEdge(int u,int v,int w) {
        e[++cnt]=(edge){v,w,head[u]},head[u]=cnt;
    }

    inline void dfs(int s,int u,int fa,int k) {
        cut[s][u]=k;
        for (re int i=head[u];i;i=e[i].nxt)
            if (e[i].v!=fa) dfs(s,e[i].v,u,min(k,e[i].w));
    }
}

namespace Flow {
    struct edge { int v,w,nxt; } e[M<<1];
    int head[N],ori[M<<1],e_cnt=1;
    inline void addEdge(int u,int v,int w) {
        e[++e_cnt]=(edge){v,w,head[u]},head[u]=e_cnt,ori[e_cnt]=w;
        e[++e_cnt]=(edge){u,w,head[v]},head[v]=e_cnt,ori[e_cnt]=w;
    }

    namespace dinic {
        int s,t,level[N];
        inline int bfs() {
            memset(level,0,sizeof(level)),level[s]=1;
            queue<int> Q; Q.push(s);
            while (!Q.empty()) {
                int u=Q.front(); Q.pop();
                for (re int i=head[u];i;i=e[i].nxt) {
                    int v=e[i].v,w=e[i].w;
                    if (w&&!level[v]) level[v]=level[u]+1,Q.push(v);
                }
            }
            return level[t]!=0;
        }
        inline int dfs(int u,int cpflow) {
            if (u==t||!cpflow) return cpflow;
            int addflow=0;
            for (re int i=head[u];i;i=e[i].nxt) {
                int v=e[i].v,w=e[i].w;
                if (w&&level[v]==level[u]+1) {
                    int tmpadd=dfs(v,min(cpflow,w));
                    e[i].w-=tmpadd,e[i^1].w+=tmpadd;
                    cpflow-=tmpadd,addflow+=tmpadd;
                    if (!cpflow) break;
                }
            }
            if (!addflow) level[u]=0;
            return addflow;
        }
        inline int dinic(int s_,int t_) {
            for (re int i=2;i<=e_cnt;++i) e[i].w=ori[i];
            s=s_,t=t_; int res=0;
            while (bfs()) res+=dfs(s,2e9);
            return res;
        }
    }

    int id[N],tmp[N],vis[N];
    inline void dfs(int u) {
        vis[u]=1;
        for (re int i=head[u];i;i=e[i].nxt)
            if (e[i].w&&!vis[e[i].v]) dfs(e[i].v);
    }
    inline void build(int l,int r) {
        if (l>=r) return;
        int x=id[l],y=id[l+1],cut=dinic::dinic(x,y);
        memset(vis,0,sizeof(vis)),dfs(x);
        int p=l,q=r;
        for (re int i=l;i<=r;++i)
            vis[id[i]]?tmp[p++]=id[i]:tmp[q--]=id[i];
        for (re int i=l;i<=r;++i) id[i]=tmp[i];
        GHT::addEdge(x,y,cut),GHT::addEdge(y,x,cut);
        build(l,p-1),build(q+1,r);
    }
}

inline void init() {
    memset(Flow::head,0,sizeof(Flow::head)),Flow::e_cnt=1;
    memset(GHT::head,0,sizeof(GHT::head)),GHT::cnt=0;
}

int main() {
    int T=read();
    while (T--) {
        init();
        n=read(),m=read();
        for (re int i=1;i<=m;++i) {
            int u=read(),v=read(),w=read();
            Flow::addEdge(u,v,w);
        }
        for (re int i=1;i<=n;++i) Flow::id[i]=i;
        Flow::build(1,n);
        for (re int i=1;i<=n;++i) GHT::dfs(i,i,0,2e9);
        Q=read();
        while (Q--) {
            int x=read(),res=0;
            for (re int i=1;i<n;++i)
                for (re int j=i+1;j<=n;++j)
                    if (cut[i][j]<=x) ++res;
            printf("%d\n",res);
        }
        if (T) puts("");
    }
    return 0;
}
最后修改:2019 年 09 月 27 日 01 : 32 PM

发表评论