HDU3338 Kakuro Extension(最大流+思维构图)

#include <bits/stdc++.h>
using namespace std;

const int maxn = 100 + 5;
const int maxm = 1e6 + 5;
const int inf = 0x3f3f3f3f;
int n, m, d[maxn*maxn], sp, tp, maxflow;
int head[maxn*maxn], tot;
struct point{
    int top, lft, type;
} mp[maxn][maxn];
struct edge{
    int to, w, next;
} ed[maxm];
int num[maxn][maxn], ans[maxn*maxn];
char s[10];
inline void add( int u, int v, int w ){
    ed[++tot].to = v; ed[tot].w = w; ed[tot].next = head[u]; head[u] = tot;
    ed[++tot].to = u; ed[tot].w = 0; ed[tot].next = head[v]; head[v] = tot;
}

inline void init(){
    memset( head, -1 ,sizeof(head) );
    tot = 1;
}

inline void map_set(){
    for( int i=0; i<n; i++ )
        for( int j=0; j<m; j++ ){
            int cnt = 0;
            if( mp[i][j].type==4 ){
                for( int k=j+1; k<m && mp[i][k].type==1; k++ )
                    add( num[i][k], num[i][j], 8 ), cnt ++;
                add( num[i][j], tp, mp[i][j].top-cnt );
            }else if( mp[i][j].type==2 ){
                for( int k=j+1; k<m && mp[i][k].type==1; k++ )
                    add( num[i][k], num[i][j], 8 ), cnt ++;
                add( num[i][j], tp, mp[i][j].top-cnt );
                cnt = 0;
                for( int k=i+1; k<n && mp[k][j].type==1; k++ )
                    add( num[i][j]-1, num[k][j], 8 ), cnt ++;       //type==2时，让num[i][j]-1代表左下数字的点
                add( sp, num[i][j]-1, mp[i][j].lft-cnt );
            }else if( mp[i][j].type==3 ){
                for( int k=i+1; k<n && mp[k][j].type==1; k++ ) 
                    add( num[i][j], num[k][j], 8 ), cnt ++;
                add( sp, num[i][j], mp[i][j].lft-cnt );
            }
        }
}

inline bool bfs(){
    memset( d, 0, sizeof(d) );
    queue<int> q;
    d[sp] = 1;
    q.push(sp);
    while( !q.empty() ){
        int x = q.front();
        q.pop();
        for( int i=head[x]; i!=-1; i=ed[i].next ){
            int y = ed[i].to;
            if( ed[i].w && !d[y] ){
                d[y] = d[x] + 1;
                q.push(y);
                if( y==tp ) return 1;
            }
        }
    }
    return 0;
}

inline int min( int a, int b ){
    return a<b ? a:b;
}

inline int dfs( int x, int flow ){
    if( x==tp ) return flow;
    int rest = flow, k;
    for( int i=head[x]; i!=-1 && rest; i=ed[i].next ){
        int y = ed[i].to;
        if( ed[i].w && d[y]==d[x]+1 ){
            k = dfs( y, min(rest, ed[i].w) );
            if(!k) d[y] = 0;
            ed[i].w -= k;
            ed[i^1].w += k;
            rest -= k;
        }
    }
    return flow - rest;
}

inline void dinic(){
    int flow = maxflow = 0;
    while( bfs() ){
        while( flow=dfs(sp, inf) ) maxflow += flow;     //该题里 maxflow没什么用，可以不求
    }
}

inline void output(){
    memset( ans, 0, sizeof(ans) );                  //ans记录答案
    for( int i=head[sp]; i!=-1; i=ed[i].next ){     //超级源点sp是虚拟的，正好利用它进行遍历
        int u = ed[i].to;
        for( int j=head[u]; j!=-1; j=ed[j].next )
            ans[ed[j].to] = 8 - ed[j].w + 1;        //最后要加回1(其实这里为什么要减去边权而不直接用边权我不是很懂，求dalao指教)
    }
    for( int i=0; i<n; i++ ){
        putchar(‘_‘);
        for( int j=1; j<m; j++ ){
            if( mp[i][j].type!=1 ) printf(" _");
            else printf("%2d", ans[num[i][j]]);
        }
        puts("");
    }
}

int main(){
    // freopen("in.txt", "r", stdin);
    while( ~scanf("%d%d", &n, &m) ){
        init();
        int nump = 0;
        for( int i=0; i<n; i++ )
            for( int j=0; j<m; j++ ){
                scanf("%s", s);
                if( s[0]==‘.‘ ){
                    mp[i][j].type = 1;
                    mp[i][j].top = mp[i][j].lft = 0;
                }else if( s[0]==‘X‘ ){
                    if( s[4]==‘X‘ ) mp[i][j].type = 0;
                    else{
                        mp[i][j].type = 4;
                        mp[i][j].lft = 0; 
                        mp[i][j].top = (s[4]-‘0‘)*100 + (s[5]-‘0‘)*10 + s[6]-‘0‘;
                    }
                }else{
                    if( s[4]==‘X‘ ){
                        mp[i][j].type = 3;
                        mp[i][j].lft = (s[0]-‘0‘)*100 + (s[1]-‘0‘)*10 + s[2]-‘0‘;
                        mp[i][j].top = 0;
                    }else{
                        mp[i][j].type = 2;
                        mp[i][j].lft = (s[0]-‘0‘)*100 + (s[1]-‘0‘)*10 + s[2]-‘0‘;
                        mp[i][j].top = (s[4]-‘0‘)*100 + (s[5]-‘0‘)*10 + s[6]-‘0‘;
                    }
                }
                if( mp[i][j].type==2 ) nump += 2;           //type == 2的时候就是上半部分和左半部分都有数字，那么将其拆点成两个点
                else nump ++;
                num[i][j] = nump;           //将矩阵的二维编码转换成一维，便于进行操作
            }
        sp = 0; tp = nump+1;
        map_set();
        dinic();
        output();
    }

    return 0;
}