처음부터 차근차근

class2 - 10773번 제로 (자바)

_soyoung — Thu, 23 May 2024 17:39:55 +0900

문제 : https://www.acmicpc.net/problem/10773

필자는 해당 문제를 배열로 풀고 정답이라 나왔는데, 다른 사람들 풀이 보니까 stack을 쓰는 사람이 많아서 stack공부도 할겸 업로드하게 되었다.

문제에서 사용하는 개념

Stack 클래스

https://docs.oracle.com/javase/8/docs/api/java/util/Stack.html

함수명	반환 형식	기능
push(item)	void	스택에 요소 추가.
pop()	item	스택의 맨 위 요소를 제거하고 반환
peek()	item	스택의 맨 위 요소를 반환하나, 제거하지 않음.
empty()	boolean	스택이 비어 있는지 여부를 반환. 비어있으면 false, 아니면 true

문제 풀이

import java.util.*;
import java.io.*;

public class Main
{
	public static void main(String[] args) {
		try {
            BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
            BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
            
            int num = Integer.parseInt(br.readLine());
            Stack<Integer> stack = new Stack<>();
            
            for (int i = 0; i < num; i++) {
                int val = Integer.parseInt(br.readLine());
                
                if (val == 0) { // 0이면
                    stack.pop(); // 값 삭제
                    
                }
                else {
                    stack.push(val);
                }
            }
            
            int sum = 0; 
            
            
            while(!stack.empty()) {
                sum += stack.pop();
            }
            
            bw.write(String.valueOf(sum) + "\n");
            
            
            br.close();
            bw.flush();
            bw.close();
		}
		catch (IOException e) {
          System.out.println("IOException 발생");
		}
		
	}	
}

자바의 Stack 클래스를 이용한 방법이다.

그리고 아래는 배열을 이용한 방법이다.

import java.util.*;
import java.io.*;

public class Main
{
	public static void main(String[] args) {
		try {
            BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
            BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
            
            int num = Integer.parseInt(br.readLine());
            int sum = 0;        
            Integer[] arr = new Integer[num];
            
            for (int i = 0; i < num; i++) {
                arr[i] = Integer.parseInt(br.readLine());
                sum += arr[i];
                
                // 0이면
                if (arr[i] == 0) {
                    
                    // 최신값 삭제
                    for(int j = i-1; j >= 0; j--) {
                        if (arr[j] != null) {
                            sum -= arr[j];
                            arr[i] = null; // 0을 null로 만들기
                            arr[j] = null; // 최신값 null로 만들기
                            break;
                            
                        }
                    }
                    
                }
            }
            
            bw.write(String.valueOf(sum) + "\n");
            
            br.close();
            bw.flush();
            bw.close();
		}
		catch (IOException e) {
          System.out.println("IOException 발생");
		}
		
	}	
}

확실히 Stack 클래스를 쓰니까 코드가 깔끔해 진것을 볼 수 있다.

class2 - 11650번 좌표 정렬하기 (자바)

_soyoung — Thu, 23 May 2024 15:28:18 +0900

문제 : https://www.acmicpc.net/problem/11650

Arrays.sort함수는 1차원 배열 뿐만아니라 2차원 배열도 정렬 가능하다.

단, Comparator 객체를 써서 커스텀 해줘야한다.

문제에서 쓰인 개념

Arrays.sort로 2차원 배열 정렬.

Arrays.sort(dots, new Comparator<int[]>() { // 이차원 배열 정렬
    @Override
    public int compare(int[] one, int[] two) { 
        if (one[0] == two[0]) {  
            return one[1] - two[1]; // y좌표 오름차순
        }
        else {
            return one[0] - two[0]; // x좌표 오름차순
        }
    }
});

Comparator 객체의 제너럴 부분에 배열을 넣어야한다.

예를들어 arr = {{1, 2}, {2, 3}, {4, 5}...}이 있다고 가정하자.

one은 {1, 2} two는 {2, 3}이고,

