[Inflearn] 자바 알고리즘 문제풀이 #02-06 6. 뒤집은 소수

문제

설명

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

예를 들어 32를 뒤집으면 23이고, 23은 소수이다. 그러면 23을 출력한다. 단 910를 뒤집으면 19로 숫자화 해야 한다.

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

 

입력 

첫 줄에 자연수의 개수 N(3<=N<=100)이 주어지고, 그 다음 줄에 N개의 자연수가 주어진다.

각 자연수의 크기는 100,000를 넘지 않는다.

 

출력 

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

 

 

 

내 풀이

import java.util.ArrayList;
import java.util.Scanner;

public class Main2_6 {
    public ArrayList<Integer> Solution(int n, int[] arr) {
        ArrayList<Integer> answer = new ArrayList<>();
        for (int x : arr) {
            String tmp = new StringBuilder(String.valueOf(x)).reverse().toString();
            int revNum = Integer.parseInt(tmp); // 숫자 뒤집은 후 형변환

            int cnt = 0; // 소수 판별
            for (int i = 1; i <= revNum; i++) {
                if(revNum%i==0) cnt++;
            }
            if(cnt==2) answer.add(revNum);
        }
        return answer;
    }
    public static void main(String[] args) {
        Main2_6 T = new Main2_6();
        Scanner sc = new Scanner(System.in);
        int num = sc.nextInt();
        int[] arr = new int[num];
        for (int i = 0; i < num; i++) {
            arr[i] = sc.nextInt();
        }
        for(int x : T.Solution(num, arr))
            System.out.print(x + " ");
    }
}

 

 

 

선생님 풀이

import java.util.ArrayList;
import java.util.Scanner;

public class Main2_6 {
    public boolean isPrime(int num) { // 소수 판별
        if(num==1) return false;
        for (int i = 2; i < num; i++) {
            if(num%i==0) return false; // 약수가 있다면 소수X
        }
        return true;
    }

    public ArrayList<Integer> Solution(int n, int[] arr) {
        ArrayList<Integer> answer = new ArrayList<>();
        for (int i = 0; i < n; i++) {
            int tmp = arr[i];
            int res = 0;
            while (tmp > 0) { // 숫자 뒤집기
                int t = tmp % 10;
                res = res * 10 + t;
                tmp = tmp / 10;
            }
            if (isPrime(res)) answer.add(res);
        }
        return answer;
    }
    public static void main(String[] args) {
        Main2_6 T = new Main2_6();
        Scanner sc = new Scanner(System.in);
        int num = sc.nextInt();
        int[] arr = new int[num];
        for (int i = 0; i < num; i++) {
            arr[i] = sc.nextInt();
        }
        for(int x : T.Solution(num, arr))
            System.out.print(x + " ");
    }
}

먼저 풀어보면서 소수 판별하는 함수를 만들까 고민하다가 귀찮아서 안만들었는데 가독성이 훨씬 좋군...^^

그리고 숫자를 뒤집는 방법은 StringBuilder.reverse()밖에 생각이 안나서 그렇게 풀었는데 저런 공식(?)이 있었다니 !!

풀이법 알아두고 다음에 꼭 적용해봐야지 !!!!

그래도 오늘은 내 풀이가 정답 떠서 좋았다 ><

 

 

 

결과

 

 

 

알게된 점

1. StringBuilder

https://rookie-programmer.tistory.com/14

[JAVA] 자바 StringBuilder 주요 메소드

 

2. ArrayList

https://rookie-programmer.tistory.com/13

[JAVA] 자바 ArrayList