BZOJ4519: [Cqoi2016]不同的最小割

清空时没清空好

最小割树=分治+最小割
在当前集合随便找两个点,算他们之间的最小割
然后找当前最小割分割的两个集合(包含不在当前分治集合的点),就从起点遍历,每次只走未满流的边即可
然后更新这两个点集之间的最小割(包含不在当前分治集合的点,这些点之间的最小割也要更新),然后把当前集合按照最小割分成两个集合,分治下去

code:

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#define ll long long
#define inf 1e15
using namespace std;

const int maxn = 900;
const int maxm = 110000;

int n,m;
int E[maxm][2];
struct edge
{
    int y,nex;
    ll c;
    edge(){}
    edge(const int _y,const ll _c,const int _nex){y=_y;c=_c;nex=_nex;}
}a[maxm],e[maxm]; int len,fir[maxn];
inline void ins(const int x,const int y,const ll c)
{
    a[++len]=edge(y,c,fir[x]);fir[x]=len;
    a[++len]=edge(x,c,fir[y]);fir[y]=len;
}

int st,ed;
int h[maxn];
queue<int>q;
bool bfs()
{
    for(int i=1;i<=n;i++) h[i]=0;
    h[st]=1; q.push(st);
    while(!q.empty())
    {
        const int x=q.front(); q.pop();
        for(int k=fir[x];k;k=a[k].nex)
        {
            const int y=a[k].y;
            if(a[k].c&&!h[y]) h[y]=h[x]+1,q.push(y);
        }
    }
    return h[ed]!=0;
}
ll dfs(const int x,const ll flow)
{
    if(x==ed) return flow;
    ll delta=0;
    for(int k=fir[x];k;k=a[k].nex)
    {
        const int y=a[k].y;
        if(a[k].c&&h[y]==h[x]+1)
        {
            ll minc=dfs(y,min(a[k].c,flow-delta));
            a[k].c-=minc; a[k^1].c+=minc;
            delta+=minc;
        }
        if(delta==flow) return delta;
    }
    if(!delta) h[x]=0;
    return delta;
}
void clear()
{
    for(int i=1;i<=m;i++)
    {
        int ii=E[i][0],c=E[i][1]-a[ii].c;
        a[ii].c+=c,a[ii^1].c-=c;
    }
}
ll Flow()
{
    //clear();
    ll re=0;
    while(bfs()) re+=dfs(st,(ll)inf);
    return re;
}

mapint>reg;
int ans;
bool v[maxn];

void mark()
{
    for(int i=1;i<=n;i++) v[i]=false;
    q.push(st); v[st]=true;
    while(!q.empty())
    {
        const int x=q.front(); q.pop();
        for(int k=fir[x];k;k=a[k].nex)
        {
            const int y=a[k].y;
            if(a[k].c&&!v[y]) v[y]=true,q.push(y);
        }
    }
}
int p[maxn],temp[maxn];
void solve(int L,int R)
{
    if(L==R) return;
    st=p[L],ed=p[R];
    for(int i=2;i<=len;i+=2) a[i].c=a[i^1].c=(a[i].c+a[i^1].c)/2ll;
    ll tk=Flow();
    if(reg.count(tk)==0) reg[tk]=1,ans++;
    mark();
    int l=L,r=R;
    for(int i=L;i<=R;i++)
    {
        if(v[p[i]]) temp[l++]=p[i];
        else temp[r--]=p[i];
    }
    for(int i=L;i<=R;i++) p[i]=temp[i];
    solve(L,l-1); 
    solve(r+1,R);
}

int main()
{
    len=1; //
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
    {
        int x,y,c; scanf("%d%d%d",&x,&y,&c);
        E[i][0]=len+1;
        E[i][1]=c;
        ins(x,y,c);
    }
    for(int i=1;i<=n;i++) p[i]=i;
    solve(1,n);
    printf("%d\n",ans);

    return 0;
}

你可能感兴趣的:(BZOJ,最小割树)