题解 | #疯牛病I#

疯牛病I

https://www.nowcoder.com/practice/2066902a557647ce8e85c9d2d5874086

知识点:二维数组,广度优先遍历

遍历二维数组,找出元素值为2的所有位置,将其加入到队列中,每过一分钟,k-1,将队列中的所有元素出队,判断四个方向中是否有值为1的元素,将其更新为2,并入队,重复以上操作,直至k=0,最后,遍历一遍数组,得到所有值为1的元素个数,即为答案。

Java题解如下

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param pasture int整型二维数组 
     * @param k int整型 
     * @return int整型
     */

    
    public int healthyCows (int[][] pasture, int k) {
        // write code here
        int[][] dirs = new int[][]{{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
        int m = pasture.length;
        int n = pasture[0].length;
        Queue<int[]> queue = new ArrayDeque<>();
        for(int i = 0; i < m; i++) {
            for(int j = 0; j < n; j++) {
                if(pasture[i][j] == 2) {
                    queue.offer(new int[]{i, j});
                }
            }
        }
        while(k > 0) {
            k--;
            int size = queue.size();
            for(int i = 0; i < size; i++) {
                int[] poll = queue.poll();
                for(int[] dir: dirs) {
                    int x = poll[0] + dir[0];
                    int y = poll[1] + dir[1];
                    if(x >= 0 && x < m && y >= 0 && y < n && pasture[x][y] == 1) {
                        pasture[x][y] = 2;
                        queue.offer(new int[]{x, y});
                    }
                }
            }
        }
        int count = 0;
        for(int i = 0; i < m; i++) {
            for(int j = 0; j < n; j++) {
                if(pasture[i][j] == 1) {
                    count++;
                }
            }
        }
        return count;
    }
}

全部评论

相关推荐

点赞 评论 收藏
分享
码农索隆:《211》《转正4k》😂,真给我整笑了
点赞 评论 收藏
分享
2025-12-28 16:32
重庆邮电大学 Java
程序员花海:1.技能放最后,来面试默认你都会,技能没啥用 2.实习写的看起来没啥含金量,多读读部门文档,包装下 接LLM这个没含金量 也不要用重构这种 不会给实习生做的 3.抽奖这个还是Demo项目,实际在公司里面要考虑策略,满减,触发点,触发规则 库存 之类的,不是这个项目这么简单 4.教育背景提前,格式为 教育背景 实习 项目 技能 自我评价
简历被挂麻了,求建议
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务