Dinic(亲笔)

今天真有成就感啊。

参照"王欣上《浅谈基于分层思想的网络流算法》.doc"亲手实现了一下dinic算法。

p<-s;

While outdegree(s)>0

    u<-p.top;

    if u!=t

        if outdegree(u)>0

            设(u,v)为层次图中的一条边;

            p<-v;  

        else

            从p和层次图中删除点u,

            以及和u连接的所有边;

    else

        增广p（删除了p中的饱和边）;

        令p.top为p中从s可到达的最后顶点;

end while

Source Code
Problem: 1698  User: HEU_daoguang 
Memory: 920K  Time: 0MS 
Language: G++  Result: Accepted 
Source Code 
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
using namespace std;
#define E 30000
#define V 500
#define inf 0xffff
struct Edge
{
    int u,v,c,next;
}edge[E];
int n,m,cnt;
int dist[V];
int head[V];
int que[V];
int sta[V];

int film[25][10];
int week,sum,s,t;

void init(){
    cnt=0;
    memset(head,-1,sizeof(head));
}
void addedge(int u,int v,int c){
    edge[cnt].u=u;edge[cnt].v=v;edge[cnt].c=c;
    edge[cnt].next=head[u];head[u]=cnt++;

    edge[cnt].u=v;edge[cnt].v=u;edge[cnt].c=0;
    edge[cnt].next=head[v];head[v]=cnt++;
}

int dinic(int s,int t){
    int ans=0;
    while(true){
        int left,right,u,v;

        memset(dist,-1,sizeof(dist));
        left=right=0;
        que[right++]=s;
        dist[s]=0;

        while(left<right){//bfs构造层次网络
            u=que[left++];
            for(int k=head[u];k!=-1;k=edge[k].next){
                u=edge[k].u;
                v=edge[k].v;
                if(edge[k].c>0 && dist[v]==-1){
                    dist[v]=dist[u]+1;
                    que[right++]=v;
                    if(v==t){
                        left=right;
                        break;
                    }
                }
            }
        }

        if(dist[t]==-1) break;//汇点不在层次网络中,算法结束.

        int top=0;
        int now=s;

        while(true){//层次图中进行一次dfs增广
            if(now!=t){//dfs未到汇点
                int k;
                for(k=head[now];k!=-1;k=edge[k].next){//判断now之后是否有可用点(入度是否为0)
                    if(edge[k].c>0 && dist[edge[k].u]+1==dist[edge[k].v]) break;
                }
                if(k!=-1){//如果now之后有可用点(入度>0)
                    //cout << top << endl;
                    sta[top++]=k;//边入栈
                    now=edge[k].v;
                }
                else{//now之后没有可用点
                    if(top==0) break;
                    dist[edge[sta[--top]].v]=-1;//从p和层次图中删除点u以及u连接的所有边
                    now=edge[sta[top]].u;
                }

            }
            else{//dfs到汇点
                int flow=inf,ebreak;
                //(1)增广p
                for(int i=0;i<top;i++){
                    if(flow>edge[sta[i]].c){
                        flow=edge[sta[i]].c;
                        ebreak=i;
                    }
                }
                ans+=flow;
                for(int i=0;i<top;i++){
                    edge[sta[i]].c-=flow;//正向边减流
                    edge[sta[i]^1].c+=flow;//反向边加流
                }
                //(2)退至p中从源点可到达的最后一个顶点
                now=edge[sta[ebreak]].u;
                top=ebreak;
            }
        }

    }
    return ans;
}

void build()
{
    init();
    s=0,t=7*week+n+1;
    for(int i=1;i<=n;i++){
        addedge(s,i,film[i][8]);
        for(int j=1;j<=film[i][9];j++){
                for(int k=1;k<=7;k++){
                if(film[i][k]) addedge(i,n+(j-1)*7+k,1);
            }
        }
    }
    for(int i=1;i<=week;i++)
        for(int j=1;j<=7;j++)
            addedge(n+(i-1)*7+j,t,1);
}
int main()
{
    //freopen("in.txt","r",stdin);
    int ncase;
    scanf("%d",&ncase);
    while(ncase--)
    {

        sum=0,week=0;
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=9;j++)
            {
                scanf("%d",&film[i][j]);
            }
            week=max(week,film[i][9]);
            sum+=film[i][8];
        }
        build();
        int ans=dinic(s,t);
        if(ans==sum) printf("Yes\n");
        else printf("No\n");
    }
    return 0;
}