algorithm

뒤집은 소수

juuuuuuun 2024. 4. 29. 18:39
 
설명

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

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

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

 

입력

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

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

 

출력

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

 

예시 입력 1 

9
32 55 62 20 250 370 200 30 100

 

예시 출력 1

23 2 73 2 3

 

소스 코드 1

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

public class Main {
    public static ArrayList<Integer> solution(int N,String[] nums) {
        int[] solNum=new int[N];
        String str;
        int max = -1;
        ArrayList<Integer> ans = new ArrayList<>();
        
        //입력받은 정수들을 뒤집어서 int형 변수로 solNum에 저장 및 최대값 찾기
        for (int i = 0; i < N; i++) {
            str = new StringBuilder(nums[i]).reverse().toString();
            solNum[i] = Integer.parseInt(str);
            if (solNum[i] > max) {
                max = solNum[i];
            }
        }

        //최대값으로 배열 만들어서 소수 배열 만들기 / 소수 -> 1 , 소수X -> 0 
        int arr[] = new int[max + 1];
        arr[1] = 1;
        for (int i = 2; i <= max; i++) {
            if (arr[i] == 0) {
                for (int j = i; j <= max; j += i) {
                    arr[j] = 1;
                }
                arr[i] = 0;
            }
        }

        //solNum(뒤집은 수) 중 소수(arr값이 0) 인 수만 출력
        for (int i = 0; i < solNum.length; i++) {
            if (arr[solNum[i]] == 0) {
                ans.add(solNum[i]);
            }
        }
        return ans;
    }

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int N = sc.nextInt();
        String[] nums = new String[N];
        for (int i = 0; i < N; i++) {
            nums[i] = sc.next();
        }

        for (Integer i : solution(N, nums)) {
            System.out.print(i + " ");
        }

    }
}

 

소스 코드 2 

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

public class Main {

	//소수 참,거짓 판별 함수
    public static boolean isPrime(int res) {
    
        //1이면 소수가 아님
        if (res == 1) {
            return false;
        }
        
        //2부터 자신의 수 -1 까지 돌며 나눠지는 수가 있다면 소수가 아님
        for (int i = 2; i < res; i++) {
            if (res % i == 0) {
                return false;
            }
        }
        //위의 조건 모두 아니라면 소수
        return true;
    }


    public static ArrayList<Integer> solution(int N,int[] nums) {
        ArrayList<Integer> ans = new ArrayList<>();
        
        //숫자 뒤집는 로직
        for (int i = 0; i < N; i++) {
            int tmp = nums[i];
            int res = 0;
            while (tmp > 0) {
                int t = tmp % 10;
                res = res * 10 + t;
                tmp = tmp / 10;
            }
            if (isPrime(res)) { //소수 판별이 참이라면 ans에 값을 추가한다.
                ans.add(res);
            }
        }
        return ans;
    }

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int N = sc.nextInt();
        int[] nums = new int[N];
        for (int i = 0; i < N; i++) {
            nums[i] = sc.nextInt();
        }

        for (Integer i : solution(N, nums)) {
            System.out.print(i + " ");
        }

    }
}