Task Schedule(最大流是否满流)

题目链接：Task Schedule

题意：有N个项目，M台机器。给出每个项目的开始时间和结束时间以及在完成这个项目所需要的时间。
每台机器每天只能处理一个项目，一个项目在不同的天可以被不同的机器处理。问你是否能按时完成所有项目。

思路：

1. 将源点和任务相连，容量是需要的天数。
2. 然后将任务需要在允许的天数内建边，容量为1。
3. 最后将所有的天数和汇点建边，容量为m，因为每天最多有m个，因为每个任务至少一天，如果同时工作，所以最多为m。

参考大佬博客：Task Schedule

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<queue>
#include<stack>
#include<vector>
#include<map>
#include<set>
#define inf 0x3f3f3f3f
using namespace std;

typedef long long ll;

const int N = 1010;

struct node
{
    int c;
    int to;
    int next;
} edge[1100000];

int tot;
int n,m;
int head[N];
int level[N];

bool bfs(int s, int d)
{
    memset(level, 0, sizeof(level));
    level[s] = 1;
    queue<int> q;
    q.push(s);
    while(q.size())
    {
        int u = q.front();
        if(u == d) return true;
        q.pop();
        for(int i = head[u]; i != -1; i = edge[i].next)
        {
            int v = edge[i].to;
            int c = edge[i].c;
            if(!level[v] && c)
            {
                q.push(v);
                level[v] = level[u]+1;
            }
        }
    }
    return false;
}

int dfs(int u, int d, int f)
{
    if(u == d) return f;
    int t = 0;
    int tmp;
    for(int i = head[u]; i != -1; i = edge[i].next)
    {
        int v = edge[i].to;
        int c = edge[i].c;
        if(c && level[v] == level[u]+1 && (tmp = dfs(v, d, min(f, c))))
        {
            edge[i].c -= tmp;
            edge[i^1].c += tmp;
            t += tmp;
            f -= tmp;
            if(!f) break;
        }
    }
    if(t) return t;
    level[u] = -1;
    return 0;
}

int Dinic(int s, int d)
{
    int ans = 0;
    while(bfs(s, d))
    {
        int tmp = dfs(s, d, inf);
        if(!tmp) break;
        ans += tmp;
    }
    return ans;
}

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

    edge[tot].c = 0;
    edge[tot].to = u;
    edge[tot].next = head[v];
    head[v] = tot++;
}

int main()
{
    int t,u,v,w;
    scanf("%d",&t);
    for(int ca = 1; ca <= t; ca++)
    {
        tot = 0;
        memset(head, -1, sizeof(head));
        int mn = inf,mx = -1;
        int sum = 0;
        scanf("%d %d",&n,&m);
        for(int i = 1; i <= n; i++)
        {
            scanf("%d %d %d",&w,&u,&v);
            sum += w;
            add(0, i, w);//将源点和每一个任务相连，权值为每个任务需要的天数
            if(u < mn) mn = u;
            if(v > mx) mx = v;
            for(int j = u; j <= v; j++)
            {
                add(i, j+n, 1);//将每一个任务与他们需要完成得天数区间内相连，权值为1，代表可在此天处理该项目
            }
        }
        int s = 0,d = mx+n+1;
        for(int i = mn+n; i < d; i++)
        {
            add(i, d, m);//最后将每个天数与汇点相连，权值为m,表示这一天可以有运行m个任务，因为每个任务至少需要一个机器1天，故最多运行m个任务
        }
        printf("Case %d: ",ca);
        if(sum == Dinic(s, d))
            printf("Yes\n");
        else
            printf("No\n");
        printf("\n");
    }
    return 0;
}