[BZOJ4823][最小割]CQOI2017：老C的方块

BZOJ4823
可以看出来是最小割，关键在于怎么建边表示出格子之间的关系
观察发现每个这样的格子在如下路径上总会在第二步跨过关键边

所以我们可以把格子染色，关键边两边分别染成红色和黄色，然后和红色格子相邻的染成绿色，黄色格子相邻的染成蓝色
则如上的路径经过的颜色序列是确定的
那么就可以在颜色的分层图上做最大流求出最小割了

Code：

#include<bits/stdc++.h>
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define INF 0x3f3f3f3f
using namespace std;
inline int read(){
    int res=0,f=1;char ch=getchar();
    while(!isdigit(ch)) {if(ch=='-') f=-f;ch=getchar();}
    while(isdigit(ch)) {res=(res<<1)+(res<<3)+(ch^48);ch=getchar();}
    return res*f; 
}
int mv[2][3][2]={-1,0,1,0,0,1,-1,0,1,0,0,-1};
const int N=1e5+5,M=1e6+5;
int n,m,s,t,p,tot;
int head[N],nxt[M],vis[M],c[M],st[N];
inline void add(int x,int y,int z){
    vis[tot]=y;nxt[tot]=head[x];head[x]=tot;c[tot++]=z;
    vis[tot]=x;nxt[tot]=head[y];head[y]=tot;c[tot++]=0;
}
inline bool bfs(){
    memset(st,-1,sizeof(st));
    queue<int>q;
    st[s]=0;q.push(s);
    while(q.size()){
        int x=q.front();q.pop();
        for(int i=head[x];i!=-1;i=nxt[i])
            if(c[i] && st[vis[i]]==-1){
                q.push(vis[i]);
                st[vis[i]]=st[x]+1;
                if(vis[i]==t) return 1;
            }
    }
    return 0;
}
int dfs(int v,int flow){
    if(v==t) return flow;
    int rest=flow,k;
    for(int i=head[v];i!=-1 && rest;i=nxt[i])
        if(c[i] && st[vis[i]]==st[v]+1){
            k=dfs(vis[i],min(rest,c[i]));
            if(!k) st[vis[i]]=-1;
            c[i]-=k;c[i^1]+=k;
            rest-=k;
        }
    return flow-rest;
}
int dinic(){
    int flow,mxflow=0;
    while(bfs())
        while(flow=dfs(s,INF)) mxflow+=flow;
    return mxflow;
}
vector<int>v[N];
map< pair<int,int>,pair<int,int> >Mp; 
int main(){
    memset(head,-1,sizeof(head));
    m=read();n=read();p=read();
    s=0,t=p+1;
    for(int i=1,x,y,z;i<=p;i++){
        y=read();x=read();z=read();v[x].pb(y);
        Mp[mp(x,y)]=mp(i,z);
    }
    for(int i=1;i<=n;i++) sort(v[i].begin(),v[i].end());
    for(int i=1,x,y,xw,yw,be,lx,ly,rx,ry;i<=n;i++)
        for(int j=0;j<v[i].size();j++){
            x=i,y=v[i][j]; 
            if((x&1)&&y%4==1){
                if(j<v[i].size()-1&&v[i][j+1]==y+1) add(Mp[mp(x,y)].fi,Mp[mp(x,y+1)].fi,min(Mp[mp(x,y)].se,Mp[mp(x,y+1)].se));
            }
            else if((x&1)&&y%4==2){
                for(int k=0;k<3;k++) if(Mp.find(mp(x+mv[0][k][0],y+mv[0][k][1]))!=Mp.end()) add(Mp[mp(x,y)].fi,Mp[mp(x+mv[0][k][0],y+mv[0][k][1])].fi,INF);
            }
            else if((x&1)==0&&y%4==0){
                if(j>0&&v[i][j-1]==y-1) add(Mp[mp(x,y)].fi,Mp[mp(x,y-1)].fi,min(Mp[mp(x,y)].se,Mp[mp(x,y-1)].se));
            }
            else if((x&1)==0&&y%4==3){
                for(int k=0;k<3;k++) if(Mp.find(mp(x+mv[1][k][0],y+mv[1][k][1]))!=Mp.end()) add(Mp[mp(x,y)].fi,Mp[mp(x+mv[1][k][0],y+mv[1][k][1])].fi,INF);
            }
            else if(((x+y)&1)&&(x&1)){
                for(int k=0;k<3;k++) if(Mp.find(mp(x+mv[0][k][0],y+mv[0][k][1]))!=Mp.end()) add(Mp[mp(x,y)].fi,Mp[mp(x+mv[0][k][0],y+mv[0][k][1])].fi,INF);
                add(s,Mp[mp(x,y)].fi,Mp[mp(x,y)].se);
            }
            else if((x&1)&&((x+y)&1)==0){
                for(int k=0;k<3;k++) if(Mp.find(mp(x+mv[1][k][0],y+mv[1][k][1]))!=Mp.end()) add(Mp[mp(x+mv[1][k][0],y+mv[1][k][1])].fi,Mp[mp(x,y)].fi,INF);
                add(Mp[mp(x,y)].fi,t,Mp[mp(x,y)].se);
            }
            else if(((x+y)&1)&&(x&1)==0){
                for(int k=0;k<3;k++) if(Mp.find(mp(x+mv[1][k][0],y+mv[1][k][1]))!=Mp.end()) add(Mp[mp(x,y)].fi,Mp[mp(x+mv[1][k][0],y+mv[1][k][1])].fi,INF);
                add(s,Mp[mp(x,y)].fi,Mp[mp(x,y)].se);
            }
            else{
                for(int k=0;k<3;k++) if(Mp.find(mp(x+mv[0][k][0],y+mv[0][k][1]))!=Mp.end()) add(Mp[mp(x+mv[0][k][0],y+mv[0][k][1])].fi,Mp[mp(x,y)].fi,INF);
                add(Mp[mp(x,y)].fi,t,Mp[mp(x,y)].se);
            }
        }
    cout<<dinic();
    return 0;
}