LeetCode562. 矩阵中最长的连续1线段 （简单动态规划）

562. 矩阵中最长的连续1线段
简单题。
$f[i][j][k]$多一个状态k用来枚举方向即可。

class Solution {
public:
    vector<vector<int>> vs = {{0,-1},{-1,0},{-1,-1},{-1,1}};
    int longestLine(vector<vector<int>>& mat) {
        int m = mat.size(), n = m > 0 ? mat[0].size() : 0;
        if(m == 0 || n == 0) return 0;
        vector<vector<vector<int>>> f(m,vector<vector<int>>(n,vector<int>(4)));

        int ans = 0;
        for(int i = 0; i < m; i++) {
            for(int j = 0; j < n; j++) {
                if(mat[i][j] == 0) continue;
                int res = 1;
                for(int k = 0; k < 4; k++) {
                    f[i][j][k] = 1;
                }
                // 其实只有两个状态的枚举是有效的
                for(int k = 0; k < 4; k++) {
                    int x = vs[k][0] + i;
                    int y = vs[k][1] + j;
                    if(x >= 0 && y >= 0 && y < n) {
                        f[i][j][k] = max(f[i][j][k],f[x][y][k] + 1);
                        res = max(res,f[i][j][k]);
                    }
                }
                ans = max(ans,res);
            }
        }
        return ans;
    }
};