2022.3.14阿里笔试第二题

原创已于 2022-03-15 11:02:06 修改 · 1k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#java #算法

于 2022-03-14 20:52:48 首次发布

算法专栏收录该内容

33 篇文章

订阅专栏

这是一个关于矩阵遍历和得分计算的问题，输入为一个n行m列的矩阵，其中1表示人，0表示聚光灯。每个聚光灯可以朝上下左右四个方向照射，如果能照到人，则得分。程序通过遍历矩阵并计算每个聚光灯的得分，最后输出所有聚光灯的总得分。示例给出了一个4*2的矩阵，并输出了相应的得分计算过程。

严正声明：转载请注明出处！！！

题目描述：
小红拿到一个n行m列的矩阵，矩阵种用1表示人，0表示聚光灯，每个聚光灯可以朝着上、下、左、右四个方向照射（照射的距离是无穷大的），若是一个方向上至少有一个人，那么小红就获得了1分。

输入描述：
第一行两个整数n, m（1≤n,m≤10 $^3$ ）
接下来m行，每行m个整数a $_{i,j}$ ，用来表示矩阵中的位置是人还是聚光灯(0≤a $_{i,j}$ ≤1)。

输出描述：
输出一行，一个整数表示这些灯的分数的总和。

测试案例1：

2 4
0 1 0 0
1 0 1 0

输出：

说明：

(1, 1)的灯可以获得2分(向右、向下)
(1, 3)的灯可以获得2分（向左、向下）
(1, 4)的灯可以获得1分（向下）
(2, 2)的灯可以获得3分（向左、向右、向上）
(2, 4)的灯可以获得1分（向左）
总和9分。

这是其中一个写法，欢迎有更好的补充，大家互相学习。

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int m = sc.nextInt();
        int[][] arr = new int[n][m];
        for(int i = 0;i < n;i++){
            for(int j = 0;j < m;j++){
                arr[i][j] = sc.nextInt();
            }
        }
        int res = 0;
        for(int i = 0;i < n;i++){
            for(int j = 0;j < m;j++){
                if(arr[i][j] == 0){
                    continue;
                }else{
                    res += count(arr, i, j);
                }
            }
        }
        System.out.println(res);
    }

    public static  int count(int[][] arr, int i, int j){
        int count = 0;
        for(int x = j - 1;x >= 0;x--){
            if(arr[i][x] == 1){
                break;
            }
            count++;
        }
        for(int x = j + 1;x < arr[0].length;x++){
            if(arr[i][x] == 1){
                break;
            }
            count++;
        }

        for(int y = i - 1;y >= 0;y--){
            if(arr[y][j] == 1){
                break;
            }
            count++;
        }

        for(int y = i + 1;y < arr.length;y++){
            if(arr[y][j] == 1){
                break;
            }
            count++;
        }
        return count;
    }
}