聚光灯矩阵

一个n行m列，矩阵中1表示人，0表示灯光，每个灯可以朝上下左右四个方向照射，距离为无限大，若某个方向有人，则积一分。
问，矩阵中所有的灯可以获得多少分

int a[1000][1000], t, row[1000][1000], col[1000][1000]; //写函数外面，数组有点大
int m, n;
    cin >> n >> m;

    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            cin >> t;
            a[i][j] = t;
        }
    }
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            row[i][j] = a[i][j];
            col[i][j] = a[i][j];
        }
    }
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            if (j > 0)
                row[i][j] += row[i][j - 1];
            if (i > 0)
                col[i][j] += col[i - 1][j];
        }
    }
    int s = 0;
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            if (a[i][j] == 0) {
                if (j>0&&row[i][j - 1] > 0)//左边
                    s++;
                if (row[i][m - 1] - row[i][j] > 0)//右边
                    s++;
                if (i>0&&col[i - 1][j] > 0)//上面
                    s++;
                if (col[n - 1][j] - col[i][j] > 0)//下面
                    s++;
            }
        }
    }
    cout << s;

思路，
row[i][j]记录矩阵a里面第i行中，从0到j元素之和
col[i][j]记录矩阵a里面第j列中，从0到i元素之和