[JAVA] SWEA 5986 - 새샘이와 세 소수

hyng·2022년 4월 14일
0

SWEA

목록 보기
71/78
post-custom-banner

import java.util.*;
class Solution
{
	public static void main(String args[]) throws Exception
	{
		Scanner sc = new Scanner(System.in);


		ArrayList<Integer> primNums = new ArrayList<>();
		getPrimeNum(999, primNums);

		int T = sc.nextInt();

		StringBuffer sb = new StringBuffer();


		for (int tc = 1; tc <= T; tc++) {
			sb.append("#").append(tc).append(" ");

			int N = sc.nextInt();

			int sum  = 0;
			for(int i=0; i<primNums.size(); i++){
				if(primNums.get(i) > N){
					break;
				}
				for(int j=i; j<primNums.size(); j++){
					if(primNums.get(i) + primNums.get(j) > N){
						break;
					}
					for(int k=j; k<primNums.size(); k++){
						if(primNums.get(i) + primNums.get(j) + primNums.get(k) > N){
							break;
						}
						if(primNums.get(i) + primNums.get(j) + primNums.get(k) == N){
							sum++;
						}
					}
				}
			}

			sb.append(sum).append("\n");
		}

		System.out.println(sb);

	}

	private static void getPrimeNum(int max, ArrayList<Integer> primNums) {
		for(int i=2; i<=max; i++){
			boolean isPrime = true;
			for(int j=2; j<=Math.sqrt(i); j++){
				if(i % j == 0) {
					isPrime = false;
					break;
				}
			}
			if(isPrime){
				primNums.add(i);
			}
		}
	}
}
profile
공부하고 알게 된 내용을 기록하는 블로그
post-custom-banner

0개의 댓글