c++T337181 刷墙

Question

题目

我的代码TLE了

#include <iostream>
#include <vector>

using namespace std;

int main() {
    int n, m, k;
    cin >> n >> m >> k;
    
    vector<vector<int>> wall(n, vector<int>(m, 0));
    
    for (int i = 0; i < k; i++) {
        int x, y, z;
        cin >> x >> y >> z;
        
        if (x == 0) {
            for (int j = 0; j < m; j++) {
                wall[y - 1][j] = z;
            }
        } else {
            for (int j = 0; j < n; j++) {
                wall[j][y - 1] = z;
            }
        }
    }
    
    int blueCount = 0;
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            if (wall[i][j] == 1) {
                blueCount++;
            }
        }
    }
    
    cout << blueCount << endl;
    
    return 0;
}

Answer 1

让阿豪来帮你解答，本回答参考chatgpt3.5编写提供，如果还有疑问可以评论或留言

问题：如何优化这个代码来避免超时？ 回答： 这个代码的问题在于，它使用了两次嵌套的循环来遍历整个矩阵，这会导致时间复杂度为O(n^2)，而在本题中，n和m最大可以达到1000，所以这种算法是不可行的。 解决这个问题的方法是，我们可以使用一个大小为1001的数组来记录每一行/列最高的墙的高度。这样，在每个位置上，我们只需要检查它所在的行/列上的最高墙的高度，并判断是否和当前墙的高度相等即可。这样时间复杂度就降为了O(k+max(n,m))，可以通过本题。 下面给出优化后的代码： #include #include using namespace std; int main() { int n, m, k; cin >> n >> m >> k; vector row(n+1, 0); // 记录每一行的最高墙高度 vector col(m+1, 0); // 记录每一列的最高墙高度 for (int i = 0; i < k; i++) { int x, y, z; cin >> x >> y >> z; if (x == 0) { for (int j = 1; j <= m; j++) { row[y] = max(row[y], z); } } else { for (int j = 1; j <= n; j++) { col[y] = max(col[y], z); } } } int blueCount = 0; for (int i = 1; i <= n; i++) { for (int j = 1; j <= m; j++) { if (row[i] == col[j] && row[i] != 0) { // 检查所在行/列的最高墙高度是否相等 blueCount++; } } } cout << blueCount << endl; return 0; }