모든 테스트 케이스랑 반례 찾아서 넣어도 맞게 나옵니다

작성한 질문수 35

1

안녕하세요 가인님 ㅎㅎ

코드를 보시면 거리를 기반으로 판단하게 되는데요. 만약 M이 1일 때 저 로직은 깨지게 됩니다.

isLeft 변수가 “어느 쪽으로 이동 거리가 더 짧은지”만 알려줄 뿐, 사과가 바구니의 왼쪽에 떨어졌는지 오른쪽에 떨어졌는지를 직접 알려주진 않습니다.

가인님 코드를 보시면

• leftLength = abs(l - apple) 은 사과와 바구니 왼쪽 끝 사이의 거리

• rightLength = abs(r - apple) 은 사과와 오른쪽 끝 사이의 거리

을 상징하고 두 값 중 작은 쪽이 “짧은 이동”이긴 하지만,

• 사과가 이미 바구니 범위 (l ≤ apple ≤ r) 안에 있다면 이동이 전혀 필요 없고,

• 사과가 바구니의 왼쪽 밖에(apple < l) 있으면 반드시 왼쪽으로 가야 하며,

• 사과가 오른쪽 밖에(apple > r) 있으면 반드시 오른쪽으로 가야 합니다.

따라서 만약 abs를 쓸거면 다음과 같이 써야 합니다.

#include <iostream>
#include <cmath>
using namespace std;

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int N, M, J;
    cin >> N >> M >> J;

    int l = 1;
    int r = M;
    long long cnt = 0;

    for (int i = 0; i < J; i++) {
        int apple;
        cin >> apple;

        int leftLength  = abs(l - apple);
        int rightLength = abs(r - apple);
        bool isLeft     = leftLength < rightLength;

        if (apple < l || apple > r) {
            int diff = isLeft ? leftLength : rightLength;

            if (apple < l) {
                diff = leftLength;
                isLeft = true;
            } else if (apple > r) {
                diff = rightLength;
                isLeft = false;
            }

            cnt += diff;
            if (isLeft) {
                l -= diff;
                r -= diff;
            } else {
                l += diff;
                r += diff;
            }
        }
    }

    cout << cnt;
    return 0;
}

후... 여러번 고민하다 알아냈네요 ㅎㅎ

또 질문 있으시면 언제든지 질문 부탁드립니다.

좋은 수강평과 별점 5점은 제게 큰 힘이 됩니다. :)

감사합니다.

강사 큰돌 올림.