Submission #51604244

---

Source Code Expand

async function main() {
  const [h, w, m] = parseInts(await readOneLine());
  const taxs: [number, number, number][] = [];
  for (const i of range(m)) {
    taxs.push(parseInts(await readOneLine()) as [number, number, number]);
  }

  const hFinalized = new Array(h).fill(false);
  let hFinalizedCount = 0;
  const wFinalized = new Array(w).fill(false);
  let wFinalizedCount = 0;

  const colorCounts = new Map<number, number>();

  for (const i of range(m)) {
    const [h1w2, index, color] = taxs[taxs.length - 1 - i];
    if (h1w2 === 1) {
      if (hFinalized[index - 1]) {
        continue;
      } else {
        hFinalized[index - 1] = true;
        hFinalizedCount++;

        colorCounts.set(color, (colorCounts.get(color) ?? 0) + (w - wFinalizedCount));

        if (hFinalizedCount === h) {
          break;
        }
      }
    } else {
      if (wFinalized[index - 1]) {
        continue;
      } else {
        wFinalized[index - 1] = true;
        wFinalizedCount++;

        colorCounts.set(color, (colorCounts.get(color) ?? 0) + (h - hFinalizedCount));

        if (wFinalizedCount === w) {
          break;
        }
      }
    }
  }

  const notFinalizedCount = hFinalized.filter((i) => !i).length * wFinalized.filter((i) => !i).length;
  if (notFinalizedCount !== 0) {
    colorCounts.set(0, (colorCounts.get(0) ?? 0) + notFinalizedCount);
  }

  const colors = Array.from(colorCounts.keys()).sort((a, b) => a - b);
  console.log(colors.length);
  for (const color of colors) {
    console.log(color, colorCounts.get(color));
  }
}

// --------------------------------------------------------------------------
function sum(array: number[]): number {
  let sum = 0;
  for (const i of array) {
    sum += i;
  }
  return sum;
}

const parseInts = (string: string) => string.split(" ").map((i) => parseInt(i));

function* range(max: number) {
  for (let i = 0; i < max; i++) {
    yield i;
  }
}

// --------------------------------------------------------------------------
function createReadOneLine() {
  const generator = (async function* () {
    const readline = require("readline");
    const rl = readline.createInterface({ input: process.stdin });
    for await (const line of rl) {
      yield line as string;
    }
  })();

  return async function readOneLine() {
    return (await generator.next()).value as string;
  };
}

const readOneLine = createReadOneLine();

main();

Submission Info

Compile Error

Judge Result