nyoj 489 哭泣天使

点击打开题目链接

该题主要在于建边，因为一行和一列单独确定一个点，所以把所有的行和列当成点建边，容量为1，再建一个源点s，与所有的行相连，容量为每一行哭泣天使的数量，建立一个汇点t，与所有的列相连，容量为相应列哭泣天使的容量，然后求最大流即可。

#include<iostream>
#include<cstdio>
#include<cstring>
#define clr(a,b) memset(a,b,sizeof(a))
#define M 0xfffff
using namespace std;
const int I=600;
int map[I][I],pre[I],dis[I],gap[I];
int A[I],B[I];


int SAP(int s,int t)
{
    clr(gap,0);
    clr(dis,0);
    clr(pre,-1);
    gap[0]=t+1;
    int u=s,ans(0);
    while(dis[s]<t+1)
    {
        int v;
        for(v=0;v<=t;v++)
        {
            if(map[u][v]>0 && dis[u]==dis[v]+1)
                break;
        }
        if(v<=t)
        {
            pre[v]=u;
            u=v;
            if(v==t)
            {
                int minflow=M;
                for(int i=v;i!=s;i=pre[i])
                    if(minflow>map[pre[i]][i])
                        minflow=map[pre[i]][i];
                ans+=minflow;
                for(int i=v;i!=s;i=pre[i])
                {
                    map[pre[i]][i]-=minflow;
                    map[i][pre[i]]+=minflow;
                }
                u=s;
            }
        }
        else
        {
            int mindis=t+10;
            for(int i=0;i<=t;i++)
            {
                if(map[u][i]>0 && mindis>dis[i]+1)
                    mindis=dis[i]+1;
            }
            --gap[dis[u]];
            if(gap[dis[u]]==0) return ans;
            dis[u]=mindis;
            gap[mindis]++;
            if(u!=s) u=pre[u];
        }
    }
    return ans;
}

int main()
{
    int ncase;
    cin>>ncase;
    while(ncase--)
    {
        int m,n;
        scanf("%d%d",&m,&n);
        int sum1(0),sum2(0);
        for(int i=1;i<=m;i++)
        {
            scanf("%d",&A[i]);
            sum1+=A[i];
        }
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&B[i]);
            sum2+=B[i];
        }
        if(sum1!=sum2)
        {
            printf("Terrible\n");
            continue;
        }
        clr(map,0);
        for(int i=1;i<=m;i++)
        {
            map[0][i]=A[i];
            for(int j=1;j<=n;j++)
            {
                map[i][m+j]=1;
                if(i<2)
                    map[m+j][m+n+1]=B[j];
            }
        }
        int ans=SAP(0,m+n+1);
        if(ans!=sum1) printf("Terrible\n");
        else printf("Not Sure\n");
    }
    return 0;
}