BZOJ 2245: [SDOI2011]工作安排

2245: [SDOI2011]工作安排

Time Limit: 20 Sec   Memory Limit: 512 MB
Submit: 1177   Solved: 567
[ Submit][ Status][ Discuss]

Description

你的公司接到了一批订单。订单要求你的公司提供n类产品，产品被编号为1~n，其中第i类产品共需要C_i件。公司共有m名员工，员工被编号为1~m员工能够制造的产品种类有所区别。一件产品必须完整地由一名员工制造，不可以由某名员工制造一部分配件后，再转交给另外一名员工继续进行制造。

我们用一个由0和1组成的m*n的矩阵A来描述每名员工能够制造哪些产品。矩阵的行和列分别被编号为1~m和1~n，A_i,j为1表示员工i能够制造产品j，为0表示员工i不能制造产品j。

如果公司分配了过多工作给一名员工，这名员工会变得不高兴。我们用愤怒值来描述某名员工的心情状态。愤怒值越高，表示这名员工心情越不爽，愤怒值越低，表示这名员工心情越愉快。员工的愤怒值与他被安排制造的产品数量存在某函数关系，鉴于员工们的承受能力不同，不同员工之间的函数关系也是有所区别的。

对于员工i，他的愤怒值与产品数量之间的函数是一个S_i+1段的分段函数。当他制造第1~T_i,1件产品时，每件产品会使他的愤怒值增加W_i,1，当他制造第T_i,1+1~T_i,2件产品时，每件产品会使他的愤怒值增加W_i,2……为描述方便，设T_i,0=0,T_i,si+1=+∞，那么当他制造第T_i,j-1+1~T_i,j件产品时，每件产品会使他的愤怒值增加W_i,j， 1≤j≤S_i+1。

你的任务是制定出一个产品的分配方案，使得订单条件被满足，并且所有员工的愤怒值之和最小。由于我们并不想使用Special Judge，也为了使选手有更多的时间研究其他两道题目，你只需要输出最小的愤怒值之和就可以了。

Input

第一行包含两个正整数m和n，分别表示员工数量和产品的种类数；

第二行包含n 个正整数，第i个正整数为C_i；

以下m行每行n 个整数描述矩阵A；

下面m个部分，第i部分描述员工i的愤怒值与产品数量的函数关系。每一部分由三行组成：第一行为一个非负整数S_i，第二行包含S_i个正整数，其中第j个正整数为T_i,j，如果S_i=0那么输入将不会留空行（即这一部分只由两行组成）。第三行包含S_i+1个正整数，其中第j个正整数为W_i,j。

Output

仅输出一个整数，表示最小的愤怒值之和。

Sample Input

2 3

2 2 2

1 1 0

0 0 1

1

2

1 10

1

2

1 6

Sample Output

24

HINT

解题思路：

裸的费用流。

源点向每件物品连边，容量为C[i], 费用为0。

每个人向汇点连S[i]+1条边，每条边的容量为T[i][j] - T[i][j-1], 费用为W[i][j]。

物品和员工之间直接按所给的矩阵连容量为无穷大的边。

结果要用long long 。

#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#define LL long long 
using namespace std;
const int MAXN = 1000 + 10;
const int MAXM = 500000 + 10;
const int INF = 0x3f3f3f3f;
int read()
{
    int x = 0, f = 1; char ch = getchar();
    while(ch < '0' || ch > '9') {if(ch == '-') f *= -1; ch = getchar();}
    while(ch >= '0' && ch <= '9') {x = x * 10 + ch - '0'; ch = getchar();}
    return x * f;
}
struct Edge
{
    int to, next, cap, flow, cost;
}edge[MAXM];
int head[MAXN], tot, pre[MAXN], dis[MAXN];
bool vis[MAXN];
int N;
void init(int n)
{
    N = n; tot = 0;
    memset(head, -1, sizeof(head));
}
void addedge(int u, int v, int cap, int cost)
{
    edge[tot].to = v; edge[tot].cap = cap;
    edge[tot].cost = cost; edge[tot].flow = 0;
    edge[tot].next = head[u]; head[u] = tot++;
    edge[tot].to = u; edge[tot].cap = 0;
    edge[tot].cost = -cost; edge[tot].flow = 0;
    edge[tot].next = head[v]; head[v] = tot++;
}
bool spfa(int s, int t)
{
    queue<int>q;
    for(int i=0;i<N;i++)
    {
        dis[i] = INF, vis[i] = false, pre[i] = -1;
    }
    dis[s] = 0, vis[s] = true, q.push(s);
    while(!q.empty())
    {
        int u = q.front();q.pop();
        vis[u] = false;
        for(int i=head[u];i!=-1;i=edge[i].next)
        {
            int v = edge[i].to;
            if(edge[i].cap > edge[i].flow && dis[v] > dis[u] + edge[i].cost)
            {
                dis[v] = dis[u] + edge[i].cost;
                pre[v] = i;
                if(!vis[v]){vis[v] = true; q.push(v);}
            }
        }
    }
    if(pre[t] == -1) return false;
    else return true;
}
int minCostMaxflow(int s, int t, LL& cost)
{
    int flow = 0;cost = 0;
    while(spfa(s, t))
    {
        int Min = INF;
        for(int i=pre[t];i!=-1;i=pre[edge[i^1].to])
        {
            if(Min > edge[i].cap - edge[i].flow)
                Min = edge[i].cap - edge[i].flow;
        }
        for(int i=pre[t];i!=-1;i=pre[edge[i^1].to])
        {
            edge[i].flow += Min;
            edge[i^1].flow -= Min;
            cost += edge[i].cost * Min;
        }
        flow += Min;
    }
    return flow;
}
int m, n, C[MAXN], A[300][300];
int S[MAXN], T[300][300], W[300][300];
int main()
{
    m = read(), n = read();
    for(int i=1;i<=n;i++) C[i] = read();
    for(int i=1;i<=m;i++) for(int j=1;j<=n;j++) A[i][j] = read();
    for(int i=1;i<=m;i++)
    {
        S[i] = read();
        for(int j=1;j<=S[i];j++) T[i][j] = read();
        for(int j=1;j<=S[i]+1;j++) W[i][j] = read();
        T[i][0] = 0, T[i][S[i]+1] = INF;
    }
    init(n + m + 2); int start = 0, end = n + m + 1;
    for(int i=1;i<=n;i++) addedge(start, i, C[i], 0);
    for(int i=1;i<=m;i++) for(int j=1;j<=n;j++) if(A[i][j]) addedge(j, i + n, INF, 0);
    for(int i=1;i<=m;i++)
    {
        for(int j=1;j<=S[i]+1;j++)
            addedge(i + n, end, T[i][j] - T[i][j-1], W[i][j]);
    }
    LL cost = 0;
    int ans = minCostMaxflow(start, end, cost);
    printf("%lld\n", cost);
    return 0;
}