백준 2828 사과 담기 게임(JAVA)

문제 설명

https://www.acmicpc.net/problem/2828

 

풀이과정

 바구니의 범위를 이용해서 풀이했다. 사과가 바구니보다 오른쪽에 있으면 바구니를 오른쪽으로 이동(++)시키고, 왼쪽에 있으면 바구니를 왼쪽으로 이동(--)시킨다.

 

정답코드

import java.io.*;
import java.util.*;

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringTokenizer st = new StringTokenizer(br.readLine());

        int N = Integer.parseInt(st.nextToken());
        int M = Integer.parseInt(st.nextToken());
        int J = Integer.parseInt(br.readLine());

        int result = 0;

        int bucket = 1;

        for (int i = 0; i < J; i++) {
            int drop = Integer.parseInt(br.readLine());

            if (drop >= bucket && drop <= bucket + M - 1) {
                continue;
            }

            if (bucket < drop) {
                while (true) {
                    if (drop >= bucket && drop <= bucket + M - 1) {
                        break;
                    }
                    bucket++;
                    result++;
                }
            } else {
                while (true) {
                    if (drop >= bucket && drop <= bucket + M - 1) {
                        break;
                    }
                    bucket--;
                    result++;
                }
            }

        }

        bw.write(String.valueOf(result));
        bw.flush();
    }
}