BJ模拟 Delight for a Cat【费用流】

题目描述

从前，有一只懒猫叫CJB。每个小时，这只猫要么在睡觉，要么在吃东西，但不能一边睡觉一边吃东西，并且这只猫会在一整个小时干同一件事情。
对于接下来的$n$个小时，CJB知道他在那n个小时睡觉和吃东西的快乐值。
为了健♂康♀地生活，在任意的连续$k$ 个整小时内，CJB要有至少$m_s$小时睡觉，至少 $m_e$ 个小时在吃东西。也就是说一共有$n−k+1$段的$k$ 小时需要满足上述条件。
你的任务是告诉CJB他接下来$n$个小时能获得的最大快乐值是多少。
$n<=1000$

解题思路：

和noi2008志愿者一样的做法。
首先，假设这只猫在所有天数都是睡觉，我们稍后再决策将哪些天数改为吃东西。
我们用 xi=0表示这只猫在第i天睡觉， xi=1表示吃东西。

则有下列不等式：

me≤x1+x2+x3+…+xk≤k−ms

me≤x2+x3+x4+…+xk+1≤k−ms

…

me≤xn−k+1+xn−k+2+…+xn≤k−ms

我们令 0≤y1≤k−ms−me

，则上述不等式可以转化为以下的等式：

x1+x2+x3+…+xk+y1=k−ms

x2+x3+x4+…+xk+1+y2=k−ms

…

xn−k+1+xn−k+2+…+xn+yn−k+1=k−ms

将这些等式两两相减，得：

x1−xk+1+y1−y2=0

x2−xk+2+y2−y3=0

…

xn−k−xn+yn−k−yn−k+1=0

(k−ms)−x1−x2−…−xk−y1=0

xn−k+1+xn−k+2+…+xn+yn−k+1−(k−ms)=0

我们把等式左边加起来，发现恰好为0，满足流量平衡。

把上述每个等式看成一个点，负的流出，正的流入（或反过来）。跑一次最小费用最大流即可。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int getint()
{
    int i=0,f=1;char c;
    for(c=getchar();(c!='-')&&(c<'0'||c>'9');c=getchar());
    if(c=='-')c=getchar(),f=-1;
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i*f;
}
const int N=1005,M=200005,inf=1e9;
const ll INF=1e17;
int n,k,ms,me,S,T,s[N],e[N];
int tot=1,first[N],nxt[M],to[M],cap[M],cost[M];
ll ans,dis[N];
bool vis[N],exist[N];
queue<int>q;
inline void add(int x,int y,int f,int w)
{
    nxt[++tot]=first[x],first[x]=tot,to[tot]=y,cap[tot]=f,cost[tot]=w;
    nxt[++tot]=first[y],first[y]=tot,to[tot]=x,cap[tot]=0,cost[tot]=-w; 
}
bool SPFA()
{
    for(int i=0;i<=T;i++)vis[i]=0,dis[i]=INF;
    dis[S]=0,q.push(S),exist[S]=1;
    while(!q.empty())
    {
        int u=q.front();q.pop();exist[u]=0;
        for(int i=first[u];i;i=nxt[i])
        {
            int v=to[i];
            if(dis[v]>dis[u]+cost[i]&&cap[i])
            {
                dis[v]=dis[u]+cost[i];
                if(!exist[v])q.push(v),exist[v]=1;
            }
        }
    }
    return dis[T]!=INF;
}
int dinic(int u,int flow)
{
    if(u==T)
    {
        ans-=dis[T]*flow;
        return flow;
    }
    int res=0;
    vis[u]=1;
    for(int i=first[u];i;i=nxt[i])
    {
        int v=to[i];
        if(dis[v]==dis[u]+cost[i]&&cap[i]&&!vis[v])
        {
            int det=dinic(v,min(flow-res,cap[i]));
            cap[i]-=det,cap[i^1]+=det;
            res+=det;if(res==flow)break;
        }
    }
    return res;
}
int main()
{
    //freopen("lx.in","r",stdin);
    n=getint(),k=getint(),ms=getint(),me=getint();
    for(int i=1;i<=n;i++)ans+=s[i]=getint();
    for(int i=1;i<=n;i++)e[i]=getint();
    S=n-k+2,T=n-k+3;
    add(S,0,k-ms,0),add(n-k+1,T,k-ms,0);
    add(0,1,k-ms-me,0),add(n-k,n-k+1,k-ms-me,0);
    if(k<=n-k)
    {
        for(int i=1;i<=k;i++)add(0,i,1,s[i]-e[i]);
        for(int i=k+1;i<=n-k;i++)add(i-k,i,1,s[i]-e[i]);
        for(int i=n-k+1;i<=n;i++)add(i-k,n-k+1,1,s[i]-e[i]);
    }
    else
    {
        for(int i=1;i<=n-k;i++)add(0,i,1,s[i]-e[i]);
        for(int i=n-k+1;i<=k;i++)add(0,n-k+1,1,s[i]-e[i]);
        for(int i=k+1;i<=n;i++)add(i-k,n-k+1,1,s[i]-e[i]);
    }
    for(int i=1;i<n-k;i++)add(i,i+1,k-ms-me,0);
    while(SPFA())dinic(S,inf);
    cout<<ans<<'\n';
    return 0;
}