CF802C Heidi and Library (hard) 费用流

题目链接

题意：
你有一个容量为$k$的空书架，现在共有$n$个请求，每个请求给定一本书$a_i$，如果你的书架里没有这本书，你就必须以$c_i$的价格购买这本书放入书架。当然，你可以在任何时候丢掉书架里的某本书。请求出完成这$n$个请求所需要的最少价钱。$n,k<=80$

题解：
一道真的挺神仙的题，我还是想不出来。

是个费用流题，我听了两种做法，都是y_immortal神仙教我的，STO sro y_immortal orz OTZ。

第一种做法的思路是我们考虑先把每天的都买进来，然后对于放在书架上的，我们看作可以原价卖出去，这样就可以实现第二次购买的时候实际上并没有新花钱，也就是把多余的花费抵消掉。建图的方法是除了源汇之外建两排点，第一排表示每一天，第二排是一个辅助点的样子（不知道该怎么解释了）。我们从源点到每一天的连一条流量为$1$，费用为$c_{a_i}$的边，表示买这本书。然后$i$向$i^{\prime }$连一条流量为$1$，费用为$0$的边，表示买了之后把这本书扔了。我们从$i$到$i+1$连一条流量为$k-1$，费用为$0$的边，表示这一天之后书架上原来能继承过来最多$k-1$本书，这一天要的那一本一定要有。另外我们要搞那个退钱操作，于是我们从上一个买进这本书的那天的下一天的辅助点向这个点连一条流量为$1$，费用为$-c_{a_i}$的边。于是我们跑一个最小费用最大流就好了。

第二种做法建图方法不太一样，点数是$O(n)$的，更优秀一点。我们先把每天的都买进来，然后考虑书架最多能帮我们省多少。我们用总的答案减去最多能省的钱得出答案。我们相邻的两个点之间都连一条流量为$k-1$，费用是$0$的边。然后我们考虑省多少，我们用差不多的思路，但是不建第二排点，我们直接从当前这一天要买的书上一次出现的那一天的下一天，向当前天连流量为$1$，费用为$-c_{a_i}$的边，然后求最小费用最大流就可以了。

我的代码写的是第二种。

代码：

#include <bits/stdc++.h>
using namespace std;

int n,k,b[100],c[100],pre[100],hed[500010],cnt;
int dep[2010],st,ed,ans,w[110],v[110],f[110],inq[110];
struct node
{
	int to,next,from,c,cost;
}a[500010];
queue<int> q;
inline void add(int from,int to,int c,int cost)
{
	a[++cnt].to=to;
	a[cnt].from=from;
	a[cnt].c=c;
	a[cnt].cost=cost;
	a[cnt].next=hed[from];
	hed[from]=cnt;
	a[++cnt].to=from;
	a[cnt].from=to;
	a[cnt].c=0;
	a[cnt].cost=-cost;
	a[cnt].next=hed[to];
	hed[to]=cnt;
}
inline void bfs()
{
	memset(w,0,sizeof(w));
	memset(v,0x3f,sizeof(v));
	w[st]=2e9;
	v[st]=0;
	q.push(st);
	while(!q.empty())
	{
		int x=q.front();
		q.pop();
		inq[x]=0;
		for(int i=hed[x];i;i=a[i].next)
		{
			int y=a[i].to;
			if(v[y]>v[x]+a[i].cost&&a[i].c)
			{
				v[y]=v[x]+a[i].cost;
				w[y]=min(a[i].c,w[x]);
				f[y]=i;
				if(!inq[y])
				{
					q.push(y);
					inq[y]=1;
				}
			}
		}
	}
	for(int i=f[ed];i;i=f[a[i].from])
	{
		a[i].c-=w[ed];
		a[i^1].c+=w[ed];
	}
}
int main()
{
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;++i)
	scanf("%d",&b[i]);
	for(int i=1;i<=n;++i)
	scanf("%d",&c[i]);
	st=n+1;
	ed=st+1;
	cnt=1;
	for(int i=1;i<=n;++i)
	{
		ans+=c[b[i]];
		if(i==1)
		add(st,i,k-1,0);
		else
		add(i-1,i,k-1,0);	
		if(pre[b[i]])
		{
			if(pre[b[i]]+1==i)
			ans-=c[b[i]];
			else
			add(pre[b[i]]+1,i,1,-c[b[i]]);
		}	
		pre[b[i]]=i;
	}
	add(n,ed,k-1,0);
	while(1)
	{
		bfs();
		if(w[ed])
		ans+=v[ed]*w[ed];
		else
		break;
	}
	printf("%d\n",ans);
	return 0;
}