正题
题目链接:https://www.luogu.com.cn/problem/CF802C
题目大意
你有一个可以放kkk本书的书架,第iii天要求书架上有第aia_iai种书,购买第iii种书的价格为cic_ici。
求满足nnn天要求的最小花费。
1≤n,k≤80,1≤ci≤1061\leq n,k\leq 80,1\leq c_i\leq 10^61≤n,k≤80,1≤ci≤106
解题思路
我们注意到每一天的限制是不能超过kkk本书延续到下一天。而书的保留我们可以考虑视为将一本书从这一天延续到下一次需要这本书的时候免费拿一本新的。
好那么现在相当于我们可以通过霸占一些区间来减少花费,但是一个格子最多被区间包含kkk次。
这和[NOI2008] 志愿者招募很像,考虑使用类似的建图。
首先我们要固定总流量为nnn,然后考虑怎么建图。这里将一个点拆为三部分记为p(x,1/2/3)p(x,1/2/3)p(x,1/2/3)
- 获取点,用来流过今天需要的书。
- 储存点,表示流过的这本书已经使用完了。
- 商店点,流过多余的流量,购买书时从这个点流过。
然后要实现以下功能
- 要求满足每天的条件,我们让p(x,1)→p(x,2)p(x,1)\rightarrow p(x,2)p(x,1)→p(x,2),流量为111,费用为−∞-\infty−∞:表示每天的条件必须满足。
- 延续这本书到下一个同样需求的点yyy,p(x,2)→p(y,1)p(x,2)\rightarrow p(y,1)p(x,2)→p(y,1),流量为111,费用为000:表示这本书用完之后可以留给后面的用。
- 购买一本书,p(x,3)→p(x,1)p(x,3)\rightarrow p(x,1)p(x,3)→p(x,1),流量为111,费用为caxc_{a_x}cax。
- 丢弃一本书,p(x,2)→p(x+1,3)p(x,2)\rightarrow p(x+1,3)p(x,2)→p(x+1,3),流量为111,费用为000。因为这本书今天已经借出了所以需要占用空间,明天才能丢掉。
- 限制存有书的数量不超过kkk:考虑到我们相当于总共有nnn点流量,在1/21/21/2类点之间的流量都相当于目前存在书柜里的,所以商店的书至少需要为n−kn-kn−k本。那么我们让p(x,3)→p(x+1,3)p(x,3)\rightarrow p(x+1,3)p(x,3)→p(x+1,3)流量为n−kn-kn−k,费用为−∞-\infty−∞,这样这些流量就必须流过了,然后再建立一条p(x,3)→p(x+1,3)p(x,3)\rightarrow p(x+1,3)p(x,3)→p(x+1,3)流量为∞\infty∞,费用为000的给多余的流过就好了。
费用为−∞-\infty−∞的我们设为一个很大的值,最后将答案把这些值加回去就好了。
code
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#define ll long long
using namespace std;
const ll N=1e4+10,inf=1e9;
struct node{ll to,next,w,c;
}a[N<<1];
ll n,k,tot,ans,op[N],c[N],las[N];
ll s,t,ls[N],f[N],pre[N],mf[N];
bool v[N];queue<ll> q;
void addl(ll x,ll y,ll w,ll c){a[++tot].to=y;a[tot].next=ls[x];ls[x]=tot;a[tot].w=w;a[tot].c=c;a[++tot].to=x;a[tot].next=ls[y];ls[y]=tot;a[tot].w=0;a[tot].c=-c;return;
}
bool SPFA(){memset(f,0x3f,sizeof(f));pre[t]=0;f[s]=0;q.push(s);v[s]=1;mf[s]=inf;while(!q.empty()){ll x=q.front();q.pop();v[x]=0;for(ll i=ls[x];i;i=a[i].next){ll y=a[i].to;if(a[i].w&&f[x]+a[i].c<f[y]){f[y]=f[x]+a[i].c;pre[y]=i;mf[y]=min(mf[x],a[i].w);if(!v[y])v[y]=1,q.push(y);}}}return pre[t];
}
ll update(){ll x=t;while(x!=s){a[pre[x]].w-=mf[t];a[pre[x]^1].w+=mf[t];x=a[pre[x]^1].to;}return mf[t]*f[t];
}
signed main()
{
// freopen("bibliotheca.in","r",stdin);
// freopen("bibliotheca.out","w",stdout);scanf("%lld%lld",&n,&k);tot=1;for(ll i=1;i<=n;i++)scanf("%lld",&op[i]);for(ll i=1;i<=n;i++)scanf("%lld",&c[i]);ll w=0;s=3*n+2;t=s+1;addl(s,2*n+1,n,0);for(ll i=1;i<=n;i++){if(las[op[i]])addl(las[op[i]]+n,i,1,0);else w=min(w+1,k);addl(i+2*n,i,1,c[op[i]]);addl(i,i+n,1,-inf);ans+=inf;addl(i+n,i+2*n+1,1,0);las[op[i]]=i;if(n>w){addl(i+2*n,i+2*n+1,n-w,-inf);ans+=(n-w)*inf;}addl(i+2*n,i+2*n+1,inf,0);}addl(3*n+1,t,n,0);int k=0;while(SPFA())ans+=update();printf("%lld\n",ans);return 0;
}