🎲 백준 1915번 가장 큰 정사각형

백준 1915번

🧸 푸는데 시간이 꽤 걸렸지만 아이디어를 스스로 생각해내서 뿌듯하다. left,right,diagonal에 dp가 아닌 arr값을 넣어 헤맸는데 반례를 찾으면서 잘못된걸 깨달았다.
💡 flat() 이라는 메서드를 알았다. 이차원 배열에서 최댓값을 구할 때 유용할게 쓸 것 같다.
💊 다른 분들의 코드를 보니 "left === up && up === diagonal" 조건 없이 "dp[n][m] = (Math.sqrt(Math.min(left, up, diagonal))+1) ** 2" 만 쓰면 되는 것을 깨달았다. 그리고 dp에 넓이를 직접 넣지 않고 마지막에 제곱해주면 더 간단하게 쓸 수 있다.

코드

const fs = require('fs'); 
const input = fs.readFileSync('/dev/stdin').toString().trim().split('\n');

const [N, M] = input.shift().split(" ").map(Number);
const arr = input.map((item) => item.split("").map(Number));
const dp = Array.from(new Array(N), () => new Array(M));

for (let n = 0; n < N; n++) {
  for (let m = 0; m < M; m++) {
    if (arr[n][m] === 1) {
      let left = 0;
      let up = 0;
      let diagonal = 0;
      //배열 범위를 벗어나지 않으면
      if (m - 1 >= 0 && n - 1 >= 0) {
        left = dp[n][m - 1];
        up = dp[n - 1][m];
        diagonal = dp[n - 1][m - 1];
      }

      if (left === up && up === diagonal) {
        dp[n][m] = (Math.sqrt(left) + 1) ** 2;
      } else {
        dp[n][m] = (Math.sqrt(Math.min(left, up, diagonal))+1) ** 2;
      }
    } else {
      dp[n][m] = 0;
    }
  }
}

console.log(Math.max(...dp.flat()));