Code & Func

第52天，有点浪的一天。survey一点没动的我。

水题，只要遍历一次计算每一行和每一列之和，然后再遍历判断是否为1，且所在行或列不止一个节点即可。

1
int countServers(vector<vector<int>>& grid) {
2
    int m = grid.size();
3
    if (m == 0) return 0;
4
    int n = grid[0].size();
5
    if (n == 0) return 0;
6
    vector<int> a(m, 0);
7
    vector<int> b(n, 0);
8

9
    for(int i = 0;i < m; i++) {
10
        for(int j = 0;j < n; j++) {
11
            if (grid[i][j]) {
12
                a[i]++;
13
                b[j]++;
14
            }
15
        }
11 collapsed lines
16
    }
17
    int res = 0;
18
    for(int i = 0;i < m; i++) {
19
        for(int j = 0;j < n; j++) {
20
            if (grid[i][j] && (a[i] > 1 || b[j] > 1))
21
                res++;
22
        }
23
    }
24
    return res;
25

26
}

还看到一种做法，先遍历一遍，每一行都记录下值为1的grid的个数，如果个数大于一，则表示这些点都是能通信的点，如果等于一，则将该点位置记录下来。

然后将对所有记录下来的点判断一次是否其所在列的点的个数大于2。

1
int countServers(vector<vector<int>>& grid) {
2
    int m = grid.size();
3
    if (m == 0) return 0;
4
    int n = grid[0].size();
5
    if (n == 0) return 0;
6
    vector<int> vec;
7
    int res = 0;
8
    for(int i = 0;i < m; i++) {
9
        vector<int> temp;
10
        for(int j = 0;j < n; j++) {
11
            if (grid[i][j]) temp.push_back(j);
12
        }
13

14
        if (temp.size() > 1) res += temp.size();
15
        else if (temp.size() == 1) {
17 collapsed lines
16
            vec.push_back(temp[0]);
17
        }
18
    }
19

20
    for(auto &j: vec) {
21
        int count = 0;
22
        for(int i = 0;i < m; i++) {
23
            if (grid[i][j]) count++;
24
            if (count > 1) {
25
                res += 1;
26
                break;
27
            }
28
        }
29
    }
30

31
    return res;
32
}