vijos P1352 最大獲利(最小割)

請不要隨便指點別人該怎麼做、每個人的人生都應該自己掌握、你給不了別人一切、你也不懂別人的憂傷、

                                                                                          微笑不代表快樂、哭泣不一定悲傷

               不努力怎麼讓關心你的人幸福、不努力怎麼讓看不起你的人絕望、

                                                                                                                                    

                                                                                                                                                              我用生命在奮鬥——lx_Zz

—————————————————————————————————————————————————————————————

—————————————————————————    華麗的分割線    ————————————————————————————

—————————————————————————————————————————————————————————————

仰慕完《最小割模型在信息學中的應用》論文之後切一下例題、發現自己太弱了。。。。orz

/* 題目： vijos 1352  */
/* 作者： lx_Zz       */
/* 時間： 2014.5.20   */
#include<stdio.h>
#include<string.h>
#include<queue>
#include<algorithm>
using namespace std;
#define N 60005
#define INF 0x7fffffff
int n,m,k;
int level[N];
struct node
{
    int to,next,cost;
}edge[500005];
int num[N];
int head[N];
int t_head[N];
int start,end;
void init()
{
    k=0;
    memset(head,-1,sizeof(head));
}
int bfs(int s,int t)//對頂點進行標號、找出層次圖
{
    memset(level,0,sizeof(level));
    queue<int>q;
    q.push(s);
    level[s]=1;
    while(!q.empty())
    {
        int now=q.front();
        q.pop();
        for(int i=head[now];i!=-1;i=edge[i].next)
        {
            int y=edge[i].to;
            if(!level[y]&&edge[i].cost>0)
            {
                level[y]=level[now]+1;
                q.push(y);
            }
        }
    }
    return level[t]!=0;//匯點是否在層次圖中
}
int dfs(int s,int cp)//在層次圖中尋找增廣路進行增廣
{
    int flow=0,temp;
    int t;
    if(s==end||cp==0)return cp;
    for(;t_head[s]+1;t_head[s]=edge[t_head[s]].next)
    {
        int y=edge[t_head[s]].to;
        if(level[s]+1==level[y])
        {
            temp=dfs(y,min(cp,edge[t_head[s]].cost));
            if(temp>0)
            {
                edge[t_head[s]].cost-=temp;
                edge[t_head[s]^1].cost+=temp;
                flow+=temp;
                cp-=temp;
                if(cp==0)break;
            }
        }
    }
    return flow;
}

int dinic()
{
    int ans=0,flow=0;
    while(bfs(start,end))//匯點不在層次圖中、算法終止
    {
        for(int i=0;i<=end;i++)
            t_head[i]=head[i];
        ans+=dfs(start,INF);
    }
    return ans;
}
void add(int x,int y,int val)
{
    edge[k].to=y;
    edge[k].cost=val;
    edge[k].next=head[x];
    head[x]=k++;

    edge[k].to=x;
    edge[k].cost=0;
    edge[k].next=head[y];
    head[y]=k++;
}

int main()
{
    //freopen("C:\\Users\\終將我要華麗的逆襲\\Desktop\\lx_Zz_in.txt","r",stdin);
    //freopen("C:\\Users\\終將我要華麗的逆襲\\Desktop\\lx_Zz_out.txt","w",stdout);

	int n,m;
	scanf("%d%d",&n,&m);
	init();
	int x;
	start=0;end=n+m+1;
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&x);
		add(start,i,x);
	}
	int sum=0;
	for(int i=1;i<=m;i++)
	{
		int a,b,c;
		scanf("%d%d%d",&a,&b,&c);
		add(a,n+i,INF);
		add(b,n+i,INF);
		add(n+i,end,c);
		sum+=c;
	}
	int ans=dinic();
	printf("%d\n",sum-ans);
    return 0;
}

非遞歸更高效的

/* 題目： vijos 1352  */
/* 作者： lx_Zz       */
/* 時間： 2014.5.20   */
#include<cstdio>
#include<queue>
#include<vector>
#include<map>
#include<string>
#include<iostream>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
const int INF=0x7ffffff;
const int MAXN=60005;//點數的最大值
const int MAXM=500005;//邊數的最大值


struct Node
{
    int from,to,next;
    int cap;
}edge[MAXM];
int tol;

int dep[MAXN];//dep爲點的層次
int head[MAXN];

int n;
void init()
{
    tol=0;
    memset(head,-1,sizeof(head));
}
void add(int u,int v,int w)//第一條變下標必須爲偶數
{
    edge[tol].from=u;
    edge[tol].to=v;
    edge[tol].cap=w;
    edge[tol].next=head[u];
    head[u]=tol++;
    edge[tol].from=v;
    edge[tol].to=u;
    edge[tol].cap=0;
    edge[tol].next=head[v];
    head[v]=tol++;
}

int BFS(int start,int end)
{
    int que[MAXN];
    int front,rear;
    front=rear=0;
    memset(dep,-1,sizeof(dep));
    que[rear++]=start;
    dep[start]=0;
    while(front!=rear)
    {
        int u=que[front++];
        if(front==MAXN)front=0;
        for(int i=head[u];i!=-1;i=edge[i].next)
        {
            int v=edge[i].to;
            if(edge[i].cap>0&&dep[v]==-1)
            {
                dep[v]=dep[u]+1;
                que[rear++]=v;
                if(rear>=MAXN)rear=0;
                if(v==end)return 1;
            }
        }
    }
    return 0;
}
int dinic(int start,int end)
{
    int res=0;
    int top;
    int stack[MAXN];//stack爲棧，存儲當前增廣路
    int cur[MAXN];//存儲當前點的後繼
    while(BFS(start,end))
    {
        memcpy(cur,head,sizeof(head));
        int u=start;
        top=0;
        while(1)
        {
            if(u==end)
            {
                int min=INF;
                int loc;
                for(int i=0;i<top;i++)
                  if(min>edge[stack[i]].cap)
                  {
                      min=edge[stack[i]].cap;
                      loc=i;
                  }
                for(int i=0;i<top;i++)
                {
                    edge[stack[i]].cap-=min;
                    edge[stack[i]^1].cap+=min;
                }
                res+=min;
                top=loc;
                u=edge[stack[top]].from;
            }
            for(int i=cur[u];i!=-1;cur[u]=i=edge[i].next)
              if(edge[i].cap!=0&&dep[u]+1==dep[edge[i].to])
                 break;
            if(cur[u]!=-1)
            {
                stack[top++]=cur[u];
                u=edge[cur[u]].to;
            }
            else
            {
                if(top==0)break;
                dep[u]=-1;
                u=edge[stack[--top]].from;
            }
        }
    }
    return res;
}

int main()
{
    //freopen("C:\\Users\\終將我要華麗的逆襲\\Desktop\\lx_Zz_in.txt","r",stdin);
    //freopen("C:\\Users\\終將我要華麗的逆襲\\Desktop\\lx_Zz_out.txt","w",stdout);

	int n,m;
	int start,end;
	scanf("%d%d",&n,&m);
	init();
	int x;
	start=0;end=n+m+1;
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&x);
		add(start,i,x);
	}
	int sum=0;
	for(int i=1;i<=m;i++)
	{
		int a,b,c;
		scanf("%d%d%d",&a,&b,&c);
		add(a,n+i,INF);
		add(b,n+i,INF);
		add(n+i,end,c);
		sum+=c;
	}
	int ans=dinic(start,end);
	printf("%d\n",sum-ans);
    return 0;
}