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

2022년 03월 25일 금요일 - 오늘도 잃었다아아 흐규ㅠㅠ

오늘 올려볼 문제는 1029번 Two City Scheduling 이라는 문제이다.

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

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

오늘 좀 길게 풀었다가 힌트 받아서 짧게 풀음

입력

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

---

풀이 및 코드

문제 설명이 어렵다... 사진 참고해주길 바란다.

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

두 수의 차이를 기준으로 정렬후에 그 기준으로 그리디를 진행하는식으로 문제를 풀었다.

이제 코드를 봐보자!

풀이코드

class Solution {
    public int twoCitySchedCost(int[][] costs) {
        int n = costs.length / 2;
        int result = 0;

        Arrays.sort(costs, (o1, o2) ->{return (o1[0] - o1[1]) - (o2[0] - o2[1]);});

        for(int i = 0; i < costs.length; i++) result += costs[i][i / n];

        return result;
    }
}

---

제출 화면

오늘은 문제를 풀긴 했지만 좀 안예쁘게 풀었다가 힌트를 받아서 짧게 풀게되었다.

내일은 한 번에 예쁘게 풀 수 있었으면 좋겠다.

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