Java 코딩테스트 공부/Java 알고리즘 공부

자바 배열 파트 알고리즘2 / 7월 13일

daramG 2022. 7. 13. 23:23

https://www.inflearn.com/course/%EC%9E%90%EB%B0%94-%EC%95%8C%EA%B3%A0%EB%A6%AC%EC%A6%98-%EB%AC%B8%EC%A0%9C%ED%92%80%EC%9D%B4-%EC%BD%94%ED%85%8C%EB%8C%80%EB%B9%84/dashboard

 

자바(Java) 알고리즘 문제풀이 : 코딩테스트 대비 - 인프런 | 강의

자바(Java)로 코딩테스트를 준비하시는 분을 위한 강좌입니다. 코딩테스트에서 가장 많이 출제되는 Top 10 Topic을 다루고 있습니다. 주제와 연동하여 기초문제부터 중급문제까지 단계적으로 구성

www.inflearn.com

 

2. 섹션 2. Array(1, 2차원 배열)

7월 12일
1. 큰 수 출력하기 03 : 05
2. 보이는 학생 07 : 59
3. 가위바위보 07 : 44
7월 13일
4. 피보나치 수열 06 : 24
5. 소수(에라토스테네스 체) 06 : 48
6. 뒤집은 소수 09 : 46

 

피보나치 수열

 

문제

입력한 숫자 만큼 피보나치 수열 출력

 

소스코드 :

import java.util.*;
class Main {
	public int[] solution(int n) {
		int[] arr = new int[n];
		arr[0] = 1;
		arr[1] = 1;
		for(int i=2; i<=n-1; i++) {
			arr[i] = arr[i-2] + arr[i-1];
		}
		return arr;
	}

	public static void main(String[] args) {
		Main T = new Main();
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		for(int x : T.solution(n)) {
			System.out.print(x+" ");
		}
	}
}

 

소수(에라토스테네스 체)

 

문제

1부터 입력한 N까지의 소수 개수 출력

 

문제풀이 : 2부터 시작해서 해당 수 제외한 해당 수의 배수들을 제거하는 방법

 

소스코드 :

import java.util.*;
class Main {
	public int solution(int n) {
		int answer = 0;
		int arr[] = new int[n+1];
		for(int i=2; i<=n; i++) {
			if (arr[i] == 0) {
				answer++;
				for(int j=i; j<=n; j+=i) {
					arr[j] = 1;
				}
			}
		}
		return answer;
	}

	public static void main(String[] args) {
		Main T = new Main();
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		System.out.print(T.solution(n));
	}
}

 

 

뒤집은 소수

 

문제

주어진 수들을 뒤집은 수가 소수인 경우의 수들을 모두 출력

출력 순서는 입력된 순서대로

ex)

입력

4

55 200 30 62

출력

2 3

 

내 풀이 :

import java.util.*;
class Main {
	public boolean isPrime(int s) {
		if (s == 1) return false;
		for (int i=2; i<s; i++) {
			if (s % i == 0) return false;
		}
		return true;
	}
	
	public ArrayList<Integer> solution(int n, String str) {
		ArrayList<Integer> answer = new ArrayList<>();
		String reverseStr = new StringBuilder(str).reverse().toString();
		// 26 03 002 55
		//int[] intArr = Stream.of(reverseStr.split(" ")).mapToInt(Integer::parseInt).toArray();
		int[] intArr = Arrays.stream(reverseStr.split(" ")).mapToInt(Integer::parseInt).toArray();
		// 26 3 2 55
		for(int i=n-1; i>=0; i--) {
			if (isPrime(intArr[i])) answer.add(intArr[i]);
		}
		return answer;
	}

	public static void main(String[] args) {
		Main T = new Main();
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		sc.nextLine();
		String str = sc.nextLine();
		// 4
		// 55 200 30 62

		for(int x : T.solution(n, str)) {
			System.out.print(x + " ");
		}
	}
}

주어진 수들을 문자열 한 줄로 받아서 StringBuilder 이용해 뒤집어버린다.

그리고 자바 8 스트림 라이브러리 이용해 공백 기준 잘라서 그대로 int배열에 집어넣고

소수인지 확인해서 그대로 출력한다. (뒤집었으니 해당 배열 역순으로 출력해야 출력 순서가 입력한 순서대로 된다.)

 

 

 

강의 풀이 :

import java.util.*;
class Main {
	public boolean isPrime(int s) {
		if (s == 1) return false;
		for (int i=2; i<s; i++) {
			if (s % i == 0) return false;
		}
		return true;
	}
	
	public ArrayList<Integer> solution(int n, int[] arr) {
		ArrayList<Integer> answer = new ArrayList<>();
		// 뒤집은 결과를 res에 저장함, 처음엔 0
		for (int i=0; i<n; i++) {
			int tmp = arr[i]; // tmp = 123
			int res=0; // 0
			while(tmp>0) {
				int t = tmp % 10; // t = 3 , 2, 1
				res = res*10 + t; // res = 3 , 32, 321
				tmp = tmp / 10; // tmp = 12 , 1, 0
			}
			if(isPrime(res)) answer.add(res);
		}
		return answer;
	}

	public static void main(String[] args) {
		Main T = new Main();
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		int[] arr = new int[n];
		for(int i=0; i<n; i++) {
			arr[i] = sc.nextInt();
		}
		for(int x : T.solution(n, arr)) {
			System.out.print(x+ " ");
		}
	}
}

처음부터 int배열로 수들을 받아서 for문으로 하나씩 뒤집은 다음 소수인지 판별해 출력한다.