hdu3572 Task Schedule--Dinic算法 & 最大流 & 链式前向星 & vector

原题链接：http://acm.hdu.edu.cn/showproblem.php?pid=3572

题意：有M个机器(代表一天可以同时干M天的工作)，有N个任务。每个任务必须在Si或者以后开始做，在Ei或者之前完成，完成每个任务必须处理Pi个时间单位，问最后是否可以完成这N个任务

这个题的建图算是经典，因为限定每个时刻每台机器只能处理一个任务，所以可以把时间点分配给各个合法的机器...具体是先设定一个超级源点S，连向各个任务，容量为该任务所需时间，各个任务连向在范围内的时间点，容量为1（保证每个时刻xxx这个条件），所有时间点连向超级汇点T，容量为机器台数，最后求最大流，等于所有机器所需时间和的就是yes

这是链式前向星建图：

#define _CRT_SECURE_NO_DEPRECATE 

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>

using namespace std;

struct Edge
{
    int v;
    int w;
    int next;
}edge[5000005];

int n, m;
int s, t;
int num;
int head[5005];
int d[5005];

void add(int u, int v, int w)
{
    edge[num].v = v;
    edge[num].w = w;
    edge[num].next = head[u];
    head[u] = num++;

    edge[num].v = u;//注意发方向的不要写错了，orz。。。。
    edge[num].w = 0;
    edge[num].next = head[v];
    head[v] = num++;
}

bool bfs()
{
    queue<int> Q;
    Q.push(s);
    memset(d, 0, sizeof(d));
    d[s] = 1;
    while (!Q.empty())
    {
        int u = Q.front();
        Q.pop();
        for (int i = head[u]; i != -1; i = edge[i].next)
        {
            int v = edge[i].v;
            if (d[v] == 0 && edge[i].w > 0)
            {
                d[v] = d[u] + 1;
                Q.push(v);
            }
        }
    }
    return d[t] != 0;
}

int dfs(int u, int flow)
{
    if (u == t)
        return flow;

    int ans = 0;
    int x = 0;
    for (int i = head[u]; i != -1; i = edge[i].next)
    {
        if (edge[i].w > 0 && d[edge[i].v] == d[u] + 1)
        {
            x = dfs(edge[i].v, min(flow-ans, edge[i].w));
            ans += x;
            edge[i].w -= x;
            edge[i ^ 1].w += x;
            if (ans == flow)
                return flow;
        }
    }
    if (ans == 0)
        d[u] = 0;
    return ans;
}

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

int main()
{
    int T;
    int sum;
    int maxx;
    int u, v, w;
    scanf("%d", &T);
    for (int cas = 1; cas <= T; cas++)
    {
        scanf("%d%d", &n, &m);
        memset(head, -1, sizeof(head));
        s = 0;
        num = 0;
        maxx = -1;
        sum = 0;

        for (int i = 1; i <= n; i++)
        {
            scanf("%d%d%d", &w, &u, &v);
            sum += w;
            maxx = max(maxx, v);

            add(s, i, w);
            for (int j = u; j <= v; j++)
                add(i, j + n, 1);
        }

        t = n + maxx + 1;
        for (int i = 1; i <= maxx; i++)
            add(i + n, t, m);

        int ans = dinic();
        printf("Case %d: ", cas);
        if (ans != sum)
            printf("No\n\n");
        else
            printf("Yes\n\n");
    }
    return 0;
}

这是摘抄自：http://www.voidcn.com/blog/stay_accept/article/p-6075386.html

#include <set>
#include <queue>
#include <vector>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
using namespace std;
const int INF=0x3f3f3f3f;
struct node{
    int u,v,cap;
    node(){}
    node(int u,int v,int cap):u(u),v(v),cap(cap){}
}es[5000005];
int R,S,T;
int dis[5005],iter[5005];
vector<int> tab[5005];
void addedge(int u, int v, int cap){
    tab[u].push_back(R);
    es[R++]=node(u,v,cap);
    tab[v].push_back(R);
    es[R++]=node(v,u,0);
}
int bfs(){
    int i,h;
    queue<int> q;
    q.push(S);
    memset(dis,INF,sizeof(dis));
    dis[S]=0;
    while(q.size()){
        h=q.front();
        q.pop();
        for(i=0;i<tab[h].size();i++){
            node &e=es[tab[h][i]];
            if(e.cap>0&&dis[e.v]==INF){
                dis[e.v]=dis[h]+1;
                q.push(e.v);
            }
        }
    }
    return dis[T]<INF;
}
int dfs(int x,int maxflow){
    int flow;
    if(x==T)
    return maxflow;
    for(int &i=iter[x];i<tab[x].size();i++){//注意这里的iter数组，目的是加快查找的作用，很巧妙
        node &e=es[tab[x][i]];
        if(dis[e.v]==dis[x]+1&&e.cap>0){
            flow=dfs(e.v,min(maxflow,e.cap));
            if(flow){
                e.cap-=flow;
                es[tab[x][i]^1].cap+=flow;
                return flow;
            }
        }
    }
    return 0;
}
int dinic(){
    int ans,flow;
    ans=0;
    while(bfs()){
        memset(iter,0,sizeof(iter));//当然每次bfs后都要初始化
        while(flow=dfs(S,INF))
        ans+=flow;
    }
    return ans;
}
int main(){                                     
    int n,m,t,i,j,u,v,w,cas,sum;                
    scanf("%d",&t);                             
    for(cas=1;cas<=t;cas++){                   
        scanf("%d%d",&n,&m);                    
        R=S=sum=0,T=1005;                      
        for(i=0;i<=T;i++)                     
        tab[i].clear();
        for(i=1;i<=n;i++){
            scanf("%d%d%d",&w,&u,&v);
            addedge(0,i,w);
            for(j=u+n;j<=v+n;j++)
            addedge(i,j,1);
            sum+=w;
        }
        for(i=1;i<=500;i++)
        addedge(i+n,T,m);
        if(sum==dinic())
        printf("Case %d: Yes\n\n",cas);
        else
        printf("Case %d: No\n\n",cas);
    }
    return 0;
}