one[0]은 1, two[0]은 2이다.

문제 풀이

import java.util.*;
import java.io.*;

public class Main
{
	public static void main(String[] args) {
		try {
            BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
            BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
            StringTokenizer st;
            
            int dotNum = Integer.parseInt(br.readLine());
            int[][] dots = new int[dotNum][2];
            
            for (int i = 0; i < dotNum; i++) {
                st = new StringTokenizer(br.readLine());
                
                int x = Integer.parseInt(st.nextToken());
                int y = Integer.parseInt(st.nextToken());
                
                dots[i][0] = x;
                dots[i][1] = y;
            }
            
            Arrays.sort(dots, new Comparator<int[]>() { // 이차원 배열 정렬
                @Override
                public int compare(int[] one, int[] two) {
                    if (one[0] == two[0]) { // x 좌표 비교
                        return one[1] - two[1]; // y좌표 오름차순
                    }
                    else {
                        return one[0] - two[0]; // x좌표 오름차순
                    }
                }
            });
            
            for(int[] dot : dots) {
                bw.write(String.valueOf(dot[0])+" "+String.valueOf(dot[1]) + "\n");
            }
            
            
            br.close();
            bw.flush();
            bw.close();
		}
		catch (IOException e) {
          System.out.println("IOException 발생");
		}
		
	}	
}

Arrays.sort를 쓰니 어렵지 않게 문제를 해결할 수 있었다.

class2 - 2775번 부녀회장이 될테야 (자바)

_soyoung — Thu, 23 May 2024 14:25:32 +0900

문제 : https://www.acmicpc.net/problem/2775

제한 시간이 0.5초로 아주 짧다.

이 문제를 봤을 때 처음 떠올렸던 방식이 아예 처음에 2차원 배열을 초기화 시키는 방식이었다.

그러나 최대 층수도 안정해져있고, for문을 돌리기에는 제한 시간이 너무 적어서 다른 방법을 고안해냈다.

그러다 떠올린 방법이 재귀함수를 이용하는 방법이었다.

어차피 k층 n호 사람 수 = k-1층 1~n호 사람 수의 합 이니까 재귀 함수를 통해서 구현할 수 있다고 생각했다.

문제 풀기전 계획

k층 sum(n호 사람 수) = k-1층 sum(1호 ~ n호 사람 수) 

arr[15] = { 1....14}


a(int k층, int n호){
	if (k == 0)
	{
		return arr[n];
	}
	else {
		int temp = 0;
		for (int i = 0; i < n호; i++) {
			temp += a(int k - 1층, int i호);
		}
		return temp;
	}
		
}

문제 풀기 전 이런식으로 큰 틀을 짠 다음에 실제 코드를 짰다.

풀이

import java.util.*;
import java.io.*;

public class Main
{
    static int[] zeroFloorPeopleNum = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14};
	public static void main(String[] args) {
		try {
            BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
            BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
            
            int caseNum = Integer.parseInt(br.readLine());
            int[] results = new int[caseNum];
            
            for (int i = 0; i < caseNum; i++) {
                int floor = Integer.parseInt(br.readLine());
                int roomNum = Integer.parseInt(br.readLine());
                
                results[i] = countPeople(floor, roomNum);
            }
            
            for(int result : results) {
                bw.write(String.valueOf(result)+"\n");
            }
            
            
            br.close();
            bw.flush();
            bw.close();
		}
		catch (IOException e) {
          System.out.println("IOException 발생");
		}
		
	}
	
	
	public static int countPeople(int floor, int roomNum) {
	    if (floor == 0) {
	        return zeroFloorPeopleNum[roomNum];
	    }
	    else {
	       int sum = 0;
	       for (int i = 1; i <= roomNum; i++) {
	           sum += countPeople(floor - 1, i);
	       }
	       
	       return sum;
	    }
	}
	
}

