Codeforces Gym 100733I The Cool Monkeys 拆点+最大流

原题链接：http://codeforces.com/gym/100733/problem/I

题意

有两颗树（只是树，不是数据结构），每棵树上有不同高度的树枝，然后有m只猴子在某棵树的前m高的树枝上睡觉。早上的时候，他们要到地上，但他们只能从一棵树上跳到另一棵树上，并且跳的高度差也有限制，每个树枝仅能被一只猴子踩过。问你，猴子们是否能够跳到另一棵树最低的m个树枝上。

题解

就每个点拆成两个点，然后连边，所有边的容量都为1，然后跑发最大流，考察是否为滿流即可。

代码

#include<iostream>
#include<stack>
#include<vector>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<cstdio>
#include<queue>
#define MAX_S (1<<10)+10
#define MAX_V 2222
#define MAX_N MAX_V
#define INF 1000000009
using namespace std;

struct edge {
    int to, cap, rev;
    bool isRev;

    edge(int t, int c, int r, bool i)
            : to(t), cap(c), rev(r), isRev(i) { }

    edge() { }
};

template <class T>
inline bool scan_d(T &ret)
{
    char c;
    int sgn;
    if(c=getchar(),c==EOF) return ; //EOF
    while(c!=' -' &&(c<'' ||c>'' )) c=getchar();
    sgn=(c==' -' )?-:;
    ret=(c==' -' )?:(c-'' );
    while(c=getchar(),c>='' &&c<='' ) ret=ret*+(c-'' );
    ret*=sgn;
    return ;
}

vector<edge> G[MAX_N];
int level[MAX_V];
int iter[MAX_V];

void init(int totNode) {
    for (int i = ; i <= totNode; i++)
        G[i].clear();
    memset(level, , sizeof(level));
    memset(iter, , sizeof(iter));
}

void add_edge(int from,int to,int cap) {
    G[from].push_back(edge (to, cap, G[to].size(),));
    G[to].push_back(edge (from, , G[from].size() - ,));
}

void bfs(int s) {
    queue<int> que;
    memset(level, -, sizeof(level));
    level[s] = ;
    que.push(s);
    while (!que.empty()) {
        int v = que.front();
        que.pop();
        for (int i = ; i < G[v].size(); i++) {
            edge &e = G[v][i];
            if (e.cap >  && level[e.to] < ) {
                level[e.to] = level[v] + ;
                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 >  && level[v] < level[e.to]) {
            int d = dfs(e.to, t, min(f, e.cap));
            if (d > ) {
                e.cap -= d;
                G[e.to][e.rev].cap += d;
                return d;
            }
        }
    }
    return ;
}

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

int ha[MAX_N],hb[MAX_N];
int m,na,nb,t;

int S=;
int T=;

bool cmp(int a,int b){
    return a>b;
}

int main() {
    cin.sync_with_stdio(false);
    cin >> m >> na >> nb >> t;
    for (int i = ; i < na; i++)cin >> ha[i];
    for (int i = ; i < nb; i++)cin >> hb[i];
    sort(ha, ha + na,cmp);
    sort(hb, hb + nb,cmp);
    for(int i=;i<m;i++)
        add_edge(S,i,);
    for(int i=;i<na;i++)
        add_edge(i,i+na,);
    for(int i=;i<=m;i++)
        add_edge((nb-i)+na+na+nb,T,);
    for(int i=;i<nb;i++)
        add_edge(i+na+na,i+na+na+nb,);
    for(int i=;i<na;i++)
        for(int j=;j<nb;j++)
            if(abs(ha[i]-hb[j])<t){
                add_edge(i+na,j+*na,);
                add_edge(j+*na+nb,i,);
            }
    int f=max_flow(S,T);
    if(f>=m){
        cout<<"S"<<endl;
        return ;
    }
    init(MAX_N);
    for(int i=;i<m;i++)
        add_edge(S,i+*na,);
    for(int i=;i<na;i++)
        add_edge(i,i+na,);
    for(int i=;i<=m;i++)
        add_edge((na-i)+na,T,);
    for(int i=;i<nb;i++)
        add_edge(i+na+na,i+na+na+nb,);
    for(int i=;i<na;i++)
        for(int j=;j<nb;j++)
            if(abs(ha[i]-hb[j])<t){
                add_edge(i+na,j+*na,);
                add_edge(j+*na+nb,i,);
            }
    f=max_flow(S,T);
    if (f >= m)
        cout << "S" << endl;
    else
        cout << "N" << endl;
    return ;
}