Bomb Enemy -- LeetCode

Given a 2D grid, each cell is either a wall 'W', an enemy 'E' or empty '0' (the number zero), return the maximum enemies you can kill using one bomb.
The bomb kills all the enemies in the same row and column from the planted point until it hits the wall since the wall is too strong to be destroyed.
Note that you can only put the bomb at an empty cell.

Example:

For the given grid

0 E 0 0
E 0 W E
0 E 0 0

return 3. (Placing a bomb at (1,1) kills 3 enemies)

思路：扫描矩阵，用rowCount表示该点所在行内有多少敌人可见。用colCount[i]表示该点所在列i内有多少敌人可见。然后该点可见的敌人为两者之和。

初始时两值为0.

当我们位于行首或者该行上一格为墙，那么在该行向右扫描直到行尾或者遇见一堵墙，更新rowCount。

当我们位于列首或者该列上一格为墙，那么在该列向下扫描直到列尾或者遇见一堵墙，更新colCount[i]。

时间复杂度O(mn)，空间复杂度O(N)。

class Solution {
public:
    int maxKilledEnemies(vector<vector<char>>& grid) {
        if (grid.size() == 0) return 0;
        int rowCount = 0, height = grid.size(), width = grid[0].size(), res = 0;
        vector<int> colCount(width, 0);
        for (int i = 0; i < height; i++) {
            for (int j = 0; j < width; j++) {
                if (!j || grid[i][j-1] == 'W') {
                    rowCount = 0;
                    for (int k = j; k < width && grid[i][k] != 'W'; k++)
                        rowCount += grid[i][k] == 'E';
                }
                if (!i || grid[i-1][j] == 'W') {
                    colCount[j] = 0;
                    for (int k = i; k < height && grid[k][j] != 'W'; k++)
                        colCount[j] += grid[k][j] == 'E';
                }
                if (grid[i][j] == '0')
                    res = std::max(res, rowCount + colCount[j]);
            }
        }
        return res;
    }
};