정답을 맞추고 다른 사람들은 어떻게 짰나 궁금해서 찾아봤는데 다른 문제랑 다르게 사람마다 풀이가 다 다르고, 다양해서 신기했다.

class2 - 2751번 수 정렬하기2 (자바)

_soyoung — Thu, 23 May 2024 10:38:10 +0900

문제 : https://www.acmicpc.net/problem/2751

해당 문제를 Arrays.sort 써서 했더니 실버5 문제임에도 불구하고 정답이 나와버려서 너무 허무했다.

그래서 다른 풀이 방법을 찾아봤다.

이 문제는 Collections.sort()를 쓰면 더 빠르게 풀수있다.

문제에서 사용하는 개념

함수	알고리즘	평균 시간복잡도	최악 시간복잡도
Arrays.sort()	dual-pivot Quicksort	O(nlogn)	O(n2)
Collections.sort()	hybrid sorting algorithm (= Timsort) (Timsort = 합병정렬 + 삽입정렬)	합병정렬 : O(nlogn) 삽입정렬 : O(n)	합병정렬 : O(nlogn) 삽입정렬 : O(n2)

https://docs.oracle.com/javase%2F7%2Fdocs%2Fapi%2F%2F/java/util/Collections.html

이런식으로 쓸 수 있다.

풀이

import java.util.*;
import java.io.*;

public class Main
{
	public static void main(String[] args) {
		try {
            BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
            BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
            
            int num = Integer.parseInt(br.readLine());
            List<Integer> arr = new ArrayList<>();
            
            for(int i = 0; i < num; i++) {
                arr.add(Integer.parseInt(br.readLine()));
            }
            
            Collections.sort(arr);
            
            for(int val : arr) {
                bw.write(String.valueOf(val)+"\n");
            }
            
            br.close();
            bw.flush();
            bw.close();
		}
		catch (IOException e) {
          System.out.println("IOException 발생");
		}
		
	}
	
}

class2 - 2609번 최대공약수와 최소공배수 (자바)

_soyoung — Thu, 23 May 2024 10:23:14 +0900

문제 : https://www.acmicpc.net/problem/2609

이 문제에서 필요한 개념

유클리드 호제법

최대공약수를 구하는 알고리즘.

큰 수를 작은 수로 나누어 떨어지게 한 뒤, 수를 반복적으로 수행하여 나머지 0이 될때까지 작동함.

public static int gcd(int num1, int num2) {
    if (num2 == 0) return num1;
    return gcd(num2, num1 % num2);
}

위와 같이 재귀방식으로 최대공약수를 구한다.

여기서 주의할점은 큰수를 num1로, 작은수를 num2로 둬야한다는 점. 안그러면 무한루프에 빠지게 된다...

최소공배수 = (두 수의 곱) / 최대공약수

문제 풀이

import java.util.*;
import java.io.*;

public class Main
{
	public static void main(String[] args) {
		try {
            BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
            BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
            StringTokenizer st = new StringTokenizer(br.readLine());
            
            int num1 = Integer.parseInt(st.nextToken());
            int num2 = Integer.parseInt(st.nextToken());
            
            // 최대공약수 구하기
            int gcd = gcd(num1, num2);
            
            // 최소공배수 구하기
            int lcm = num1 * num2 / gcd;
            
            bw.write(String.valueOf(gcd)+"\n"); // 최대공약수
            bw.write(String.valueOf(lcm)+"\n"); // 최소공배수
           
            br.close();
            bw.flush();
            bw.close();
		}
		catch (IOException e) {
          System.out.println("IOException 발생");
		}
		
	}
	
	public static int gcd(int num1, int num2) {
	    if (num2 == 0) return num1;
	    return gcd(num2, num1 % num2);
	}
}

유클리드 호제법을 알고있으면 간단하게 풀 수 있는 문제였다.

class2 - 2108번 통계학 (자바)

_soyoung — Wed, 22 May 2024 16:07:14 +0900

문제 : https://www.acmicpc.net/problem/2108

