poj Alice's Chance(最大流解题)

题目信息：Alice's Chance

解法：

    利用最大网络流，以0表示源点，1表示汇点，以源点建立与每一天的边，边的容量是1，根据题目数据建立每周的给定的一天与每个电影之间的边，容量为1，建立每个film与汇点的边，容量为该部电影所需花费的时间，求出最大流和每部电影所花费天数的和比较，相等输出yes，否则输出No

源代码：

//Accepted 2196K 922MS 
//很险差点超时 
#include<iostream>
#include<queue>
#include<cstring>
using namespace std;
#define VEX 500
int cpt[VEX][VEX],flow[VEX][VEX];
int sum;
int f[7];
int Edmonds_Karp(int src,int sink,int node)
 {
     
     int d[VEX],pre[VEX];//d是增广路长度，pre记录前驱;
     int i,j;
 
     memset(flow,0,sizeof(flow));
     while(true)
     {
         memset(pre,-1,sizeof(pre));
         d[src]=0x7fffffff;
         queue<int> que;
         que.push(src);
         while(!que.empty() && pre[1]<0)
         {
             int t=que.front();
             que.pop();
             for(i=0;i<node;++i)
             {
                 if(pre[i]<0 && (j=cpt[t][i]-flow[t][i]))
                 {
                     pre[i]=t;
                     que.push(i);
                     d[i]=min(d[t],j);
                 }
             }
             
         }
         if(pre[sink]<0) break;
         for(i=1;i!=0;i=pre[i])
         {
             flow[pre[i]][i]+=d[1];
             flow[i][pre[i]]-=d[1];
         }
     }
     for(j=i=0;i<node;j+=flow[0][i++]);
     return j;
 }
int Init()
{
   sum=0;
   int Maxw=0;//记录跨度最长的电影需要周数 
   int n,i,j;
   int D,W;
   int filmNum=0;
   memset(cpt,0,sizeof(cpt)); 
   cin>>n;
   //0 is the source,1 is the sink,and the2--21 is the films,the 22--500 is the days
   for(i=22;i<500;++i) cpt[0][i]=1;  
   while(n--)
   {
       for(i=0;i<7;++i)
           cin>>f[i];
       cin>>D>>W;
       sum+=D;
       if(Maxw<W) Maxw=W; 
       cpt[filmNum+2][1]=D;
       int temp=W*7;
       for(i=0;i<temp;i+=7)
       {
          for(j=0;j<7;++j)
          {
              cpt[22+i+j][filmNum+2]=f[j];
          }
       }   
       filmNum++;       
   }
   return 22+7*Maxw; 
}
int main()
{
    int T;//the number of test cases
    cin>>T;
    while(T--)
    {
        int node=Init();
        int temp=Edmonds_Karp(0,1,node);
        //cout<<temp<<" "<<sum<<endl;
        if(sum==temp)
            cout<<"Yes"<<endl;
        else cout<<"No"<<endl;
    }
    return 0;
    
}