[LeetCode] 1260번 문제를 풀어보았다. (ft. java)

2022년 04월 11일 월요일 - 왜 아직도 월요일..??

오늘 올려볼 문제는 1260번 Shift 2D Grid 이라는 문제이다.

사진을 클릭하면 해당 문제로 이동합니다.

오늘도 LeetCode 사이트 오늘의 문제를 가지고 왔다.

오늘은 급한 일 처리하느라 좀 늦게 풀어따아....

입력

사진에서도 볼 수 있듯이 2차원 int 배열 1개와 int 값 1개가 입력으로 들어온다.

---

풀이 및 코드

2차원 배열을 k만큼 밀어낸 2차원 배열을 리턴하는 문제다.

오늘은 처음부터 정답을 생각해냈다.

2차원 배열을 1차원으로 봐서 문제를 풀었다.

이제 코드를 봐보자!

풀이코드

class Solution {
    public List<List<Integer>> shiftGrid(int[][] grid, int k) {
        List<List<Integer>> result = new ArrayList<>();
        for(int i = 0; i < grid.length; i++) result.add(new ArrayList<>());

        int index = 0;

        k %= grid.length * grid[0].length;

        for(int i = grid.length * grid[0].length - k; i < grid.length * grid[0].length * 2 - k; i++)
        {
            int temp = i % (grid.length * grid[0].length);
            int y = temp / grid[0].length, x = temp % grid[0].length;

            if(result.get(index).size() == grid[0].length) index++;

            result.get(index).add(grid[y][x]);
        }

        return result;
    }
}

---

제출 화면

오늘은 문제가 쉬웠지만 재밌었다.

내일 문제도 재밌었으면 좋겠다.

내일도 문제를 풀어서 블로그에 글을 쓸 수 있으면 좋겠다.