최빈값 구하는데 시간이 너무 오래걸렸었다.

입력값으로 음수가 들어와서 최빈값 구할 때 배열을 하나 더 만들었어야 했었고,

여러 개 있을 때에는 최빈값 중 두 번째로 작은 값을 출력하라는 조건도 있어서 로직이 복잡했다ㅠㅠ

결국 최빈값 부분은 해답을 보고 풀었었는데, 코드를 짧고 간결하게 잘짠 사람들이 많아서 대단하게 느껴졌다...

문제에서 사용한 개념

Math.round(소수) : 소수점 첫번째자리에서 반올림 하여 정수를 리턴함. 가장 가까운 정수를 리턴한다고 생각하면 편함.

리턴 형식	매개변수
long	round(double a)
int	round(float a)

공식 문서 : https://docs.oracle.com/javase/7/docs/api/java/lang/Math.html

풀이

import java.util.*;
import java.io.*;

public class Main
{
	public static void main(String[] args) {
		try {
            BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
            BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
            
            int num = Integer.parseInt(br.readLine());
            int[] arr = new int[num];
            int sum = 0; // 합
          
            
            for (int i = 0; i < num; i++) {
                arr[i] = Integer.parseInt(br.readLine());
                sum += arr[i];
            }
            
            // 산술평균
            long arithmeticMean = Math.round((double)sum / num);
            
            // 중앙값 
            Arrays.sort(arr);
            int median = arr[(num -1) / 2];
            
            // 최빈값 
            int[] count = new int[8001]; // 빈도 수 카운팅 배열
            int mode = 0; // 최빈값
            int maxCount = 0; // 카운트 최대값
            boolean flag = false; // 두번째 최빈값 찾기 위한 플래그
            
            for (int i = 0; i < arr.length; i++) {
                count[arr[i] + 4000]++;
            }
            
            for (int i = 0; i < count.length; i++) { // 최빈값 찾기
                if (count[i] > maxCount) {
                    mode = i - 4000;
                    maxCount = count[i];
                    flag = true;
                } else if (count[i] == maxCount && flag == true) {
                    mode = i - 4000;
                    flag = false;
                }
            }
           
            // 범위
            int range = arr[num - 1] - arr[0];
            
            bw.write(String.valueOf(arithmeticMean)+"\n"); // 산술평균
            bw.write(String.valueOf(median)+"\n"); // 중앙값
            bw.write(String.valueOf(mode)+"\n"); // 최빈값
            bw.write(String.valueOf(range)+"\n"); // 범위
           
            br.close();
            bw.flush();
            bw.close();
		}
		catch (IOException e) {
          System.out.println("IOException 발생");
		}
		
	}
}

예제는 모두 맞았는데 답 제출하기만 하면 틀렸다고 나와서 한참을 찾아보다가 산술평균 반환 형식이 잘못됐다는 것을 깨달았다.

double형 매개변수를 넣었으면 리턴 형식이 long인데 int형을 써서 에러가 났었다...

class2 - 1920번 수 찾기 (자바)

_soyoung — Tue, 21 May 2024 14:37:42 +0900

문제 : https://www.acmicpc.net/problem/1920

이 문제는 시간 제한이 1초로 굉장히 짧다.

그래서 이분탐색(이진탐색) 알고리즘을 사용해야 한다.

정말 놀라운 사실이.. 자바에는 이분탐색 함수가 이미 만들어져 있다!!!

그래서 손쉽게 풀 수 있었다.

문제에서 사용되는 개념

이분탐색(이진탐색)

정렬되어 있는 리스트에서 탐색 범위를 절반씩 좁혀가며 데이터를 탐색하는 방법.

예시)

{1, 2, 3, 4, 5} 배열이 있고, 찾고자 하는 값은 2 이다.

가장 먼저 가운데 3을 찍어서 2가 3보다 큰지 작은지 같은지 비교하고, 3보다 작으면 왼쪽으로, 크면 오른쪽으로 가서 찾는다.

