POJ 1698 构图+最大流

题目链接： http://poj.org/problem?id=1698

题目大意：

　　man需要一个安排表参见演出，有n(n<=20)场，每场指定了必须表演的天数Di，和截止的周数Week_i，还指定了必须星期几表演，要求man一天最多参加一场，问能否有一个安排使得满足要求，有输出Yes，否则输出No;

分析：

　　开始自己想了个392个点的构图,后来看了题解，发现我多了一层（我拆了点，其实没有必要）……

　　源ST分别向n个场次连边（1……n），边容量为各场指定的天数Di， 各场（1……n）分别作为一个节点，向指定的星期连边（注意从第一周到第Week_i的都要），容量为1（只要大于0的整数实际都可以）， 然后每天 （n+1……n+(Week*7)）分别向汇点ED连边，容量为1;

　　求一次最大流，看maxflow是否等于sum{ Di };

　　最多372个点，果断上Edmond_Karp!

代码：

/*1698    Accepted    1328K    141MS    G++    2140B    2012-06-19 20:42:23*/
#include <cstdio>
#include <cstring>
#include <cmath>
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;

#define mpair make_pair
#define pii pair<int,int>
#define MM(a,b) memset(a,b,sizeof(a));
typedef long long lld;
typedef unsigned long long u64;
template<class T> bool up_max(T& a,const T& b){return b>a? a=b,1 : 0;}
template<class T> bool up_min(T& a,const T& b){return b<a? a=b,1 : 0;}
#define maxn 390
const int inf= 2100000000;

int n;
int ST, ED;
int g[maxn][maxn];

int Read(){
    int i, j, k, t, D, W, sum= 0, week= 0;
    scanf("%d", &n);
    ST= 0;
    MM( g, 0 );
    vector<int> a;
    for(i=1;i<=n;++i){
        a.clear();
        for(j=1;j<=7;++j){
            scanf("%d", &t);
            if( 1==t ) a.push_back( j );
        }
        scanf("%d%d", &D, &W);
        g[ST][i]= D;
        sum+= D;
        up_max( week, W );
        for(j=1;j<=W;++j)
            for(k=0;k<a.size();++k)
                g[i][ n+(j-1)*7+a[k] ]= 1; ///!!!
    }
    ED= n + 7*week + 1;
    for(i=n+1;i<=n+7*week;++i) g[i][ED]= 1;
    return sum;
}

bool vis[maxn];
int que[maxn], pre[maxn];
bool bfs(){
    fill( vis, vis+1+ED, 0 );
    int head= 0, tail= 0;
    que[tail++]= ST;
    vis[ST]= 1;
    while( head<tail ){
        int u= que[head++];
        for(int v=1;v<=ED;++v){
            if( g[u][v]>0 && !vis[v] ){
                pre[v]= u;
                if( v==ED ) return 1;
                que[tail++]= v;
                vis[v]= 1;
            }
        }
    }
    return 0;
}

int Edmond_karp(){
    int ret= 0;
    while( bfs() ){
        int t= inf;
        for(int i=ED;i!=ST;i=pre[i])
            up_min( t, g[pre[i]][i] );
        ret+= t;
        for(int i=ED;i!=ST;i=pre[i]){
            g[pre[i]][i]-= t;
            g[i][pre[i]]+= t;
        }
    }
    return ret;
}

int main()
{
    //freopen("poj1689.in","r",stdin);
    int Cas;
    cin>>Cas;
    while( Cas-- ){
        int sum= Read();
        int ret= Edmond_karp();
        if( sum==ret ) puts("Yes");
        else puts("No");
    }
}