火力网(匈牙利算法)

2.火力网

题目大意:

给出一个N*N的网格,用’.'表示空地,用’X’表示墙。在网格上放碉堡,可以控制所在的行和列,但不能穿过墙。
问:最多能放多少个碉堡?

做法:

匈牙利匹配问题,不会的可以看这个我们可以把独立的每一行好每一列(中间没有墙)当做X部和Y部,然后再把有公共顶点的连边,然后再跑匈牙利就可以了

#include
using namespace std;
const int N=25;
int sum1,sum2,n,a[N][N],mp1[N][N],mp2[N][N],k,vis[N*N],cx[N*N],cy[N*N],mp[N*N][N*N],ans;
char c;
bool find(int x)
{
    for(int i=1;i<=sum2;i++)
    {
        if(!mp[x][i]) continue;
        if(!vis[i])
        {
            vis[i]=1;
            if(cy[i]==-1||find(cy[i]))
            {
                cx[x]=i;
                cy[i]=x;
                return true;
            }
        }
    }
    return false;
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            while(1)
            {
                scanf("%c",&c);
                if(c=='.'||c=='X') break;
            }
            if(c=='.') a[i][j]=0;
            else a[i][j]=1;
        }
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            if(a[i][j]==1) continue;
            sum1++;
            k=j;
            while(1)
            {
                if(a[i][k]==1||k>n) break;
                mp1[i][k]=sum1;
                k++;
            }
            j=k;
        }
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            if(a[j][i]==1) continue;
            sum2++;
            k=j;
            while(1)
            {
                if(a[k][i]==1||k>n) break;
                mp2[k][i]=sum2;
                k++;
            }
            j=k;
        }
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            if(mp1[i][j]!=0)
            {
                mp[mp1[i][j]][mp2[i][j]]=1;
            }
        }
    }
    for(int i=1;i<=max(sum1,sum2);i++) cx[i]=cy[i]=-1;
    for(int i=1;i<=sum1;i++)
    {
        if(cx[i]==-1)
        {
            for(int j=1;j<=sum2;j++) vis[j]=0;
            ans+=find(i);
        }
        
    }
    printf("%d",ans);
    return 0;
}

你可能感兴趣的:(算法,c++,图论)