BZOJ 1061 （费用流，填”坑“ ）

1061: [Noi2008]志愿者招募

Time Limit: 20 Sec   Memory Limit: 162 MB
Submit: 2874   Solved: 1775
[ Submit][ Status][ Discuss]

Description

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

Input

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

Output

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

Sample Input

3 3
2 3 4
1 2 2
2 3 5
3 3 2

Sample Output

14

建图：

把每一天作为一个点，从源点s给最左边的时间点引流，流的大小为U（U为大整数），然后每个时间点i向时间点i+1建边（i，i+1，U-Ai，0），最后设汇点为最右端的时间点。如果所有的时间点Ai都为0的话，流出的流量也为U.

Ai不为0，那么源点到汇点的流量被截断了，就要其他支路流量来填补。第i类人建边(li,ri+1,INF,ci)。

要求的是最小费用，跑最小费用流时，要求汇点流量必须为U，cost才是最小，否则无解。

#include<iostream>
#include<string>
#include<stdio.h>
#include<string.h>
#include<vector>
#include<math.h>
#include<queue>
#include<map>
#include<set>
#include<algorithm>
using namespace std;
#define MAXN 1205
#define LL long long
#define INF 0x3f7f7f7f
const double eps = 1e-10;
struct Edge
{
    int from,to,cap,flow,cost;
    Edge(int from=0,int to=0,int cap=0,int flow=0,int cost=0)
    {
        this->from=from;
        this->to=to;
        this->cap=cap;
        this->flow=flow;
        this->cost=cost;
    }
};
vector<Edge>edges;
vector<int>g[MAXN];
bool vis[MAXN];
int dis[MAXN];
int pre[MAXN];
int a[MAXN];
int s,t;
void init()
{
    for(int i=0;i<MAXN;i++)
    {
        g[i].clear();
    }
    edges.clear();
}
void add(int from,int to,int cap,int cost)
{
    edges.push_back(Edge(from,to,cap,0,cost));
    edges.push_back(Edge(to,from,0,0,-cost));
    int sz=edges.size();
    g[from].push_back(sz-2);
    g[to].push_back(sz-1);
}
bool spfa(int &flow,int &cost)
{
    int i;
    for(i=0;i<MAXN;i++)
        dis[i]=INF;
    memset(vis,0,sizeof(vis));
    queue<int>q;
    q.push(s);
    dis[s]=0;
    vis[s]=1;
    pre[s]=0;
    a[s]=INF;
   while(!q.empty())
   {
       int u=q.front();
       q.pop();
       vis[u]=0;
       int sz=g[u].size();
       for(i=0;i<sz;i++)
       {
           Edge &e=edges[g[u][i]];
           if(e.cap>e.flow&&dis[e.to]>dis[u]+e.cost)
           {
               dis[e.to]=dis[u]+e.cost;
               pre[e.to]=g[u][i];
               a[e.to]=min(a[u],e.cap-e.flow);
               if(!vis[e.to])
               {
                   vis[e.to]=1;
                   q.push(e.to);
               }
           }
       }
   }
   if(dis[t]==INF)
    return false;
   flow+=a[t];
   cost+=dis[t]*a[t];
   int v=t;
   while(v!=s)
   {
       edges[pre[v]].flow+=a[t];
       edges[pre[v]^1].flow-=a[t];
       v=edges[pre[v]].from;
   }
   return true;
}
#define U 100000
int minCost()
{
    int flow=0,cost=0;
    while(spfa(flow,cost));
    if(flow!=U)
        cost=-1;
    return cost;
}
int n,m;

int main()
{
    int i;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        init();
        s=0,t=1000+4;
        add(s,1,U,0);
        for(i=1;i<=n;i++)
        {
            int id;
            scanf("%d",&id);
            add(i,i+1,U-id,0);
        }
        for(i=1;i<=m;i++)
        {
            int l,r,c;
            scanf("%d%d%d",&l,&r,&c);
            add(l,r+1,INF,c);
        }
        add(n+1,t,U,0);
        printf("%d\n",minCost());

    }
    return 0;
}