Virtual Barber of the Army of Mages

http://acm.hust.edu.cn/vjudge/contest/view.action?cid=45301#problem/A

题意：有n个人，k个理发师一。每个人需要理两次头发。每人有到达时间和他们等待的最大时间。求每个人能不能在他们等待的最长时间之前理完发。如果能，输出他理发的时刻。

题解：最大流……n个人和时刻连容量为1的边，源点和人之间为容量为2的边，时刻和汇点之间为容量为k的边。

注意：时刻与人建边的时候要注意是在时间段内，要减去min。输出理发时刻时应该+min。

建图还是不熟啊……怒wa了一天……唉……

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <string>
#include <vector>
#include <list>
#include <map>
#include <queue>
#include <stack>
#include <bitset>
#include <algorithm>
#include <numeric>
#include <functional>
#include <set>
#include <fstream>

using namespace std;

const int maxn=1010;
const int INF=1<<28;
struct edge{
    int to,cap,rev;
};
vector<edge> G[maxn*maxn];
int level[maxn*maxn];
int iter[maxn*maxn];
int ti[maxn*maxn][2];
void add_edge(int from,int to,int cap)
{
    G[from].push_back((edge{to,cap,(int)G[to].size()}));
    G[to].push_back((edge{from,0,(int)G[from].size()-1}));
}

void bfs(int s)
{
    memset(level, -1, sizeof(level));
    queue<int> que;
    level[s]=0;
    que.push(s);
    while (!que.empty()) {
        int v=que.front();
        que.pop();
        for (int i=0; i<G[v].size(); i++) {
            edge &e=G[v][i];
            if (e.cap>0&&level[e.to]<0) {
                level[e.to]=level[v]+1;
                que.push(e.to);
            }
        }
    }
}

int dfs(int v,int t,int f)
{
    if (v==t) {
        return f;
    }
    for (int &i=iter[v]; i<G[v].size(); i++) {
        edge &e=G[v][i];
        if (e.cap>0&&level[v]<level[e.to]) {
            int d=dfs(e.to,t,min(f,e.cap));
            if (d>0) {
                e.cap-=d;
                G[e.to][e.rev].cap+=d;
                return d;
            }
        }
    }
    return 0;
}

int max_flow(int s,int t)
{
    int flow=0;
    for (; ; ) {
        bfs(s);
        if (level[t]<0) {
            return flow;
        }
        memset(iter, 0, sizeof(iter));
        int f;
        while ((f=dfs(s, t, INF))>0) {
            flow+=f;
        }
    }
}

int n,k;

int main()
{
   // freopen("/Users/apple/Desktop/A/A/in", "r", stdin);
   // freopen("/Users/apple/Desktop/A/A/out", "w", stdout);
    while ((scanf("%d%d",&n,&k))!=EOF) {
        for (int i= 0; i < maxn; i++)
            G[i].clear();
        memset(ti, 0, sizeof(ti));
        int min1=1000000;
        int max1=-1;
        int maxtime=-1;
        for (int i=0; i<n; i++) {
            scanf("%d%d",&ti[i][0],&ti[i][1]);
            if (ti[i][0]<min1) {
                min1=ti[i][0];
            }
            if (ti[i][0]+ti[i][1]-1>max1) {
                max1=ti[i][0]+ti[i][1]-1;
            }
        }
        //cout<<min1<<" "<<max1<<endl;
        //min1--;
        int time1=max1-min1+1;
        int s=n+time1;
        int t=s+1;
       // s到人 0~n-1
        for (int i=0; i<n; i++) {
            add_edge(s, i, 2);
         //   printf("i: s %d n %d
",s,i);
        }
        //cout<<"22222222"<<endl;
        //时刻到t n~n+time1-1
        for (int i=0; i<time1; i++) {
            add_edge(n+i, t, k);
          //  printf("i: time %d t %d
",n+i,t);
        }
        //cout<<"1111111"<<endl;
        //人到时刻 
        for (int i=0; i<n; i++) {
            for (int j=ti[i][0]; j<ti[i][1]+ti[i][0]; j++) {
                add_edge(i, n+j-min1, 1);
          //      printf("i: n %d time %d
",i,n+j-min1);
            }
        }
        if (max_flow(s, t)==2*n) {
            printf("Yes
");
            for (int i=0; i<n; i++) {
                int cnt=1;
                for (int j=0; j<G[i].size(); j++) {
                    if (G[i][j].cap==0&&G[i][j].to>=n&&G[i][j].to<=n+time1-1) {
                        int res=G[i][j].to-n+min1;
                        printf("%d%s",res,cnt!=2?" ":"
");
                        cnt++;
                    }
                    if (cnt==3) {
                        break;
                    }
                }
            }
        }
        else printf("No
");
    }
    return 0;
}