POJ 3020 Antenna Placement 匈牙利算法,最大流解法 难度:1

http://poj.org/problem?id=3020

#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;
char maz[62][62];
int e[500][500];
int d[4][2]={0,1,0,-1,1,0,-1,0};
int n,m;
vector <int > G[500];
bool vis[500];
void addedge(int from,int to){
    e[from][to]=1;
    G[from].push_back(to);
}
int dfs(int s){
    vis[s]=true;
    if(s==n*m+1){//printf("dfs %d ok
",s);
    return 1;}
    for(int i=0;i<G[s].size();i++){
        if(!vis[G[s][i]]&&e[s][G[s][i]]&&dfs(G[s][i])){
            e[s][G[s][i]]=0;
            e[G[s][i]][s]=1;
          //  printf("dfs %d ok
",s);
            return 1;
        }
    }
    //printf("dfs %d failed 
",s);
    return 0;
}
int maxflow(){
    int ans=0,f;
    while(1){
        f=dfs(n*m);
        memset(vis,0,sizeof(vis));
        if(f==0)break;
        ans+=f;
    }
    return ans;
}
void printe(){
    for(int i=0;i<m*n+2;i++){
        bool f=false;
        for(int j=0;j<m*n+2;j++){
            if(e[i][j]){printf("e[%d][%d] ",i,j);f=true;}
        }
       if(f) printf("
");
    }
}
int main(){
    int t;
    scanf("%d",&t);
    while((t--)&&scanf("%d%d",&n,&m)==2){
        for(int i=0;i<=n*n+1;i++){G[i].clear();}
        memset(e,0,sizeof(e));

        for(int i=0;i<n;i++){
            scanf("%s",maz[i]);
        }
        int ans=0;
        int star=0;
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                if(maz[i][j]=='*'){
                    star++;
                    if((i+j)&1){
                        addedge(n*m,i*m+j);//n*n s n*n+1 t
                       // printf("addedge %d %d %d %d:%d to %d
",n,0,i,j,n*m,i*m+j);
                        for(int k=0;k<4;k++){
                            int nx=i+d[k][0];
                            int ny=j+d[k][1];
                            if(nx>=0&&nx<n&&ny>=0&&ny<m&&maz[nx][ny]=='*'){
                                addedge(i*m+j,nx*m+ny);
                         //       printf("addedge %d %d %d %d:%d to %d
",i,j,nx,ny,i*m+j,nx*m+ny);
                            }
                        }
                    }
                    else {
                        addedge(i*m+j,n*m+1);
                        //printf("addedge %d %d %d %d:%d to %d
",i,j,n,1,i*m+j,n*m+1);
                        for(int k=0;k<4;k++){
                            int nx=i+d[k][0];
                            int ny=j+d[k][1];
                            if(nx>=0&&nx<n&&ny>=0&&ny<m&&maz[nx][ny]=='*'){
                                addedge(i*m+j,nx*m+ny);
                                e[i*m+j][nx*m+ny]=0;
                          //      printf("addedge %d %d %d %d:%d to %d
",i,j,nx,ny,i*m+j,nx*m+ny);
                            }
                        }
                    }
                }
            }
        }

        ans=maxflow();
        printf("%d
",star-ans);
    }
    return 0;
}