Arrays.binarySearch(arr, key)

https://docs.oracle.com/javase/8/docs/api/java/util/Arrays.html

배열에서 이분탐색을 하는 메소드.

오름차순된 배열 arr에서 key인 요소를 찾음.

검색 성공 시 : key와 일치하는 요소의 인덱스를 반환( 만약 일치하는 요소가 여러 개 있다면 무작위의 인덱스를 반환 )

검색 실패 시 : 음수 반환

풀이

import java.util.*;
import java.io.*;

public class Main
{
	public static void main(String[] args) {
		try {
            BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
            BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
            StringTokenizer st;
            
            // 값 받음1
            int num1 = Integer.parseInt(br.readLine());
            st = new StringTokenizer(br.readLine());
            int[] arr1 = new int[num1];
            for (int i = 0; i < num1; i++) {
                arr1[i] = Integer.parseInt(st.nextToken());
            }
            
            Arrays.sort(arr1); // 오름차순 정렬
            
            // 값 받음2
            int num2 = Integer.parseInt(br.readLine());
            st = new StringTokenizer(br.readLine());
            int[] arr2 = new int[num2];
            int result;
            for (int i = 0; i < num2; i++) {
                arr2[i] = Integer.parseInt(st.nextToken());
                
                // 이분 탐색
                result = Arrays.binarySearch(arr1, arr2[i]);
                if (result < 0) bw.write("0\n");
                else bw.write("1\n");
            }
           
            br.close();
            bw.flush();
            bw.close();
		}
		catch (IOException e) {
          System.out.println("IOException 발생");
		}
		

	}
}

자바 Arrays 클래스의 Arrays.binarySearch(arr, key) 함수를 이용한 방법이다.

import java.util.*;
import java.io.*;

public class Main
{
	public static void main(String[] args) {
		try {
            BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
            BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
            StringTokenizer st;
            
            // 값 받음1
            int num1 = Integer.parseInt(br.readLine());
            st = new StringTokenizer(br.readLine());
            int[] arr1 = new int[num1];
            for (int i = 0; i < num1; i++) {
                arr1[i] = Integer.parseInt(st.nextToken());
            }
            
            Arrays.sort(arr1); // 오름차순 정렬
            
            // 값 받음2
            int num2 = Integer.parseInt(br.readLine());
            st = new StringTokenizer(br.readLine());
            int[] arr2 = new int[num1];
            for (int i = 0; i < num2; i++) {
                arr2[i] = Integer.parseInt(st.nextToken());
            }
            
            
            // 있는지 확인 - 이분 탐색
            for (int j = 0; j < num2; j++) {
                int result = 0; // 없음
                int end = arr1.length - 1; // 끝 인덱스
                int start = 0; // 시작 인덱스
                int middle = 0;
                while (true) {
                    middle = (end - start) / 2;
                    if (middle < 0 || end > arr1.length - 1) break;

                    if ( arr2[j] > arr1[middle]) {
                        start = middle + 1;
                    }
                    else if (arr2[j] < arr1[middle]) {
                        end = middle - 1;
                        // if (value == 0 && arr1[0] != arr2[j]) {
                        //     break;
                        // }
                    }
                    else if (arr1[middle] == arr2[j]) {
                        result = 1;
                        break;
                    }
                    
                }
                bw.write(String.valueOf(result) + "\n");
            }
           
            br.close();
            bw.flush();
            bw.close();
		}
		catch (IOException e) {
          System.out.println("IOException 발생");
		}
		

	}
}

이진 탐색을 수동으로 하는 코드이다.

문제 풀었던 과정

import java.util.*;
import java.io.*;

