1. 程式人生 > >最大流演算法模板 挑戰程式設計競賽

最大流演算法模板 挑戰程式設計競賽

const int MAX_V=1e6;
struct edge{
    int to,cap,rev;     //終點,容量,反向邊
};

vector<edge> G[MAX_V];  //圖的鄰接表表示
int level[MAX_V];       //頂點到原點的距離標號
int iter[MAX_V];        //當前弧,在其之前的邊已經沒有用了


//增加一條從from到to的容量為cap的邊
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,0,G[from].size()-1});

}

//通過BFS計算從原點出發的距離標號
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);
            }
        }
    }
}


//通過DFS尋找增廣路
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;
}


//求解從s到t的最大流
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;
        }
    }
}