coding test - python/기본기 문제

문제 / 뒤집은 소수 / Python

sillon 2022. 3. 31. 19:17
728x90
반응형

문제 제목: 뒤집은 소수

N개의 자연수가 입력되면 각 자연수를 뒤집은 후 그 뒤집은 수가 소수이면 그 수를 출력하는 프로그램을 작성하세요.

예를 들어 32를 뒤집으면 23이고, 23은 소수이다. 그러면 23을 출력 한다.

단 910를 뒤집으면 19로 숫자화 해야 한다.

첫 자리부터의 연속된 0은 무시한다.

뒤집는 함수인 def reverse(x) 와 소수인지를 확인하는 함수 def isPrime(x)를 반드시 작성하 여 프로그래밍 한다.

 

▣ 입력설명

 

첫 줄에 자연수의 개수 N(3<=N<=100)이 주어지고, 그 다음 줄에 N개의 자연수가 주어진다. 각 자연수의 크기는 100,000를 넘지 않는다.

 

▣ 출력설명

 

첫 줄에 뒤집은 소수를 출력합니다. 출력순서는 입력된 순서대로 출력합니다.

 

▣ 입력예제

 

5 32 55 62 3700 250

 

▣ 출력예제 

 

23 73

나의 풀이(오류로 실행 안됨)

n = int(input())

a = list(map(int,input().split()))

def reverse(x): #수를 뒤집는 함수
    x = list(str(x))
    x.sort(reverse =True)
    
    return x

def isPrime(x): #소수 구하기 함수
    ch = [0] * (x+1)
    for i in ragne(2,x+1):
        if ch[i] == 0:
            res = i
        for j in range(i,x+1,i):
            ch[j] = 1

    return res

for x in a:
    reverse_a = reverse(x)
    print(isPrime(reverse_a),end = ' ')

일단.. 수를 뒤집는 함수에서 제대로 구현이 안되었는데,

수만 뒤집고 int로 바꿔주면 되겠다고 생각한 바와는 달리

9100 -> 19로 구현하는데에 오류가 걸렸다

def reverse(x):
    res = 0
    while x > 0:
        
        t = x % 10
        x = x//10

        res = res*10 + t
        
    return  res

그래서 자릿수합을 구할 때 사용했던 방식처럼 while문을 이용하여 수를 뒤집어주었다.

 

reverse 함수를 실행한 결과

 

그 다음 소수 출력에서도 오류가 나왔었는데... 그 이유가

저 코드는 해당 수가 소수인지 검사하는 코드가 아니여서 그런듯

 

소수 구하는 코드를 새로 작성해주었다.

def isPrime(x):
    if x == 1:
        return False
    for i in range(2, x//2 +1):
        if x % i == 0:
            return False
    else:
        return True

앞선 풀이들에는 for문의 범위가 n+1이었지만, 

x//2

약수는 자기 자신의 절반까지만 존재한다.

따라서 for문으로 약수가 존재하는지 확인할 때 

그 수의 절반까지만 검사해도 이 수가 소수인지 아닌지 확인가능하기때문에 

range범위를 (2,x//2+1) 로 해준 것이다.

 

 

모범답안

n = int(input())

a = list(map(int,input().split()))

def reverse(x):
    res = 0
    while x > 0:
        
        t = x % 10
        x = x//10

        res = res*10 + t
        
    return  res

def isPrime(x):
    if x == 1:
        return False
    for i in range(2, x//2 +1):
        if x % i == 0:
            return False
    else:
        return True

for x in a:
    tmp = reverse(x)
    if isPrime(tmp) == True:
        print(tmp,end = ' ')

※ 알아야 할 것

- 소수 판별에 적절하게 boolean을 이용해서 출력하는 방법을 떠올리자!

- 소수 판별(약수가 있다면 false를 반환)에서 범위는 range(2, x//2 +1) , 약수는 그 수의 절반까지만 존재

 

728x90
반응형