bzoj千题计划139：bzoj2229: [Zjoi2011]最小割

#include<queue>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
 
using namespace std;
 
#define N 200
#define M 7000
const int inf=2e9;
 
int n;
 
int tot;
int front[N],nxt[M<<1],to[M<<1],val[M<<1],from[M<<1];
int lev[N],num[N];
int path[N];
int cur[N];
 
int src,decc;
 
int a[N],tmp[N];
bool use[N];
 
int dis[N][N];
 
void read(int &x)
{
    x=0; char c=getchar();
    while(!isdigit(c))  c=getchar(); 
    while(isdigit(c)) { x=x*10+c-‘0‘; c=getchar();  }
}
 
void add(int u,int v,int w)
{
    to[++tot]=v; nxt[tot]=front[u]; front[u]=tot; from[tot]=u; val[tot]=w;
    //to[++tot]=u; nxt[tot]=front[v]; front[v]=tot; from[tot]=v; val[tot]=0;
}
 
bool bfs()
{
    queue<int>q;
    for(int i=1;i<=n;++i) lev[i]=n;
    q.push(decc);
    lev[decc]=0;
    int now,t;
    while(!q.empty())
    {
        now=q.front();
        q.pop();
        for(int i=front[now];i;i=nxt[i])
        {
            t=to[i];
            if(lev[t]==n && val[i^1]) 
            {
                lev[t]=lev[now]+1;
                q.push(t);
            }
        }
    }
    return lev[src]!=n;
}
 
int augment()
{
    int now=decc,flow=inf;
    int i;
    while(now!=src)
    {
        i=path[now];
        flow=min(flow,val[i]);
        now=from[i];
    }
    now=decc;
    while(now!=src)
    {
        i=path[now];
        val[i]-=flow;
        val[i^1]+=flow;
        now=from[i];
    }
    return flow;
}
 
int isap()
{
    int flow=0;
    if(!bfs()) return 0;
    memset(num,0,sizeof(num));
    for(int i=1;i<=n;++i) num[lev[i]]++,cur[i]=front[i];
    int now=src,t;
    while(lev[src]<n)
    {
        if(now==decc)
        {
            flow+=augment();
            now=src;
        }
        bool advanced=false;
        for(int i=cur[now];i;i=nxt[i])
        {
            t=to[i];
            if(lev[t]==lev[now]-1 && val[i])
            {
                advanced=true;
                path[t]=i;
                cur[now]=i;
                now=t;
                break;
            }
        }
        if(!advanced)
        {
            int mi=n;
            for(int i=front[now];i;i=nxt[i])
                if(val[i]) mi=min(mi,lev[to[i]]);
            if(!--num[lev[now]]) break;
            num[lev[now]=mi+1]++;
            cur[now]=front[now];
            if(now!=src) now=from[path[now]];
        }
    }
    return flow;
}
 
void dfs(int x)
{
    use[x]=true;
    for(int i=front[x];i;i=nxt[i])
        if(!use[to[i]] && val[i]) dfs(to[i]);
}
 
void solve(int l,int r)
{
    if(l>=r) return;
    src=a[l]; decc=a[r];
    for(int i=2;i<=tot;i+=2) val[i]=val[i+1]=val[i]+val[i+1]>>1; 
    int flow=isap();
//    printf("%d %d %d\n",l,r,flow);
    memset(use,false,sizeof(use));
    dfs(src);
    for(int i=1;i<=n;++i)
        if(use[i])
            for(int j=1;j<=n;++j)
                if(!use[j])
                    dis[i][j]=dis[j][i]=min(dis[i][j],flow);
    int i=l,j=r;
    for(int k=l;k<=r;++k)
        if(use[a[k]]) tmp[i++]=a[k];
        else tmp[j--]=a[k];
    for(int k=l;k<=r;++k) a[k]=tmp[k];
    solve(l,i-1);
    solve(j+1,r);
}
 
int main()
{
//    freopen("mincuto.in","r",stdin);
//    freopen("mincuto.out","w",stdout);
    int T;
    read(T);
    int m,q;
    int u,v,w,x;
    int ans;
    while(T--)
    {
        tot=1;
        memset(front,0,sizeof(front));
        memset(dis,127,sizeof(dis));
        read(n); read(m);
        for(int i=1;i<=n;++i) a[i]=i;
        while(m--)
        {
            read(u); read(v); read(w);
            add(u,v,w);
            add(v,u,w);
        }
        solve(1,n);
        read(q);
        while(q--)
        {
            ans=0;
            read(x);
            for(int i=1;i<=n;++i)
                for(int j=i+1;j<=n;++j)
                    if(dis[i][j]<=x) ans++;
            cout<<ans<<‘\n‘;
        }
        cout<<‘\n‘;
    }
    return 0;
}