聚光灯矩阵
一个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元素之和