洛谷 P3980 [NOI2008]志愿者招募 费用流

题目描述

申奥成功后，布布经过不懈努力，终于成为奥组委下属公司人力资源部门的主管。布布刚上任就遇到了一个难题：为即将启动的奥运新项目招募一批短期志愿者。经过估算，这个项目需要N 天才能完成，其中第i 天至少需要Ai 个人。 布布通过了解得知，一共有M 类志愿者可以招募。其中第i 类可以从第Si 天工作到第Ti 天，招募费用是每人Ci 元。新官上任三把火，为了出色地完成自己的工作，布布希望用尽量少的费用招募足够的志愿者，但这并不是他的特长！于是布布找到了你，希望你帮他设计一种最优的招募方案。

输入输出格式

输入格式：
第一行包含两个整数N, M，表示完成项目的天数和可以招募的志愿者的种类。 接下来的一行中包含N 个非负整数，表示每天至少需要的志愿者人数。 接下来的M 行中每行包含三个整数Si, Ti, Ci，含义如上文所述。为了方便起见，我们可以认为每类志愿者的数量都是无限多的。

输出格式：
仅包含一个整数，表示你所设计的最优方案的总费用。

输入输出样例

输入样例#1：
3 3
2 3 4
1 2 2
2 3 5
3 3 2
输出样例#1：
14
说明

1 ≤ N ≤ 1000，1 ≤ M ≤ 10000，题目中其他所涉及的数据均 不超过2^31-1。

分析：
想到网络流以后，一个直观的想法就是源点连人，人连可以被雇佣的日子，日子连汇点，然后跑最小费用最大流。可以发现一滴流量（雇佣一个人）只能用其中一天，其实每天都是能用的。

那我们可以把点串联或直接跨过这个区间，显然串联的话不同类的人会相互影响，那就只能直接跨过这个区间了。

我们设每天都要雇佣$k$个志愿者，对于一类志愿者，可以管$[l,r]$，那么从$l$到$r+1$连一条流量为$k$，费用为$c_i$的边，流一滴流量相当于雇佣一个志愿者。第$i$天到第$i+1$天连一条边，流量为$k-a_i$，费用为$0$，表示第$i$天需要$a_i$个志愿者。因为假设每天要$k$个志愿者，第$i$天实际只要$a_i$个志愿者，相当于雇佣了$k-a_i$个免费的只在第$i$天的志愿者。

我们可以把$k$设为$inf$，只保证费用最小，但是费用流会为了更大的流而花费更多的钱，但是每一滴从源点到汇点的流都只表示$[1,n]$都雇佣了一个志愿者，所以我们希望的最大流是$inf$，所以从源点向$1$点流量为$inf$，费用为$0$的边，$n+1$到汇点连流量为$inf$，费用为 0 0 <script type="math/tex" id="MathJax-Element-4935">0</script>的边，问题就解决了。

代码：

#include <iostream>
#include <cstdio>
#include <cmath>
#include <queue>
#define LL long long

const int inf=2147483647;
const int maxn=1007;
const int maxe=40007;

using namespace std;

int n,m,x,y,c,s,t,cnt;
LL ans;
int pre[maxn],ls[maxn],dis[maxn],v[maxn];

struct edge{
    int x,y,w,c,op,next;
}g[maxe];

queue <int> q;

void add(int x,int y,int w,int c)
{
    g[++cnt]=(edge){x,y,w,c,cnt+1,ls[x]};
    ls[x]=cnt;
    g[++cnt]=(edge){y,x,0,-c,cnt-1,ls[y]};
    ls[y]=cnt;
}

bool spfa() 
{
    for (int i=s;i<=t;i++)
    {
        dis[i]=inf/2;
        v[i]=0;
    }
    while (!q.empty()) q.pop();
    dis[s]=0;
    v[s]=1;
    q.push(s);
    while (!q.empty())
    {
        int x=q.front();
        q.pop();
        for (int i=ls[x];i>0;i=g[i].next)
        {
            int y=g[i].y;
            if ((g[i].w) && (dis[x]+g[i].c<dis[y]))
            {
                dis[y]=dis[x]+g[i].c;
                pre[y]=i;
                if (!v[y])
                {
                    v[y]=1;
                    q.push(y);
                }
            }
        }
        v[x]=0;
    }
    if (dis[t]==inf/2) return 0;
    return 1;
}

void mcf()
{
    int flow=inf,i=t;
    while (i!=s)
    {
        flow=min(flow,g[pre[i]].w);
        i=g[pre[i]].x;
    }
    i=t;
    while (i!=s)
    {
        g[pre[i]].w-=flow;
        g[g[pre[i]].op].w+=flow;
        ans+=(LL)g[pre[i]].c*1ll*flow;
        i=g[pre[i]].x;
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    s=0; t=n+2;
    add(s,1,inf,0);
    add(n+1,t,inf,0);
    for (int i=1;i<=n;i++)
    {
        scanf("%d",&x);
        add(i,i+1,inf-x,0);
    }
    for (int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&x,&y,&c);
        add(x,y+1,inf,c);
    }   
    while (spfa()) 
      mcf();
    printf("%lld",ans);
}