Leetcode - Bomb Enemy

My code:

public class Solution {
    public int maxKilledEnemies(char[][] grid) {
        if (grid == null || grid.length == 0 || grid[0].length == 0) {
            return 0;
        }
        
        int max = 0;
        int row = 0;
        int[] cols = new int[grid[0].length];
        
        for (int i = 0; i < grid.length; i++) {
            for (int j = 0; j < grid[0].length; j++) {
                if (j == 0 || grid[i][j - 1] == 'W') {
                    row = updateRow(i, j, grid);
                }
                if (i == 0 || grid[i - 1][j] == 'W') {
                    cols[j] = updateCol(i, j, grid);
                }
                if (grid[i][j] == '0') {
                    max = Math.max(max, row + cols[j]);
                }
            }
        }
        return max;
    }
    
    private int updateRow(int i, int j, char[][] grid) {
        int counter = 0;
        for (int k = j; k < grid[0].length; k++) {
            if (grid[i][k] == 'W') {
                break;
            }
            else if (grid[i][k] == 'E') {
                counter++;
            }
        }
        return counter;
    }
    
    private int updateCol(int i, int j, char[][] grid) {
        int counter = 0;
        for (int k = i; k < grid.length; k++) {
            if (grid[k][j] == 'W') {
                break;
            }
            else if (grid[k][j] == 'E') {
                counter++;
            }
        }
        return counter;
    }
}

reference:
https://discuss.leetcode.com/topic/48742/simple-dp-solution-in-java/2

这道题目我自己也做了出来。
首先是 brute-force,
然后思考,哪里可以加cache
发现,从列角度看,如果左侧是'0',那么,当前这个点,这一行的炸死敌人数,就等于左边这一点,这一行的数。
同理,从行角度看,如果上侧是'0',那么,当前这个点,这一列的炸死敌人数,就等于上边这一点,这一列的数。
用这个原理加cache
因为每个点都有 行炸死个数,列炸死个数。
我搞了两个二维数组作为cache
int[][] rows
int[][] cols

其实原理和上面的解法原理是相同的。只是cache规模太大,浪费了很多时间。
我的解法如下:

My code:

public class Solution {
    public int maxKilledEnemies(char[][] grid) {
        if (grid == null || grid.length == 0 || grid[0].length == 0) {
            return 0;
        }
        
        int max = 0;
        int[][] cols = new int[grid.length][grid[0].length];
        int[][] rows = new int[grid.length][grid[0].length];
        for (int i = 0; i < grid.length; i++) {
            for (int j = 0; j < grid[0].length; j++) {
                if (grid[i][j] == '0') {
                    max = Math.max(max, place(i, j, grid, rows, cols));
                }
            }
        }
        return max;
    }
    
    private int place(int i, int j, char[][] grid, int[][] rows, int[][] cols) {
        int row = 0;
        int col = 0;
        boolean colHit = false;
        boolean rowHit = false;
        if (i > 0 && grid[i - 1][j] == '0') {
            col += cols[i - 1][j];
            colHit = true;
        }
        if (j > 0 && grid[i][j - 1] == '0') {
            row += rows[i][j - 1];
            rowHit = true;
        }
        
        if (!colHit) {
            // up
            for (int k = i - 1; k >= 0; k--) {
                if (grid[k][j] == 'W') {
                    break;
                }
                else if (grid[k][j] == 'E') {
                    col++;
                }
            }
            // down
            for (int k = i + 1; k < grid.length; k++) {
                if (grid[k][j] == 'W') {
                    break;
                }
                else if (grid[k][j] == 'E') {
                    col++;
                }
            }
        }
        if (!rowHit) {
            // left
            for (int k = j - 1; k >= 0; k--) {
                if (grid[i][k] == 'W') {
                    break;
                }
                else if (grid[i][k] == 'E') {
                    row++;
                }
            }
            // right
            for (int k = j + 1; k < grid[0].length; k++) {
                if (grid[i][k] == 'W') {
                    break;
                }
                else if (grid[i][k] == 'E') {
                    row++;
                }
            }
        }
        
        cols[i][j] = col;
        rows[i][j] = row;
        return col + row;
    }
}

Anyway, Good luck, Richardo! -- 09/20/2016

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 背景 一年多以前我在知乎上答了有关LeetCode的问题, 分享了一些自己做题目的经验。 张土汪:刷leetcod...
    土汪阅读 12,775评论 0 33
  • 转至元数据结尾创建: 董潇伟,最新修改于: 十二月 23, 2016 转至元数据起始第一章:isa和Class一....
    40c0490e5268阅读 1,776评论 0 9
  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 134,973评论 19 139
  • My code: 这道题木我没有做出来。看了答案才想出来。一开始我写了我的解法,就是看右边的值是否比自己小。小的化...
    Richardo92阅读 730评论 0 1
  • title: TED:孩子们只会跟喜欢的人学习date: 20160404听周围的同事朋友说现在的一些老师,没有耐...
    kevinHan阅读 405评论 0 0