bzoj2324: [ZJOI2011]营救皮卡丘 费用流

先求最短路 然后把每个点拆点。分为入点和出点,由于基地有先后顺序所以入点和出点不能直接连边。

我们得在出点往入点连边。然后每个超级源向每个入点连流量为1的边，出点向每个汇连流量为1的边。

0和n+1~2*n是入点。1到n是出点

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
#include <cmath>
#include <queue>
using namespace std;
#define maxn 444444
#define INF 0x3f3f3f3f
struct node
{
    int u,v,f,c,next;
}e[maxn];
int a[320][320],head[maxn],dist[maxn],vis[maxn],pre[maxn];
int n,m,p,en,st,ed,maxflow,mincost;
void add(int u,int v,int f,int c)
{
    e[en].u=u;
    e[en].v=v;
    e[en].c=c;
    e[en].f=f;
    e[en].next=head[u];
 
    head[u]=en++;
    e[en].u=v;
    e[en].v=u;
    e[en].c=-c;
    e[en].f=0;
    e[en].next=head[v];
    head[v]=en++;
}
int spfa()
{
    int u,v;
    for(int i=0;i<=ed;i++)
        pre[i]=-1,vis[i]=0,dist[i]=INF;
    dist[st]=0;
    vis[st]=1;
    queue<int>q;
    q.push(st);
    while(!q.empty())
    {
        u=q.front();
        q.pop();
        for(int i=head[u];i!=-1;i=e[i].next)
        {
            v=e[i].v;
            if(e[i].f>0&&dist[u]+e[i].c<dist[v])
            {
                dist[v]=dist[u]+e[i].c;
                pre[v]=i;
                if(!vis[v])
                {
                    vis[v]=1;
                    q.push(v);
                }
            }
        }
        vis[u]=0;
    }
    if(dist[ed]==INF)
        return 0;
    return 1;
}
void add2()
{
    int v;
    int maxf=INF;
    for(v=pre[ed];~v;v=pre[e[v].u])
        maxf=min(maxf,e[v].f);
    for(v=pre[ed];~v;v=pre[e[v].u])
    {
        e[v].f-=maxf;
        e[v^1].f+=maxf;
 
    }
    maxflow+=maxf;
    mincost+=maxf*dist[ed];
}
void floyd()
{
    for(int i=1;i<=n;i++) a[i][i]=0;
    for(int k=0;k<=n;k++)
    {
        for(int i=0;i<=n;i++)
        {
            for(int j=0;j<=n;j++)
            {
                if (i == j) continue;
                a[i][j]=min(a[i][k]+a[k][j],a[i][j]);
                if (k==j&&i<j&&a[i][j] < INF)
                add(i==0?0:i+n,j,INF,a[i][j]);
            }
        }
    }
}
void build()
{
    add(st,0,p,0);
    for(int i = 1; i <= n; i++)
    {
        add(i,ed,1,0);
        add(st,i+n,1,0);
    }
}
void init()
{
    maxflow=0;
    st=n*2+1;
    ed=st+1;
    memset(head,-1,sizeof(head));
}
int main()
{
    memset(a,0x3f3f3f3f,sizeof(a));
    scanf("%d%d%d",&n,&m,&p);
    for(int i=1;i<=m;i++)
    {
        int u,v,w;
        scanf("%d%d%d",&v,&u,&w);
        a[u][v]=a[v][u]=min(a[u][v],w);
    }
    init();
    floyd();
 
    build();
    while(spfa())
    {
        add2();
    }
    printf("%d\n",mincost);
    return 0;
}