본문으로 바로가기

[level] 점찍기 JAVA

category 코딩테스트/programmers 2023. 1. 4. 20:45

https://school.programmers.co.kr/learn/courses/30/lessons/140107

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr


문제 설명

좌표평면을 좋아하는 진수는 x축과 y축이 직교하는 2차원 좌표평면에 점을 찍으면서 놀고 있습니다. 진수는 두 양의 정수 k, d가 주어질 때 다음과 같이 점을 찍으려 합니다.

  • 원점(0, 0)으로부터 x축 방향으로 a*k(a = 0, 1, 2, 3 ...), y축 방향으로 b*k(b = 0, 1, 2, 3 ...)만큼 떨어진 위치에 점을 찍습니다.
  • 원점과 거리가 d를 넘는 위치에는 점을 찍지 않습니다.

예를 들어, k가 2, d가 4인 경우에는 (0, 0), (0, 2), (0, 4), (2, 0), (2, 2), (4, 0) 위치에 점을 찍어 총 6개의 점을 찍습니다.

정수 k와 원점과의 거리를 나타내는 정수 d가 주어졌을 때, 점이 총 몇 개 찍히는지 return 하는 solution 함수를 완성하세요.

 


 

내가 생각한 2가지 조건

- 이중 for문을 사용하여 k씩 증가

- 좌표의 거리가 d 보다 작거나 같다  -> if(i*i + j*j <= d*d)

 

1. 첫번째 시도 -> 실패..테스트 케이스까지만 정답

import java.util.*;

class Solution {
    public long solution(int k, int d) {
        long answer = 0;
     
        for(long i = 0; i <= d; i+=k){
            for(long  j = 0; j <= d; j+=k){ 
                if(i*i + j*j <= d*d){
                     answer++;
                } 
            }
        }
        
        return answer;
    }
}

 

 

2. 두번째 시도 -> 성공!!

 - 2중 for문을 사용하니.. 시간 초과를 발생한듯

단일 for문에서 x 좌표의 값을 구한 후 -> y 좌표의 최대값/4

 * long y = (long) Math.sqrt((long) Math.pow(d, 2) - (long) Math.pow(x, 2)) / k;

 * Math.sqrt : 루트함수

 * Math.pow : 제곱함수

import java.util.*;

class Solution {
    public long solution(int k, int d) {
        long answer = 0;
     
        for(long i = 0; i <= d/k; i++){
            long x = i*k;
            long y = (long) Math.sqrt((long) Math.pow(d, 2) - (long) Math.pow(x, 2)) / k;
            answer += y + 1;
        }
        
        return answer;
    }
}