春招实习笔试题,是简单的bfs
题目
公司的程序员不够用了,决定把产品经理都转变为程序员以解决开发时间长的问题。
在给定的矩形网格中,每个单元格可以有以下三个值之一:
值0代表空单元格;
值1代表产品经理;
值2代表程序员;
每分钟,任何与程序员(在4个正方向上)相邻的产品经理都会变成程序员。
返回直到单元格中没有产品经理为止所必须经过的最小分钟数。
如果不可能,返回-1。
以下是一个4分钟转变完成的示例:
2 1 1 2 2 1 2 2 2 2 2 2 2 2 2
1 1 0 -> 2 1 0 -> 2 2 0 -> 2 2 0 -> 2 2 0
0 1 1 0 1 1 0 1 1 0 2 1 0 2 2
输入格式
不固定多行(行数不超过10),毎行是按照空格分割的数字(不固定,毎行数字个数不超过10)。
其中每个数组项的取值仅为0、1、2三种。
读取时可以按行读取,直到读取到空行为止,再对读取的所有行做转换处理。
输出格式
如果能够将所有产品经理变成程序员,则输出最小的分钟数。
如果不能够将所有的产品经理变成程序员,则返回-1.
输入样例1:
1 2 1
1 1 0
0 1 1
输出
3
输入样例2:
0 2
1 0
输出
-1
数据读取部分就不写了
思路是用visited数组来记录访问次序,最后再遍历找最大或没有被访问到的;
public int bfs(int[][] input, int n, int m){
int[][] visit = new int[n][m];//本来用来记录访问,现在记录长度
Queue<int[]> queue = new PriorityQueue<>();
for(int i=0;i<n;i++){
for(int j=0;j<m;j++){
if(input[i][j]==2){
visit[i][j] = 1;
queue.add(new int[] {i, j});
}
}
}
final int[] dx = {0,1,0,-1};
final int[] dy = {-1,0,1,0};
while(!queue.isEmpty()){
int temp[] = queue.poll();
int x = temp[0], y = temp[1], d = visit[x][y];
for(int i = 0; i<4; i++) {
int nx = x + dx[i], ny = y + dy[i];
if (nx >= 0 && nx < n && ny>=0 && ny<m && input[nx][ny] == 1 && visit[nx][ny] == 0){
visit[nx][ny] = d + 1;
queue.add(new int[] {nx, ny});
}
}
}
//再遍历找最大或没有被访问到的
int result = 0;
for(int i=0;i<n;i++){
for(int j=0;j<m;j++){
if(input[i][j] == 1 && visit[i][j] == 0)//没有被访问到的
return -1;
if(visit[i][j] > result)
result = visit[i][j];
}
}
return result - 1;
}