[백준 1978] 소수 찾기

문제

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

1978번: 소수 찾기

---

내가 작성한 코드

import java.io.*;
import java.util.ArrayList;
import java.util.StringTokenizer;

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

        int num = Integer.parseInt(br.readLine());
        StringTokenizer st = new StringTokenizer(br.readLine());

        int count = 0;

        for(int i = 0; i<num; i++){
            int input = Integer.parseInt(st.nextToken());
            boolean state = true;

            if(input == 1){
                state = false;
            }
            else if(input == 2 || input == 3){
                state = true;
            }
            else{
                for(int j = 2; j*j<=input; j++){
                    if(input%j==0){
                        state = false;
                        break;
                    }
                }
            }

            if(state == true){
                count++;
            }
        }

        bw.write(count+"");
        bw.close();
    }
}

소수 찾는 법

누군가의 체?????? 이거 써야 된다더라

i*i가 내가 소수인지 아닌지 판별하고자 하는 친구보다 작을 때만 반복문이 돌아가도록 하면 된다