const inputLines = require("fs").readFileSync("/dev/stdin").toString().trim().split("\n"); const [rowCount, colCount, sharkCount] = inputLines[0].split(" ").map(Number); class Shark { constructor(speed, direction, size) { this.speed = speed; this.direction = direction; this.size = size; } } const deltaRow = [-1, 1, 0, 0]; const deltaCol = [0, 0, 1, -1]; const sharkGrid = new Array(rowCount + 1); const nextSharkGrid = new Array(rowCount + 1); for (let row = 0; row <= rowCount; row++) { sharkGrid[row] = new Array(colCount + 1); nextSharkGrid[row] = new Array(colCount + 1); for (let col = 0; col <= colCount; col++) { sharkGrid[row][col] = []; nextSharkGrid[row][col] = []; } } function isValidPosition(r, c) { return 1 <= r && r <= rowCount && 1 <= c && c <= colCount; } function simulateSharkMovement() { for (let row = 1; row <= rowCount; row++) { for (let col = 1; col <= colCount; col++) { for (const shark of sharkGrid[row][col]) { let currentRow = row; let currentCol = col; let effectiveSpeed = shark.speed; while (effectiveSpeed > 0) { if (!isValidPosition(currentRow + deltaRow[shark.direction], currentCol + deltaCol[shark.direction])) { shark.direction = shark.direction ^ 1; } currentRow += deltaRow[shark.direction]; currentCol += deltaCol[shark.direction]; effectiveSpeed--; } nextSharkGrid[currentRow][currentCol].push(shark); } } } for (let row = 1; row <= rowCount; row++) { for (let col = 1; col <= colCount; col++) { if (nextSharkGrid[row][col].length > 1) { nextSharkGrid[row][col].sort((firstShark, secondShark) => secondShark.size - firstShark.size); nextSharkGrid[row][col] = [nextSharkGrid[row][col][0]]; } sharkGrid[row][col] = nextSharkGrid[row][col]; nextSharkGrid[row][col] = []; } } } for (let i = 1; i <= sharkCount; i++) { const [inputRow, inputCol, speed, direction, size] = inputLines[i] .split(" ") .map(Number); const cycle = direction < 2 ? 2 * rowCount - 2 : 2 * colCount - 2; sharkGrid[inputRow][inputCol].push(new Shark(speed % cycle, direction - 1, size)); } let totalCaughtSize = 0; for (let currentCol = 1; currentCol <= colCount; currentCol++) { for (let row = 1; row <= rowCount; row++) { if (sharkGrid[row][currentCol].length > 0) { totalCaughtSize += sharkGrid[row][currentCol][0].size; sharkGrid[row][currentCol] = []; break; } } simulateSharkMovement(); } console.log(totalCaughtSize);

제출 #19702

제출 정보

제출 결과

제출 코드

찾은 반례

입력:

오답 출력:

정답 출력:

입력:

오답 출력:

정답 출력:

입력:

오답 출력:

정답 출력: