poj 3020 Antenna Placement(java + 匈牙利算法)

来源:互联网 发布:广州seo顾问 编辑:程序博客网 时间:2024/06/14 07:37

package 匈牙利算法;

import java.util.Scanner;

/**问题请参考http://poj.org/problem?id=3020
* @author rayli

* @date:2014-7-31 下午1:09:33
* @解题思路:请参考http://blog.csdn.net/lyy289065406/article/details/6647040
*
*/
public class AntennaPlacement
{
static int map[][];//城市结点之间的示意图
static boolean city[][];//根据城市结点构造二分图
boolean vist[];
int link[] = new int[401];


void search(int m, int n)
{//根据图map[][]构造二分图city[][]
int dir[][] = {{0, -1}, {0, 1}, {-1, 0}, {1, 0}};

for(int i=0; i<m; i++)
for(int j=0; j<n; j++)
{
if(map[i][j] != 0)
{
for(int k=0; k<4; k++)
{
int x = i + dir[k][0];
int y = j + dir[k][1];

if(x >=0 && x <m && y >= 0 && y < n)
if(map[x][y] != 0)
{
city[map[i][j]-1][map[x][y]-1] = true;
}
}


}
}
}

boolean dfs(int x, int n)
{
for(int y=0; y<n; y++)
{
if(city[x][y] && !vist[y])
{
vist[y] = true;

if(link[y] == 0 || dfs(link[y], n))
{
link[y] = x;
return true;
}
}
}
return false;
}

int algorithm(int n)
{//匈牙利算法
int M = 0;
for(int i=0; i<n; i++)
{
vist = new boolean[n];

if(dfs(i, n))
M++;
}
return M;
}

void output(int m, int n)
{//输出为无向二分图的最小路径覆盖 = 顶点数 – 最大二分匹配数/2
System.out.println((m - n/2));
}

public static void main(String args[])
{
Scanner cin = new Scanner(System.in);

int n = cin.nextInt();//要测试的数量

while(n-->0)
{
int r = cin.nextInt();//行
int c = cin.nextInt();//列

char str[][] = new char[r][c];//用来存储输入的字符串
map = new int[r][c];

for(int j=0; j<r; j++)
str[j] = cin.next().toCharArray();

int num = 0;//记录结点序号

//构造数字地图
for(int i=0; i<r; i++)
for(int j=0; j<c; j++)
{
if(str[i][j] == '*')
map[i][j] = ++num;
}

city = new boolean[num][num];//为二分图分配地址

AntennaPlacement ap = new AntennaPlacement();
ap.search(r, c);//根据图map[][]构造二分图city[][]
int tmp = ap.algorithm(num);//最大二分匹配数
ap.output(num, tmp);
}
}
}

原创粉丝点击