1277. Cops and Thieves

http://acm.timus.ru/problem.aspx?space=1&num=1277

拆点建图  dinic 加一点小优化

代码：

#include<iostream>

#include<cstdio>

#include<cstring>

#include<string>

#include<map>

#include<vector>

#include<stack>

#include<set>

#include<map>

#include<queue>

#include<algorithm>

#include<cmath>

#define LL long long

using namespace std;



const int INF=0x3f3f3f3f;

const int N=205;

const int M=100005;

int head[N],I;

struct node

{

    int j,next;

    int flow;

}edge[M];

int L[N];//距离标号

int st,nd;//

void add(int i,int j,int flow)//建图

{

    edge[I].j=j;

    edge[I].flow=flow;

    edge[I].next=head[i];

    head[i]=I++;

}

bool bfs(int x1,int x2)//从汇点向源点搜索并标号 如果搜不到返回 false

{                       //和从源点向汇点搜索并标号相比 在dfs的时候会节省一定的时间

    memset(L,-1,sizeof(L));

    queue<int>qt;

    qt.push(x1);

    L[x1]=0;

    while(!qt.empty())

    {

        int x=qt.front();

        qt.pop();

        for(int t=head[x];t!=-1;t=edge[t].next)

        {

            int j=edge[t].j;

            if(edge[t^1].flow>0&&L[j]==-1)//注意这里的边是 edge[t^1].flow 因为我们是反向搜索

            {

                L[j]=L[x]+1;

                qt.push(j);

            }

        }

    }

    if(L[x2]==-1)

    return false;

    return true;

}

int dfs(int x,int sum)

{

    if(x==nd)

    return sum;

    int tmp=sum;

    for(int t=head[x];t!=-1;t=edge[t].next)

    {

        int j=edge[t].j;

        if(edge[t].flow>0&&L[x]==L[j]+1)

        {

            int w=dfs(j,min(tmp,edge[t].flow));

            edge[t].flow-=w;

            edge[t^1].flow+=w;

            tmp-=w;

            if(tmp==0)

            break;

        }

    }

    return (sum-tmp);

}

void init(int n,int m,int s,int f)

{

    for(int i=1;i<=n;++i)

    {

        int k;

        cin>>k;

        if(i==s||i==f)

        continue;

        add(i,i+n,k);

        add(i+n,i,k);

    }

    while(m--)

    {

        int l,r;

        cin>>l>>r;

        if(l==s&&r==f)

        {add(l,r,INF);add(r,l,0);continue;}

        if(r==s&&l==f)

        {add(r,l,INF);add(l,r,0);continue;}

        if(l==s)

        {add(l,r,INF);add(r,l,0);}

        else

        {add(l+n,r,INF);add(r,l+n,0);}

        if(r==s)

        {add(r,l,INF);add(l,r,0);}

        else

        {add(r+n,l,INF);add(l,r+n,0);}

    }

}

int main()

{

    //freopen("data.in","r",stdin);

    int K;

    while(scanf("%d",&K)!=EOF)

    {

        memset(head,-1,sizeof(head));

        I=0;

        int n,m,s,f;

        cin>>n>>m>>s>>f;

        st=s;//初始化源点和汇点

        nd=f;

        init(n,m,s,f);//建图

        int ans=0;//最大流

        while(bfs(nd,st))//从汇点向源点bfs

        {

            int k;

            while((k=dfs(st,INF)))

            {ans+=k;if(ans>K) break;}

            if(ans>K)//优化在这里 如果ans大于K了以后 就没有必要再搜了

            break;//因为答案一定是"NO"了 而且K的上界是10000 所以会很快

        }

        if(ans<=K)

        cout<<"YES"<<endl;

        else

        cout<<"NO"<<endl;

    }

    return 0;

}