hdu 3572 Task Schedule 最大流 Dinic算法

题意：

有N个任务，每个任务只能在其对应的S天到E天执行，需要的时间为P，不要求连续执行。

有M台机器，每个机器同一时间只能执行一项任务，每项任务同一时间只能被一台机器执行。问是否存在一种时间安排使任务都能被完成。

这题之前只学了EK算法，超时妥妥的，于是去学了Dinic算法。

同时这题也难在建图：

从源点出发到每个任务点i建立一条边(s,i,p) ;

因为不要求连续执行，但同一时间每项任务只能被一台机器执行(容量为1)，所以如果任务i可以在某一天j被执行，则建立一条边(i,j,1);

每一天j(其实并不一定需要每天都建边，只要保证有任务被执行的那些天有边即可)到汇点都有边(j,t,M);

只要求出整个图的最大流，比较是否等于任务需要的总天数即可。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
#include<algorithm>
using namespace std;
#define maxn 1100
#define INF 0xfffffff
int deep[maxn],N,M,cnt,head[maxn];
int s,t;
struct Edge
{
    int v,cap;
    int next;
}edge[maxn*maxn];
int bfs()
{
    memset(deep,-1,sizeof(deep));
    queue<int>q;
    q.push(s);
    deep[s]=0;
    while(!q.empty())
    {
        int now=q.front();
        q.pop();
        for(int nex=head[now];nex!=-1;nex=edge[nex].next)
        {
            if(deep[edge[nex].v]==-1&&edge[nex].cap>0)
            {
                deep[edge[nex].v]=deep[now]+1;
                q.push(edge[nex].v);
            }
        }
    }
    if(deep[t]==-1) return 0;
    return 1;
}
int Min(int a,int b)
{
    return a<b?a:b;
}
int dfs(int pos,int flow)
{
    if(pos==t) return flow;
    int t=0,sum=0;
    for(int nex=head[pos];nex!=-1;nex=edge[nex].next)
    {
        if(deep[edge[nex].v]==deep[pos]+1&&edge[nex].cap>0)
        {
            t=dfs(edge[nex].v,Min(flow-sum,edge[nex].cap));
            if(t>0)
            {
                edge[nex].cap-=t;
                edge[nex^1].cap+=t;
                sum+=t;
                if(sum==flow) break;
            }
            else deep[edge[nex].v]=-1;
        }
    }
    return sum;//当前点可增广的最大流量
}
int dinic()
{
    int sum=0;
    while(bfs())
        sum+=dfs(s,INF);
    return sum;
}
void add(int u,int v,int w)
{
    //u->v
    edge[cnt].v=v;
    edge[cnt].cap=w;
    edge[cnt].next=head[u];
    head[u]=cnt++;
    //v->u
    edge[cnt].v=u;
    edge[cnt].cap=0;
    edge[cnt].next=head[v];
    head[v]=cnt++;
}
int main()
{
    int T,Case;
    int S,E,P;
    scanf("%d",&T);
    for(Case=1;Case<=T;Case++)
    {
        s=0;
        cnt=0;
        memset(head,-1,sizeof(head));
        scanf("%d%d",&N,&M);
        int sum=0,m=-1;
        for(int i=1;i<=N;i++)
        {
            scanf("%d%d%d",&P,&S,&E);
            sum+=P;
            m=m>E?m:E;
            add(s,i,P);
            for(int j=S;j<=E;j++)
                add(i,j+N,1);
        }
        t=N+m+1;//任务数+最晚完成天数+1
        for(int i=1;i<=m;i++)
            add(i+N,t,M);
        printf("Case %d: ",Case);
        int ans=dinic();
        if(ans==sum) printf("Yes\n\n");
        else printf("No\n\n");
    }
}