public class Main
{
	public static void main(String[] args) {
		try {
            BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
            BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
            StringTokenizer st;
            
            int num1 = Integer.parseInt(br.readLine());
            st = new StringTokenizer(br.readLine());
            int[] arr1 = new int[num1];
            for (int i = 0; i < num1; i++) {
                arr1[i] = Integer.parseInt(st.nextToken());
            }
            
            int num2 = Integer.parseInt(br.readLine());
            st = new StringTokenizer(br.readLine());
            int[] arr2 = new int[num1];
            int result = 0;
            for (int i = 0; i < num2; i++) {
                arr2[i] = Integer.parseInt(st.nextToken());
                for (int j = 0; j < num1; j++) {
                    if (arr2[i] == arr1[j]) {
                        result = 1; // 존재함
                    }
                }
                
                bw.write(String.valueOf(result) + "\n");
                result = 0;
            }
        

           
            br.close();
            bw.flush();
            bw.close();
		}
		catch (IOException e) {
          System.out.println("IOException 발생");
		}
	}
}

처음에 이런식으로 이분탐색을 사용하지 않고 풀었다가 시간초과가 났었다.

그래서 이분탐색 알고리즘을 사용하여 문제를 풀었다.

class2 - 1676번 팩토리얼 0의 개수 (자바)

_soyoung — Tue, 21 May 2024 11:09:21 +0900

문제 : https://www.acmicpc.net/problem/1676

문제의 최댓값인 500!은 자릿수만 1135자 가까이 돼서 일반적인 int문으로 데이터를 담는것이 불가능하다.

그래서 BigInteger을 써야한다.

문제에서 사용되는 개념

BigInteger

타입	범위	기본형/참조형	저장 위치
int	-2,147,483,648 ~ 2,147,483,647	기본형(value type)	Stack
BigInteger	무한 (Infinity)	참조형(reference type)	Heap

선언 방법

import java.math.BigInteger;
BigInteger num = new BigInteger("1");

연산

num.subtract(BigInteger.valueOf(2))
num.multiply(BigInteger.valueOf(2))
num.divide(BigInteger.valueOf(2))
num.remainder(BigInteger.valueOf(2))

형변환

BigInteger.valueOf(1); // int -> BigIntger
num.intValue(); //BigIntger -> int
num.longValue(); //BigIntger -> long
num.floatValue(); //BigIntger -> float
num.doubleValue(); //BigIntger -> double
num.toString(); //BigIntger -> String

풀이

import java.util.*;
import java.io.*;
import java.math.BigInteger;

public class Main
{
	public static void main(String[] args) {
		try {
            BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
            BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

            // 팩토리얼 구하기
            int num = Integer.parseInt(br.readLine());
            BigInteger fac = new BigInteger("1");
            for (int i = num; i > 0; i--) {
                fac = fac.multiply(BigInteger.valueOf(i));
            }
            
            char[] arr = String.valueOf(fac).toCharArray();
            
            int count = 0;    
            
            for (int i = arr.length - 1; i >= 0; i--) {
                if (arr[i] == '0') {
                    count++;
                }
                else {
                    break;
                }
            }
            
            bw.write(String.valueOf(count));
            
            br.close();
            bw.flush();
            bw.close();
		}
		catch (IOException e) {
          System.out.println("IOException 발생");
		}
	}
}

처음에 int로 담았다가 틀려서 BigInteger 사용해서 다시 짰다.

class2 - 1929번 소수 찾기 (자바)

_soyoung — Sun, 19 May 2024 21:34:56 +0900

문제 : https://www.acmicpc.net/problem/1929

해당문제는 변수는 1000000까지 될 수 있으나 제한 시간이 2초여서 알고리즘을 사용해야한다.

문제 풀 때 필요한 개념

에라토스테네스의 체

고대 그리스의 수학자 에라토스테네스가 만들어 낸 소수를 찾는 방법.

임의의 수 N 까지의 소수를 구하고자 할 때,

2부터 √N (제곱근) 까지 돌며 모든 배수들을 소수에서 제외시키는 방식이다.

2 * 4 = 4 * 2 처럼 모든 약수를 구하기에는 겹치는 부분이 많아 비효율적이다.

