2024春招美团前端

编程题

小美的平衡矩阵

思路

通过前缀和计算 0 的个数
然后对应奇数的值是0，因为奇数的平方得来的0和1 不可能相等，我们只看偶数，然后从对应下标 [i][i] 的位置开始看，我们算这块区域的 0 的个数是不是刚好等于偶数平方的一半即可

javascript 复制代码

const rl = require("readline").createInterface({ input: process.stdin });
var iter = rl[Symbol.asyncIterator]();
const readline = async () => (await iter.next()).value;

void (async function () {
    // Write your code here
    // while ((line = await readline())) {
    //     let tokens = line.split(" ");
    //     let a = parseInt(tokens[0]);
    //     let b = parseInt(tokens[1]);
    //     console.log(a + b);
    // }

    let n = parseInt(await readline());

    let zero = [];
    for (let i = 0; i < n; i++) {
        let str = await readline();
        zero.push([]);
        for (let p = 0; p < n; p++) {
            if (str[p] === "0") zero[i][p] = p === 0 ? 1 : zero[i][p - 1] + 1;
            else zero[i][p] = p === 0 ? 0 : zero[i][p - 1];
        }
    }

    // console.log(zero);

    // 奇数不用考虑是0
    for (let i = 0; i < n; i++) {
        let count = 0;
        if ((i + 1) % 2 === 1) {
            console.log(0);
            continue;
        }
        for (let p = i; p < n; p++) {
            for (let q = i; q < n; q++) {
                // 计算1和0是否相等  也就是说要是 i*i/2 的数量
                let sum0 = 0;
                for (let k = 0; k < i + 1; k++) {
                    // console.log("--", zero[p - k][q], zero[p - k][q - k]);
                    sum0 += zero[p - k][q];
                    if (q - i > 0) sum0 -= zero[p - k][q - i - 1];
                    // console.log('i,j***',p-k,q,p-k,q-k-i)
                }
                // console.log('sum0',sum0)
                if (sum0 === ((i + 1) * (i + 1)) / 2) count++;
            }
        }
        console.log(count);
    }

    // 前缀和
})();

小美的数组询问

思路：

题目说的是0的数字会变成别的数字，因此我们先计算出本身数组的值，然后计算0的个数，0的个数乘以 l 就是最小值，乘以r就是最大值

javascript 复制代码

const rl = require("readline").createInterface({ input: process.stdin });
var iter = rl[Symbol.asyncIterator]();
const readline = async () => (await iter.next()).value;

void async function () {
    // Write your code here
    // while(line = await readline()){
    //     let tokens = line.split(' ');
    //     let a = parseInt(tokens[0]);
    //     let b = parseInt(tokens[1]);
    //     console.log(a + b);
    // }

    let n,q,arr;
    line=await readline()
    n=line.split(' ')[0]
    q=line.split(' ')[1]

    line=await readline()
    arr=line.split(' ')
    let arrSum=0,zeroNum=0;
    for(let i=0;i<arr.length;i++){
        arr[i]=parseInt(arr[i])
        arrSum+=arr[i]
        if(arr[i]===0) zeroNum++
    }

    for(let i=0;i<q;i++){
        line=await readline()
        let left=parseInt(line.split(' ')[0]),right=parseInt(line.split(' ')[1])
        // 0 的元素随机取值
        console.log(arrSum+zeroNum*left,arrSum+zeroNum*right)

    }
}()