Programmers
[프로그래머스 Level.2] 두 원 사이의 정수 쌍 (연습문제) (Java)
Devtraces
2023. 5. 11. 20:16
문제 링크
https://school.programmers.co.kr/learn/courses/30/lessons/181187#
프로그래머스
코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.
programmers.co.kr
코딩테스트 연습 > 연습문제 > 두 원 사이의 정수 쌍
문제 설명
x축과 y축으로 이루어진 2차원 직교 좌표계에 중심이 원점인 서로 다른 크기의 원이 두 개 주어집니다.
반지름을 나타내는 두 정수 r1, r2가 매개변수로 주어질 때, 두 원 사이의 공간에 x좌표와 y좌표가 모두 정수인 점의 개수를 return하도록 solution 함수를 완성해주세요.
※ 각 원 위의 점도 포함하여 셉니다.
제한 사항
- 1 ≤ r1 < r2 ≤ 1,000,000
입출력 예
r1 | r2 | result |
2 | 3 | 20 |
입출력 예 설명
그림과 같이 정수 쌍으로 이루어진 점은 총 20개 입니다.
나의 코드
class Solution {
public long solution(int r1, int r2) {
long answer = 0;
// 각 원의 넓이
long areaR1 = (long) Math.pow(r1, 2);
long areaR2 = (long) Math.pow(r2, 2);
for(int x=0; x<r2; x++) {
// y = 루트(r^2 - x^2)
// x에 따라 테두리 포함 원의 내부에 있는 가장 큰 정수 y좌표
long maxY2 = (long) Math.sqrt(areaR2 - (long) Math.pow(x, 2));
long maxY1 = (long) Math.sqrt(areaR1 - (long) Math.pow(x, 2));
// 0 <= x < r2 까지 구했으므로 *4 해주면 r2 원의 테두리 + r1 과 r2 사이 내부의 모든 점들이 된다.
answer += (maxY2 - maxY1) * 4;
// r1 테두리 점들을 구해 더해준다. maxY1이 정수이면 작은 원의 테두리의 점이므로 포함시켜야 한다.
// 0 <= x < r1 까지 구했으므로 *4 해주면 r1 원의 테두리의 모든 점들이 된다.
if(x < r1 && Math.sqrt((areaR1 - Math.pow(x, 2))) % 1 == 0) answer += 4;
}
return answer;
}
}
풀이
- r1 원의 테두리에 해당하는 점들과 r1 원과 r2 원의 사이에 해당하는 점과 r2 원의 테두리에 해당하는 점들을 모두 구해야 하는 문제이다.
- x^2 + y^2 = r^2을 이용하여 x를 0부터 r 전까지 순회하면서 각 x마다 가능한 y 좌표의 개수를 구한다.
- 각 x마다 y좌표 최대는 루트(r^2 - x^2) 이고 y좌표 최대를 내림차순하여 정수로 변경하면 가능한 정수 y좌표의 개수가 된다. (수가 클 수 있으므로 int형이 아닌 long형으로 변환하는 것에 주의한다.)
- 예를 들어 r=3일 때 x는 0부터 2까지 순회하며 x=0일 때 y좌표 최대는 3.0, y좌표 최대는 2.828... , x=1일 때 y좌표 최대는 2.236... 이고 따라서 소숫점을 날리면 x=0일 때 가능한 y좌표는 3개, x=1일 때 가능한 y좌표는 2개, x=2일 때 가능한 y좌표는 2개가 된다.
- 이렇게 구한 좌표들을 *4 하면 반지름 r의 모든 점들을 구할 수 있다. (시계 or 반시계 방향으로 한바퀴 돌리면 각 4분면에 모든 좌표들이 찍히는 것과 같다.)
- 이렇게 r2로 가능한 모든 좌표들을 구하여 r1으로 가능한 모든 좌표들을 구해 빼주면 r1의 테두리에 속하는 점들을 제외한 r1 원과 r2 원 사이의 점들과 r2 테두리에 속하는 점들이 된다.
- 따라서 r1 테두리에 속하는 점들을 구하여 더해주면 된다.
- 해당 점들을 구하는 방법도 x를 0부터 r1 전까지 순회하면서 y좌표 최대를 구했을 때 해당 y좌표 최대가 정수인 것들의 개수를 구하여 *4 해서 더해주면 된다.