Q 16283 Farm

💡 문제 요약 및 분석

하루에 양 한 마리가 먹는 사료 a, 염소 한 마리가 먹는 사료 b, 전체 n 마리, 소비한 전체 사료 w 그램이 주어지면 양과 염소가 몇 마리인지 구하는 문제. 단순한 수학 문제인 것으로 보인다.

💡 알고리즘 설계

생각해낸 방법은 무식하게 더해보는 방법이다. a 와 b 를 n 번 더한 값이 w 이고 a, b 를 몇 번 더했느냐가 문제의 답이 될 것이다.

예를 들어 n = 9 인 경우,
(a * 9) + (b * 0);
(a * 8) + (b * 1);
(a * 7) + (b * 2);
...

💡 코드

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);

        int a = sc.nextInt();
        int b = sc.nextInt();
        int n = sc.nextInt();
        int w = sc.nextInt();

        int sheep = 0;
        int goat = 0;

        // a,b 값이 같을 경우를 먼저 체크한다.
        if (a == b) {
            // a,b 값이 같으면서 전체 마리 수가 3마리 이상이면 가능한 해가 두 개 이상이므로 -1 을 출력한다.
            if (n == 2) {
                System.out.println(1 + " " + 1);
            } else {
                System.out.println(-1);
            }
        } else if (a != b) {
            int count = 0; // 가능한 해가 몇 개 있는지 체크하는 변수
            for (int i = 0; i <= n; i++) {
                if (a * (n-i) + b * i == w) {
                    sheep = n - i;
                    goat = i;
                    count++;
                }
            }
            // 가능한 해가 2개 이상이면 -1 을 출력한다.
            if (count > 1) {
                System.out.println(-1);
            }
            // 양 혹은 염소가 0마리인 경우는 -1 을 출력한다.
            if (sheep == 0 || goat == 0) {
                System.out.println(-1);
            } else {
                System.out.println(sheep + " " + goat);
            }
        }

    }
}

💡 틀린 부분 수정

접근은 좋았으나 과도한 조건문 사용이 실패의 원인이 되었다. for 문으로 모든 경우의 수를 체크하므로 a,b 가 동일한지 여부는 체크할 필요가 없었다. 또한 기존에는 정답이 아닌 경우에 조건문을 걸어서 -1 을 출력하도록 만들었는데, 생각보다 오답 조건이 많았고 이 부분에서 놓치는 부분이 발생하여 예상치 못하게 출력이 두 번 이루어지는 등 문제가 있었던 듯 하다. 조금 더 단순하게 정답인 조건만 조건문을 걸어 출력하게 바꿔서 정답 코드를 작성했다.

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);

        int a = sc.nextInt();
        int b = sc.nextInt();
        int n = sc.nextInt();
        int w = sc.nextInt();

        int sheep = 0;
        int goat = 0;

        int count = 0; // 가능한 해가 몇 개 있는지 체크하는 변수
        for (int i = 1; i < n; i++) {
            if (a * (n-i) + b * i == w) {
                sheep = n - i;
                goat = i;
                count++;
            }
        }
        if (count == 1) {
            System.out.println(sheep + " " + goat);
        } else {
            System.out.println(-1);
        }
    }
}

💡 시간 복잡도, 공간 복잡도

시간 복잡도	공간 복잡도
O(n)	O(1)

💡 다른 풀이

💡 느낀점 및 기억할 정보

• 과도한 반복문, 조건문에 주의하자. 생각나는대로 코딩하다보니 복잡해지는 것이 나의 안좋은 습관인 듯 하다 ㅜㅜ 최대한 단순하게 작성해야 실패할 확률이 줄어든다.