알고리즘/프로그래머스 1단계

프로그래머스 - 정수 제곱근 판별 - C++

게임만드는학생 2023. 9. 23. 11:55

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

 

프로그래머스

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

programmers.co.kr

 

#include <string>
#include <vector>
#include <cmath>
using namespace std;

long long solution(long long n) {
    long long answer = 0;
    
    long long a = sqrt(n);
    
    for(long long i = 0; i<=a;i++)
    {
        if(i*i==n)
        {
            return (i+1)*(i+1);
        }
    }
    
    return -1;
}

 

설명 

주어진 n의 양의정수인 제곱근을 찾아서 있다면 제곱근 + 1의 제곱을 리턴하고 양의정수인 제곱근이 없다면 -1을 리턴하는 문제이다.

 

sqrt 함수는 cmath 헤더파일에 포함된 함수로 매개변수의 제곱근을 리턴해준다. 

for문을 a까지 돌면서 i*i가 a이면 제곱근인 것이다.

if가 참이되면 i+1의 제곱을 리턴하여 함수를 종료하고

그렇지 않고 for문이 종료되면 -1을 리턴한다.