LeetCode[JS] > 2718. Sum of Matrix After Queries

문제링크

https://leetcode.com/problems/sum-of-matrix-after-queries/description/

문제풀이

이 문제를 아무 생각없이 순서대로 구현하여 답을 도출 시 memory heap size 초과를 맞딱드리게 된다. 2차원 배열을 만들어 해당 배열을 조작하는 대신 나는 Set 객체를 이용하여 연산횟수를 최소화 하였다. 뒤에 있는 쿼리일수록 기존 배열의 값을 override 하기 때문에 맨뒤의 쿼리부터 순회하면서 값을 계산하면 된다. 

/**
 * @param {number} n
 * @param {number[][]} queries
 * @return {number}
 */
var matrixSumQueries = function (n, queries) {
    let sum = 0;

    let removedRow = new Set();
    let removedColumn = new Set();

    for (let i = queries.length - 1; i >= 0; --i) {
        let [type, index, val] = queries[i];

        if (type && !removedColumn.has(index)) {
            // 1 -> column
            removedColumn.add(index);
            sum += (n - removedRow.size) * val;
        } else if (!type && !removedRow.has(index)) {
            // 0 -> row
            removedRow.add(index);
            sum += (n - removedColumn.size) * val;
        }
    }

    return sum;
};