본문 바로가기
코딩테스트 준비/백준

[백준] 10158번 - 개미 [Java]

by mwzz6 2024. 12. 6.

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

 

[백준] 10158번 - 개미 [Java]
[백준] 10158번 - 개미 [Java]


1.  아이디어

 

개미의 최종 위치를 x좌표와 y좌표를 분리해서 계산해도 똑같다는 점을 활용한다.


2. 문제풀이

 

개미의 x좌표는 가로 길이의 2배만큼 이동했을 때, y좌표는 세로 길이의 2배만큼 이동했을 때 각각 제자리로 돌아온다는 점을 캐치하면 간단하게 해결할 수 있다.

 

제자리로 돌아오는 것은 (w * 2), (h * 2)로 나누었을 때 몫이므로 최종 위치는 나머지로 구할 수 있다. 이때 나머지가 w, h보다 크면 벽에 한번 더 부딫히는 것까지 고려하면 이를 식으로 세울 수 있어서 삼항 연산자를 활용해서 구현했다.


3. 코드

 

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));
        StringBuilder sb = new StringBuilder();
        StringTokenizer st;

        st = new StringTokenizer(br.readLine());
        int w = Integer.parseInt(st.nextToken());
        int h = Integer.parseInt(st.nextToken());

        st = new StringTokenizer(br.readLine());
        int p = Integer.parseInt(st.nextToken());
        int q = Integer.parseInt(st.nextToken());

        int t = Integer.parseInt(br.readLine());

        int x = (p + t) % (w * 2) < w ? (p + t) % w : w * 2 - (p + t) % (w * 2);
        int y = (q + t) % (h * 2) < h ? (q + t) % h : h * 2 - (q + t) % (h * 2);

        sb.append(x).append(" ").append(y);

        bw.write(sb.toString());
        bw.flush();
    }
}

4. 후기