【牛客】美團2024年春招第一場筆試【技術】

【牛客】美團2024年春招第一場筆試【技術】

1. 小美的平衡矩陣

#include <iostream>
#include <vector>
using namespace std;

int main() {
    int n;
    cin >> n;
    vector<vector<int>> nums;
    for (int i = 0; i < n; i++) {
        fgetc(stdin);
        vector<int> nums_;
        for (int j = 0; j < n; j++) {
            int num = fgetc(stdin) - '0';
            nums_.push_back(num);
        }
        nums.push_back(nums_);
    }

    vector<vector<int>> pre(n + 1, vector<int>(n + 1, 0));
    for (int i = 1; i < n + 1; i++) {
        for (int j = 1; j < n + 1; j++) {
            pre[i][j] = pre[i][j - 1] + (pre[i - 1][j] - pre[i - 1][j - 1]) + nums[i - 1][j
                        - 1];
        }
    }

    vector<int> res(n, 0);
    for (int i = 2; i < n + 1; i++) {
        for (int j = 2; j < n + 1; j++) {
            for (int k = 2; k <= min(i, j); k += 2) {
                if (pre[i][j] - pre[i - k][j] - pre[i][j - k] + pre[i - k][j - k] == k * k /
                        2) {
                    res[k - 1] += 1;
                }
            }
        }
    }

    for (int i = 0; i < n; i++) {
        cout << res[i] << endl;
    }
    return 0;
}

思路：

1. 因為矩陣只有0和1，所以可以使用二維字首和，只佔任意一個i*i矩陣，如果總和為i*i/2，則符合要求。
2. 遍歷整個矩陣，作為小矩陣的右下角，遍歷可能的邊長i，判斷是否符合要求。

遇到的問題：

1. 輸入問題：做力扣習慣了，一開始直接使用int接受行，發現會溢位，需要用fgetc()一個一個接受。