变身程序员 BFS 字节跳动笔试

春招实习笔试题,是简单的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;
    }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值