餐巾计划问题——费用流spfa

　　这题就是spfa随便打打就好。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
inline int read(){
    int t=1,num=0;
    char c=getchar();
    while(c>'9'||c<'0'){if(c=='-')t=-1;c=getchar();}
    while(c>='0'&&c<='9'){num=num*10+c-'0';c=getchar();}
    return num*t;
}
struct edge{
    int to,cap,cost,next;
}g[10010];
int h[2010],d[2010],used[2010],que[100010],last[2010];
int k=1,INF=0x7fffffff,ans=0;
void add(int f,int t,int c1,int c2){
    g[++k].next=h[f];h[f]=k;g[k].to=t;g[k].cap=c1;g[k].cost=c2;
    g[++k].next=h[t];h[t]=k;g[k].to=f;g[k].cap=0;g[k].cost=-c2;
}
bool spfa(int s,int t){
    memset(last,0,sizeof(last));
    memset(d,127/3,sizeof(d));INF=d[0];
    memset(used,0,sizeof(used));
    int tail,head;
    head=tail=50002;
    que[head]=s;used[s]=1;d[s]=0;
    while(head>=tail){
        int x=que[tail++];
        for(int i=h[x];i;i=g[i].next){
            if(g[i].cap&&d[x]+g[i].cost<d[g[i].to]){
                d[g[i].to]=d[x]+g[i].cost;
                last[g[i].to]=i;
                if(!used[g[i].to]){
                    if(d[g[i].to]<d[que[tail]])que[--tail]=g[i].to;
                    else que[++head]=g[i].to;
                    used[g[i].to]=1;
                }
            }
        }
        used[x]=0;
    }
    return d[t]!=INF;
}
void mcf(int t){
    int minn=INF;
    for(int i=last[t];i;i=last[g[i^1].to])minn=min(minn,g[i].cap);
    for(int i=last[t];i;i=last[g[i^1].to]){
        ans+=g[i].cost*minn;
        g[i].cap-=minn;
        g[i^1].cap+=minn;
    }
}
int main()
{
    int n,x,y,p,f,s;
    n=read();p=read();x=read();f=read();y=read();s=read();
    for(int i=1;i<=n;i++){
        int now=read();
        add(n*2+1,i,INF,p);
        add(n*2+1,i+n,now,0);
        add(i,n*2+2,now,0);
        if(i+1<=n)add(i+n,i+n+1,INF,0);
        if(i+x<=n)add(i+n,i+x,INF,f);
        if(i+y<=n)add(i+n,i+y,INF,s);
    }
    while(spfa(n*2+1,n*2+2))mcf(n*2+2);
    printf("%d",ans);
    return 0;
}

本文由Yzyet编写，网址为www.cnblogs.com/Yzyet。非Yzyet同意，禁止转载，侵权者必究。