[Java] BOJ 20157 화살을 쏘자!

20157번: 화살을 쏘자!

---

테스트 코드

2
2 2
-2 -2
답: 1​

---

정답 코드

• HashMap - key : 각도 / value : 풍선 수
• 사분면으로 나눠서 풀이해야함
• 사분면으로 나누지 않을 경우 테스트 코드에서 걸리는 문제 발생
• x, y가 0일때도 생각해야함

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

public class Main {

    static BufferedReader br;
    static BufferedWriter bw;
    static StringTokenizer st;
    static int N;
    static List<HashMap<Double, Integer>> list;

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

        N = Integer.parseInt(br.readLine());
        list = new ArrayList<>();
        for (int i = 0; i < 4; i++) {
            list.add(new HashMap<>());
        }
        for (int i = 0; i < N; i++) {
            st = new StringTokenizer(br.readLine(), " ");
            double x = Double.parseDouble(st.nextToken());
            double y = Double.parseDouble(st.nextToken());

            double d = x / y;

            int key = -1;
            if (x >= 0 && y > 0) {
                key = 0;
            } else if (x < 0 && y >= 0) {
                key = 1;
            }else if (x <= 0 && y < 0) {
                key = 2;
            } else if (x > 0 && y <= 0) {
                key = 3;
            }
            
            if (list.get(key).containsKey(d)) {
                list.get(key).put(d, list.get(key).get(d) + 1);
            } else {
                list.get(key).put(d, 1);
            }
        }

        int max = 0;
        for (int i = 0; i < 4; i++) {
            for (double d : list.get(i).keySet()) {
                max = Math.max(max, list.get(i).get(d));
            }            
        }

        System.out.println(max);
    }
}