그래서 주어진 수의 제곱근 까지만 약수를 구하면 효율적으로 소수를 구할 수 있다.

1. 2부터 √N 까지의 배수를 제외시킨다.

2. 배수를 구할 때는 자기 자신을 뺀다.

3. 제외된 수는 배수를 구하지 않고 넘어간다.

풀이

import java.io.*;
import java.util.*;

public class Main {
	public static void main(String[] args) {
		try {
            BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
            BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
            StringTokenizer st;
            
            st = new StringTokenizer(br.readLine());
            int start = Integer.parseInt(st.nextToken());
            int end = Integer.parseInt(st.nextToken());
            
            Boolean[] arr = new Boolean[end+1];
            
            Arrays.fill(arr, true); // 초기값 세팅
            arr[0] = false; // 0은 무조건 false로 세팅한다.
            arr[1] = false; // 1은 무조건 false로 세팅한다.
            int real_end = (int)Math.sqrt((double)end); 
            
            for (int i = 2; i <= real_end; i++) {	
                if (!arr[i]) continue;
            	for (int j = 2; i*j <= end; j++) {
            	    arr[i*j] = false;
            	}
            }
            
            
            // 출력
            for (int i = start; i <= end; i++) {
                if (arr[i]) {
                    bw.write(String.valueOf(i)+"\n");
                }
            }
           

            // close
            br.close();
            bw.flush();
            bw.close();
		}
		catch (IOException e) {
          System.out.println("IOException 발생");
		}	
	}
}

문제푼 과정

package backjoon;

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
import java.util.List;
import java.util.StringTokenizer;

public class q_1929 {
	public static void main(String[] args) {
		try {
            BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
            BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
            StringTokenizer st;
            
            st = new StringTokenizer(br.readLine());
            int start = Integer.parseInt(st.nextToken());
            int end = Integer.parseInt(st.nextToken());
            
            List<Integer> results = new ArrayList<>();
            
            for (int i = start; i < end; i++) {	
            	for (int j = 2; j < 1000000; j++) {
            		if (i == 2 || i == j) {
            			results.add(i);
            			break;
            		}
            		else if (i == 1 || i % j == 0) {
            			break;
            		}
            	}
            }
            
            
            // 출력
            for (Integer result : results) {
            	 bw.write(String.valueOf(result)+"\n");
            }
           

            // close
            br.close();
            bw.flush();
            bw.close();
		}
		catch (IOException e) {
          System.out.println("IOException 발생");
		}	
	}
}

이렇게 했더니 시간초과가 나서 다른 방법을 찾아보게 되었고,

이 문제는 알고리즘을 이용해서 풀어야 되는 문제임을 알게되었다.

class2 - 1978번 소수 찾기 (자바)

_soyoung — Sun, 19 May 2024 21:13:22 +0900

문제 : https://www.acmicpc.net/problem/1978

전체적으로 쉽게 풀 수 있는 문제였다.

문제 분석 :

소수 = 1과 자신만의 숫자를 약수로 가지는 수

그러므로 1과 자기자신이 아닐 경우에 나눠서 0이 되면 count를 하지 않는다.

소스

package backjoon;

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.StringTokenizer;

public class q_1978 {
	public static void main(String[] args) {
		try {
            BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
            BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
            StringTokenizer st;
            
            int num = Integer.parseInt(br.readLine());            
            
            int count = 0;
            int temp;
            st = new StringTokenizer(br.readLine());
            for (int i = 0; i < num; i++) {
            	temp = Integer.parseInt(st.nextToken());
            	
            	for (int j = 2; j < 1000; j++) {
            		if (j == temp) {
            			count++;
            			break;
            		}
            		else if (temp == 1 || temp % j == 0) {
            			break;
            		}
            		
            	}
            }
            
            
            // 출력
            bw.write(String.valueOf(count)+"\n");

            // close
            br.close();
            bw.flush();
            bw.close();
		}
		catch (IOException e) {
          System.out.println("IOException 발생");
		}	
	}
}