hdu 3572 Task Schedule （dinic 网络流）

题目链接：哆啦A梦传送门

题意：有N个任务，M个机器，现在每个任务要在[si,ei]时间段里完成pi个时刻工作，每个任务可以由多个机器完成，但每个机器在同一时刻只能运行一个任务。现在问：能否完成在规定的时间的全部任务？

 

题解：

我们先建模：源点到每个任务建边，权值为该任务的pi。每个任务连它的时刻，权值为1，最后我们将每个时刻与汇点相连，权值为M。

 

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;

const int N=6e5;
const int INF=0x3f3f3f3f;

struct edge{
    int from,to,w,next;
}e[N<<1];
int cnt;

int dep[1111],head[2111];
//queue<int> que;


void init(){
    memset(head,-1,sizeof(head));
    cnt=0;
}
void add(int u,int v,int w)
{
    e[cnt].from=u;
    e[cnt].to=v;
    e[cnt].w=w;
    e[cnt].next=head[u];
    head[u]=cnt++;

     e[cnt].from=v;
    e[cnt].to=u;
    e[cnt].w=0;
    e[cnt].next=head[v];
    head[v]=cnt++;
}

int bfs(int s,int t)
{
//    while(!que.empty()) que.pop();
    queue<int> que;
    memset(dep,-1,sizeof(dep));

    dep[s]=0;
    que.push(s);

    while(!que.empty()){
        int u=que.front();
        que.pop();
//        if(u==t) return 1;
            for(int i=head[u];i!=-1;i=e[i].next){
                int w=e[i].w;
                int v=e[i].to;
            if(w&&dep[v]==-1){
                dep[v]=dep[u]+1;
                que.push(v);
                if(v==t) return 1;
            }
        }

    }
    return 0;
}

int dfs(int u,int mi,int t){
    if(u==t||mi==0) return mi;
    int tmp;
    int ret=0;
//    for(int v=1;v<=t;v++){
        for(int i=head[u];i!=-1;i=e[i].next){
            int w=e[i].w;
            int v=e[i].to;
        if(w&&dep[v]==dep[u]+1&&mi-ret>0){ ///mi-ret>0少一次递归，优化
           tmp=dfs(v,min(mi-ret,w),t) ;
           if(!tmp) dep[v]=0;
           e[i].w-=tmp;
            e[i^1].w+=tmp;
            ret+=tmp;
            if(ret==mi) return ret;
        }
    }
    return ret;
}

int dinic(int s,int t)
{
    int ans=0;
    while(bfs(s,t)){
            ans+=dfs(s,INF,t);
    }
    return ans;
}

bool vis[600];
int main()
{

    int ncase,T=0;
    scanf("%d",&ncase);

    while(ncase--)
    {
        memset(vis,0,sizeof(vis));
        init();
        int n,m;
        scanf("%d%d",&n,&m);

        int s=0,t=n+500+1;
        int sum=0;
        for(int i=1;i<=n;i++)
        {
            int pi,si,ei;
            scanf("%d%d%d",&pi,&si,&ei);
            sum+=pi;
            add(s,500+i,pi);
            for(int j=si;j<=ei;j++){
                add(i+500,j,1);
                vis[j]=1;
            }

        }

        for(int i=1;i<=500;i++){
            if(vis[i])
            add(i,t,m);
        }

        printf("Case %d: ",++T);
        if(dinic(s,t)==sum){
            printf("Yes\n\n");
        }
        else printf("No\n\n");

    }

    return 0;
}