BZOJ1001: [BeiJing2006]狼抓兔子

//法一
#include<bits/stdc++.h>
#define rep(i,l,r) for (int i=l;i<=r;i++)
using namespace std;
const int maxn=1000005,inf=0x3f3f3f3f;
int n,m,tot=1,S,T;
long long ans;
int read(){
    int ans=0,f=1,c=getchar();
    while(c<‘0‘||c>‘9‘) {if(c==‘-‘) f=-1; c=getchar();}
    while(c>=‘0‘&&c<=‘9‘) {ans=ans*10+(c-‘0‘); c=getchar();}
    return ans*f;
}    
int head[maxn],d[maxn],q[maxn],cur[maxn];
struct node{int flow,obj,pre;}e[6*maxn];
void add(int a,int b,int w){
    tot++; e[tot].obj=b; e[tot].flow=w; e[tot].pre=head[a]; head[a]=tot;
    tot++; e[tot].obj=a; e[tot].flow=w; e[tot].pre=head[b]; head[b]=tot;
}
int bfs(){
    rep(i,1,n*m) d[i]=-1;
    queue<int> q; q.push(S); d[S]=0;
    while (!q.empty()){
        int u=q.front(); q.pop();
        for (int j=head[u];j;j=e[j].pre){
            int v=e[j].obj;
            if (d[v]==-1&&e[j].flow) d[v]=d[u]+1,q.push(v);
        }    
    }
    if (d[T]==-1) return 0;
    return 1;
}
int dfs(int x,int a){
    if(x==T||a==0) return a;
    int f,flow=0;
    for(int &i=cur[x];i;i=e[i].pre){
        int now=e[i].obj;
        if(d[now]==d[x]+1&&(f=dfs(now,min(a,e[i].flow)))>0){
            e[i].flow-=f;
            e[i^1].flow+=f;
            flow+=f;
            a-=f;
            if(!a) break;
        }
    }
    return flow;
}
int main()
{
    int x;
    n=read(); m=read();
    S=1; T=n*m;
    for(int i=1;i<=n;i++)
        for(int j=1;j<m;j++) {x=read();int v=(i-1)*m+j; add(v,v+1,x);}
    for(int i=1;i<n;i++)
        for(int j=1;j<=m;j++) {x=read(); int v=(i-1)*m+j; add(v,v+m,x);}
    for(int i=1;i<n;i++)
        for(int j=1;j<m;j++) {x=read(); int v=(i-1)*m+j; add(v,v+m+1,x);}
    while(bfs()){
        for(int i=1;i<=n*m;i++) cur[i]=head[i]; ans=ans+dfs(S,inf);
    }
    printf("%lld\n",ans);
    return 0;
}

//法二
#include<bits/stdc++.h> 
using namespace std;
 
int n,m;
#define maxn 2000011
#define maxm 6000011
#define LL long long
struct Edge{int to,next,v;}edge[maxm];int first[maxn],le=2;
void in(int x,int y,int v) {Edge &e=edge[le];e.to=y;e.v=v;e.next=first[x];first[x]=le++;}
void insert(int x,int y,int v) {in(x,y,v);in(y,x,v);}
 
int tot;
struct qnode
{
    int id;LL v;
    bool operator > (const qnode &b) const {return v>b.v;}
};
priority_queue<qnode,vector<qnode>,greater<qnode> > q;
LL dis[maxn];bool vis[maxn];
LL dijkstra(int s,int t)
{
    for (int i=1;i<=tot;i++) dis[i]=1e18;
    dis[s]=0;
    q.push((qnode){s,0});
    while (!q.empty())
    {
        const int now=q.top().id; q.pop();
        if (vis[now]) continue;
        vis[now]=1;
        for (int i=first[now];i;i=edge[i].next)
        {
            const Edge &e=edge[i];
            if (dis[e.to]>dis[now]+e.v)
            {
                dis[e.to]=dis[now]+e.v;
                q.push((qnode){e.to,dis[e.to]});
            }
        }
    }
    return dis[t];
}
 
int main()
{
    scanf("%d%d",&n,&m);
    if (n==1 || m==1)
    {
        if (m==1) swap(n,m);
        int ans=0x3f3f3f3f;
        for (int i=1,x;i<m;i++) scanf("%d",&x),ans=min(ans,x);
        printf("%d\n",ans);
        return 0;
    }
    tot=(n-1)*(m-1)*2+2; int s=tot-1,t=tot;
    for (int j=1,x;j<m;j++) scanf("%d",&x),insert(t,j,x);
    for (int i=2;i<=n-1;i++)
    {
        int u=(i-1)*(m-1),v=(i-2)*(m-1),w=(n-1)*(m-1);
        for (int j=1,x;j<m;j++)
        {
            scanf("%d",&x);
            insert(u+j,v+j+w,x);
        }
    }
    for (int j=1,x;j<m;j++) scanf("%d",&x),insert((n-2)*(m-1)+j+(n-1)*(m-1),s,x);
     
    for (int i=1,x;i<n;i++)
    {
        int w=(n-1)*(m-1),u=(i-1)*(m-1);
        scanf("%d",&x);
        insert(s,u+1+w,x);
        for (int j=2;j<m;j++)
        {
            scanf("%d",&x);
            insert(u+j-1,u+w+j,x);
        }
        scanf("%d",&x);
        insert(t,u+m-1,x);
    }
     
    for (int i=1;i<n;i++)
    {
        int u=(i-1)*(m-1),v=(n-1)*(m-1);
        for (int j=1,x;j<m;j++)
        {
            scanf("%d",&x);
            insert(u+j,u+j+v,x);
        }
    }
    printf("%lld\n",dijkstra(s,t));
    return 0;
}