HDU-4322-Candy

无比伦比坑爹的一个题啊~这个题数据有问题，最后一个矩阵如果不用二维数组保存输入的话就会无限WA~~

       这个题想到网络流应该不难，建图想了半天。后来才想到优先满足孩子们满意的糖果，这样的话就可以建立最大费用最大流，建立源点和汇点，建图如下：

1、源点和每个糖果相连，费用为0，流量为1；

2、孩子和汇点相连，分2种情况，一种是sa[i](第i个孩子要的满意度) %k==0这种情况直接连向汇点一条费用为-k，流量为sa[i]/k的边。对于sa[i]%k!=0，那么拆边，除了连前面的那条边，另外连一条费用为-sa[i]%k，流量为1的边；

3、对于所给矩阵每个为1的值，将对应的糖果和孩子连一条费用为0，流量为1的边；

     剩下的就是模板了，最后统计下缺多少就行了

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
const int inf=1<<29;
const int maxn=50;
const int maxm=maxn*maxn;
int e,st,des,n,m,k,sa[14],head[maxn],pnt[maxm],nxt[maxm],flow[maxm],cost[maxm],dist[maxn];
int pre[maxn],cnt[maxn],map[maxn][maxn];
bool vis[maxn];
queue<int> q;
void AddEdge(int u,int v,int c,int f)
{
    pnt[e]=v;nxt[e]=head[u];flow[e]=f;cost[e]=c;head[u]=e++;
    pnt[e]=u;nxt[e]=head[v];flow[e]=0;cost[e]=-c;head[v]=e++;
}
bool Spfa(int st,int des)
{
    for(int i=st;i<=des;i++)
    {
        dist[i]=inf;
        pre[i]=-1;
    }
    dist[st]=0;
    q.push(st);
    while(!q.empty())
    {
        int u=q.front();
        vis[u]=0;
        q.pop();
        for(int i=head[u];i!=-1;i=nxt[i])
        {
            if(flow[i]&&dist[pnt[i]]>dist[u]+cost[i])
            {
                dist[pnt[i]]=dist[u]+cost[i];
                pre[pnt[i]]=i;
                if(!vis[pnt[i]])
                {
                    vis[pnt[i]]=1;
                    q.push(pnt[i]);
                } 
            }
        }
    }
    return dist[des]!=inf;
}

int maxflow(int st,int des)
{
    int ans=0;
    while(Spfa(st,des))
    {
        int mini=inf;
        for(int i=pre[des];i!=-1;i=pre[pnt[i^1]])
        {
            mini=min(mini,flow[i]);
        }
        for(int i=pre[des];i!=-1;i=pre[pnt[i^1]])
        {
            flow[i]-=mini;
            flow[i^1]+=mini;
        }
        ans+=mini;
    }
    return ans;
}
int main()
{
    int T,cas=1;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d%d",&n,&m,&k);
        e=st=0;des=n+m+1;
        memset(head,-1,sizeof(head));
        memset(cnt,0,sizeof(cnt));
        for(int i=1;i<=n;i++)
            AddEdge(st,i,0,1);
        for(int i=1;i<=m;i++)
        {
            scanf("%d",&sa[i]);
            AddEdge(n+i,des,-k,sa[i]/k);
            if(sa[i]%k)
                AddEdge(n+i,des,-(sa[i]%k),1);
        }
        for(int i=1;i<=m;i++)
            for(int j=1;j<=n;j++)
            {
                int val;
                scanf("%d",&map[i][j]);
            }
        for(int i=1;i<=m;i++)
            for(int j=1;j<=n;j++)
                if(map[i][j])
                    AddEdge(j,n+i,0,1);
        int res=n-maxflow(st,des);
        for(int i=head[des];i!=-1;i=nxt[i])
            cnt[pnt[i]-n]+=flow[i];
        bool is=true;
        for(int i=1;i<=m;i++)
        {
            if(cnt[i]*k<sa[i])
                res-=sa[i]-cnt[i]*k;
            if(res<0)
            {
                is=false;
                break;
            }
        }
        if(is)
            printf("Case #%d: YES\n",cas++);
        else
            printf("Case #%d: NO\n",cas++);
    }
    return 0;
}