鹰角网络2021/8/19笔试 最后一题

import java.util.*;


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

    int count = 0;
    public int getSurrend(int[][] maze, int x, int y, int flag) {
        int[][] dir = {{-1, 0}, {1 , 0}, {0, -1},{0, 1}};
        int result = 0;
        for(int i = 0; i < dir.length; ++i) {
            int xt = x + dir[i][0];
            int yt= y + dir[i][1];
            if( xt< 0 || yt < 0 || xt >= maze.length || yt >= maze[0].length) {
                continue;
            }
            if(maze[xt][yt] == 1 - flag) {
                maze[xt][yt] = -1;
                result += getSurrend(maze, xt, yt, 1 - flag);
            }

        }
        if(flag == 0) {
            result += 1;
        }
        if(flag == 1) {
            count ++;
        }
        return result;
    }


    public long pathOfZeroAndOne (int[][] maze) {
        // write code here

        int result = 0;
        for(int i = 0; i < maze.length; ++i) {
            for (int j = 0; j < maze[0].length; ++j) {
                if(maze[i][j] == 1) {
                    count = 0;
                    maze[i][j] = -1;
                    int temp = getSurrend(maze, i, j, 1);
                    result += count * temp;
                }
            }
        }
        return result;

    }

}