poj1698--最大流(Dinic)

题目大意：

爱丽丝要拍电影，有n部电影，规定爱丽丝每天只能拍一部电影，每部电影在每个礼拜只有固定的几天可以拍电影，只可以拍前面w个礼拜，并且这部电影要拍d天,问爱丽丝能不能拍完所有的电影。

思路：

建图。点1~350代表天数（因为最多只有350天），点351~370代表电影（最多只有20部电影）。从源点向每部电影连一条容量为d的边，从每部电影向可以拍摄它的那几天连一条容量为1的边，从每天向汇点连一条容量为1的边（限制每天只能拍一部电影）。再求一遍最大流，看其是否与sum(d[i])相等，如果是，就输出Yes，否则输出No。

具体看代码。

#include<iostream>
#include<cstdio>
#include<vector>
#include<queue>
#include<cstring>
using namespace std;
#define INF 2147483647
vector<int>g[];
queue<int>q;
struct Edge{
    int to,from,cap,flow;
}edge[];
int i,j,k,s,t,n,m,x,y,sum,w,d,cur[],T,dist[];
bool a[],vis[];
void addedge(int x,int y,int cap){
    edge[m++]=(Edge){y,x,cap,};
    edge[m++]=(Edge){x,y,,};
    g[x].push_back(m-);
    g[y].push_back(m-);
}
bool bfs(){
    memset(vis,,sizeof(vis));
    vis[s]=;
    dist[s]=;
    q.push(s);
    while(!q.empty()){
        int x=q.front();q.pop();
        for(int i=;i<g[x].size();i++){
            Edge& e=edge[g[x][i]];
            if(e.cap>e.flow&&!vis[e.to]){
                vis[e.to]=;
                dist[e.to]=dist[x]+;
                q.push(e.to);
            }
        }
    }
    return vis[t];
}
int dfs(int x,int a){
    if(x==t||!a)return a;
    int flow=,f;
    for(int& i=cur[x];i<g[x].size();i++){
        Edge& e=edge[g[x][i]];
        if(dist[x]+==dist[e.to]&&(f=dfs(e.to,min(a,e.cap-e.flow)))){
            e.flow+=f;
            edge[g[x][i]^].flow-=f;
            flow+=f;
            a-=f;
            if(!a)break;
        }
    }
    return flow;
}
int maxflow(){
    int flow=;
    while(bfs()){
        memset(cur,,sizeof(cur));
        flow+=dfs(s,INF);
    }
    return flow;
}
int main()
{
    scanf("%d",&T);
    while(T--){
        scanf("%d",&n);
        sum=m=;t=+n;
        for(i=;i<=n;i++){
            for(j=;j<=;j++)scanf("%d",&a[j]);
            scanf("%d%d",&d,&w);
            sum+=d;
            addedge(s,i+,d);
            for(j=;j<=;j++)
            if(a[j])for(k=;k<w;k++)addedge(i+,k*+j,);
        }
        for(i=;i<=;i++)addedge(i,t,);
        if(maxflow()==sum)printf("Yes\n");else printf("No\n");
        for(i=;i<=t;i++)g[i].clear();
    }
    return ;
}

poj1698