poj 1698Alice's Chance 最大流模板题

//建立一个超级源点和一个超级汇点
//从超级源点到每一个film的权值为需要在这个film工作的天数D
//然后从film到每个星期的第j天为一条权值为1的边
//从每个星期的第j天到超级汇点的权值为1
//这样就可以只需要验证从超级源点到超级汇点的最大流是否和所有film的天数之和是否相等
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
#define maxn 1010
#define inf 0x7fffffff
struct node
{
    int to;
    int cap;
    int next;
}edge[maxn*1000];
int nedge=0;
int head[maxn];
queue<int> que;
int dis[maxn];
int st=0;
int en=1001;
void add(int u,int v,int cap)
{
    edge[nedge].to=v;
    edge[nedge].cap=cap;
    edge[nedge].next=head[u];
    head[u]=nedge++;
    edge[nedge].to=u;
    edge[nedge].cap=0;
    edge[nedge].next=head[v];
    head[v]=nedge++;
}
int bfs()
{
    int i;
    memset(dis,-1,sizeof(dis));
    while(que.size())
    que.pop();
    dis[st]=0;
    que.push(st);
    while(que.size())
    {
        int u=que.front();
        que.pop();
        for(i=head[u];i!=-1;i=edge[i].next)
        {
           int v=edge[i].to;
           if(dis[v]<0&&edge[i].cap>0)
           {
               dis[v]=dis[u]+1;
               que.push(v);
           }
        }
    }
    if(dis[en]>0)
    return 1;
    return 0;
}
int dfs(int x,int mx)
{
    if(x==en)
    return mx;
    int i;
    int ans=0;
    int a;
    for(i=head[x];i!=-1;i=edge[i].next)
    {
        int v=edge[i].to;
        if(dis[v]==dis[x]+1&&edge[i].cap>0&&(a=dfs(v,min(mx,edge[i].cap))))
        {
            edge[i].cap-=a;
            edge[i^1].cap+=a;
            ans+=a;
            mx-=a;
            if(!mx)
            break;
        }
    }
    if(!ans)
    dis[x]=-1;
    return ans;
}
int main()
{
   // freopen("in.txt","r",stdin);
    int T;
    scanf("%d", &T);
    while(T--)
    {
        int N,i,j,D,W,vis[maxn],hash[maxn];
        int sum = 0;
        memset(head,-1,sizeof(dis));
        memset(vis,0,sizeof(vis));
        nedge=0;
        scanf("%d" , &N);
        for(i = 1; i <= N ;i++)
        {
            memset(hash,0,sizeof(hash));
            for(j = 1 ; j <= 7 ;j++)
            {
                int t;
                scanf("%d", &t);
                hash[j] = t;
            }
            scanf("%d %d" , &D ,&W);
            add(st, i ,D);
            sum += D;
            while(W--)
            {
                for(j=1; j<= 7 ;j++)
                if(hash[j])
                {
                    add(i, 50+j+W*10 ,1);
                    if(!vis[50+j+W*10])  
                    {
                        add(50+j+W*10, en ,1);
                         vis[50+j+W*10] = 1;
                    }
                }
            }
        }
        int ans = 0;
        int res;
        while(bfs())
        while(res=dfs(0,inf))
        ans+=res;
        if(ans == sum)
        printf("Yes\n");
        else
        printf("No\n");
    }
    return 0;
}