제곱수 판별하기

#include <stdio.h>
#include <math.h>

int solution(int n) {
    int num = (int)sqrt(n);  
    if (num * num == n) {
        return 1; 
    } 
    else 
        return 2;  
}​

import math

def solution(n):
    num = int(math.sqrt(n))  
    if num * num == n:
        return 1  
    else:
        return 2