553 · 炸弹袭击

最新推荐文章于 2024-11-15 18:38:47 发布

Pioneer.C

最新推荐文章于 2024-11-15 18:38:47 发布

阅读量104

点赞数

分类专栏：动态规划专项训练文章标签： java 算法 csp

本文链接：https://blog.csdn.net/weixin_45845865/article/details/121545735

版权

动态规划专项训练专栏收录该内容

12 篇文章 0 订阅

订阅专栏

描述

给定一个二维矩阵, 每一个格子可能是一堵墙 W,或者一个敌人 E 或者空 0 (数字 '0'), 返回你可以用一个炸弹杀死的最大敌人数. 炸弹会杀死所有在同一行和同一列没有墙阻隔的敌人。由于墙比较坚固，所以墙不会被摧毁.

你只能在空的地方放置炸弹.

样例

样例1

输入:
grid =[
     "0E00",
     "E0WE",
     "0E00"
]
输出: 3
解释:
把炸弹放在 (1,1) 能杀3个敌人

样例2

输入:
grid =[
     "0E00",
     "EEWE",
     "0E00"
]
输出: 2
解释:
P把炸弹放在 (0,0) 或 (0,3) 或 (2,0) 或 (2,3) 能杀2个敌人

思路：通过4种方向的动态规划，解决问题

递推公式：

1.当前位置是wall，dp[i][j] = 0;

2.当前位置是enemy，dp[i][j]=前一种状态+1;

3.当前位置是0，dp[i][j] = 前一种状态;

public class Solution {
    /**
     * @param grid: Given a 2D grid, each cell is either 'W', 'E' or '0'
     * @return: an integer, the maximum enemies you can kill using one bomb
     */
    public int maxKilledEnemies(char[][] grid) {
        // write your code here
        if(grid.length==0) return 0;
        int m = grid.length;
        int n = grid[0].length;
        int[][] up = new int[m][n];
        int[][] down = new int[m][n];
        int[][] left = new int[m][n];
        int[][] right = new int[m][n];
        for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                int a = m-i-1;
                int b = n-j-1;
                if(i==0)
                {
                    if(grid[i][j]=='E') up[i][j] = 1;
                    else up[i][j] = 0;
                }
                else{
                    if(grid[i][j]=='E') up[i][j] = up[i-1][j]+1;
                    else if(grid[i][j]=='W') up[i][j] =0;
                    else up[i][j] = up[i-1][j]; 
                }

                if(i==0)
                {
                    if(grid[a][j]=='E') down[a][j] = 1;
                    else down[a][j] = 0;
                }
                else{
                    if(grid[a][j]=='E') down[a][j] = down[a+1][j]+1;
                    else if(grid[a][j]=='W') down[a][j] =0;
                    else down[a][j] = down[a+1][j]; 
                }

                if(j==0)
                {
                    if(grid[i][j]=='E') left[i][j] = 1;
                    else left[i][j] = 0;
                }
                else{
                    if(grid[i][j]=='E') left[i][j] = left[i][j-1]+1;
                    else if(grid[i][j]=='W') left[i][j] =0;
                    else left[i][j] = left[i][j-1]; 
                }

                if(j==0)
                {
                    if(grid[i][b]=='E') right[i][b] = 1;
                    else right[i][b] = 0;
                }
                else{
                    if(grid[i][b]=='E') right[i][b] = right[i][b+1]+1;
                    else if(grid[i][b]=='W') right[i][b] =0;
                    else right[i][b] = right[i][b+1]; 
                }
            

            }
        }
        int x=-1,y=-1;
        int[][] arr = new int[m][n];
        for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                arr[i][j] = up[i][j] +down[i][j] +right[i][j]+left[i][j]; 
                //System.out.println(up[i][j] +" "+down[i][j]+" "+right[i][j]+" "+left[i][j]);
            }
        }

        for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                if(grid[i][j]=='0'){
                    if(x==-1){
                        x=i;
                        y=j;
                    }else if(arr[x][y]<arr[i][j]){
                        x = i;
                        y=j;
                    }
                }
                
            }
        }
        if(x==-1) return 0;
        return arr[x][y];
    }
}