bzoj 4108: [Wf2015]Catering|带上下界最小费用可行流

终于搞懂 上下界网络流了!

注意边的数量

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int MAXN=1200;
const int INF=1000000000;
int tot=1,g[MAXN*10],num[MAXN*20],nnext[MAXN*20],flow[MAXN*20],cost[MAXN*20];
int n,k,s,t,ss,tt,s0,ans;
int d[MAXN],team[MAXN*40],head,tail;
int pre[MAXN],id[MAXN];
bool b[MAXN];
void add(int x,int y,int f,int c)
{
	tot++;nnext[tot]=g[x];g[x]=tot;num[tot]=y;cost[tot]=c;flow[tot]=f;
}
bool spfa()
{
	memset(d,100,sizeof(d));
	head=tail=0;
	d[ss]=0;team[++tail]=ss;
	while(head<tail)
	{
		int x=team[++head];
		for(int i=g[x];i;i=nnext[i])
		if(flow[i]>0)
		{
			int tmp=num[i];
			if(d[tmp]>d[x]+cost[i])
			{
				d[tmp]=d[x]+cost[i];
				pre[tmp]=x;
				id[tmp]=i;
				if(!b[tmp]) team[++tail]=tmp;
			}
		}
	}
	if(d[tt]==d[0]) return false;
	return true;
}
int main()
{
	scanf("%d %d",&n,&k);
	s=n*2+1,t=s+1;ss=t+1,tt=ss+1,s0=tt+1;
	add(s0,s,k,0);add(s,s0,0,0);
	add(t,s0,INF,0);add(s0,t,0,0);
	for(int i=1;i<=n;i++) {int x;scanf("%d",&x);add(s,i,1,x);add(i,s,0,-x);}
	for(int i=1;i<n;i++)
		for(int j=i+1;j<=n;j++)
		{
			int x;
			scanf("%d",&x);
			add(i+n,j,1,x);add(j,i+n,0,-x);
		}
	for(int i=1;i<=n;i++) add(i+n,t,1,0),add(t,i+n,0,0);
	for(int i=1;i<=n;i++) add(ss,i+n,1,0),add(i+n,ss,0,0);
	for(int i=1;i<=n;i++) add(i,tt,1,0),add(tt,i,0,0);
	while(spfa())
	{
		int tmp=tt;//cout<<tt<<' '<<ss<<endl;
		while(tmp!=ss)
		{
			ans+=cost[id[tmp]];
			flow[id[tmp]]-=1;
			flow[id[tmp]^1]+=1;
			tmp=pre[tmp];
		}
	}
	cout<<ans<<endl;
	return 0; 
}