Node.js) 백준 28438번: 행렬 연산 (행렬 계산하기)

https://www.acmicpc.net/problem/28438

28438번: 행렬 연산 (행렬 계산하기)

const readline = require('readline').createInterface({
	input: process.stdin,
	output: process.stdout,
});

let flag = false;
let matrix;
const mapR = new Map();
const mapC = new Map();
readline.on('line', function (line) {
	if (flag) {
		const command = line.split(' ').map(Number);
		if (command[0] == 1) {
			const [_, r, v] = command;
			if (mapR.has(r - 1)) {
				const cnt = mapR.get(r - 1);
				mapR.set(r - 1, cnt + v);
			} else {
				mapR.set(r - 1, v);
			}
		} else {
			const [_, c, v] = command;
			if (mapC.has(c - 1)) {
				const cnt = mapC.get(c - 1);
				mapC.set(c - 1, cnt + v);
			} else {
				mapC.set(c - 1, v);
			}
		}
	} else {
		flag = true;
		const [N, M, Q] = line.split(' ').map(Number);
		matrix = Array.from(Array(N), () => Array(M).fill(0));
	}
}).on('close', function () {
	mapR.forEach((v, k) => {
		for (let c = 0; c < matrix[0].length; c++) {
			matrix[k][c] += v;
		}
	});

	mapC.forEach((v, k) => {
		for (let r = 0; r < matrix.length; r++) {
			matrix[r][k] += v;
		}
	});

	console.log(matrix.map((v) => v.join(' ')).join('\n'));
	process.exit();
});