結果

問題 No.843 Triple Primes
コンテスト
ユーザー k_6101
提出日時 2019-12-08 01:05:00
言語 Java
(openjdk 23)
結果
AC  
実行時間 443 ms / 2,000 ms
コード長 2,183 bytes
コンパイル時間 2,806 ms
コンパイル使用メモリ 81,884 KB
実行使用メモリ 49,448 KB
最終ジャッジ日時 2024-12-27 12:51:36
合計ジャッジ時間 17,863 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 42
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.io.InputStream;
import java.io.PrintWriter;
import java.lang.reflect.Array;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.PriorityQueue;
import java.util.Scanner;
import java.util.Set;
import java.util.Stack;
import java.util.TreeMap;
import java.util.TreeSet;

import static java.util.Comparator.*;

public class Main {
    public static void main(String[] args) {
        PrintWriter out = new PrintWriter(System.out);
        Solver solver = new Solver(System.in, out);
        solver.solve();
        out.close();
    }
}
class Solver {
	Scanner sc;
	PrintWriter out;
    public Solver(InputStream in, PrintWriter out) {
    	sc = new Scanner(in);
    	this.out = out;
    }
    // ==================================================================
    // エラトステネスの篩(ふるい）　→　素数の表を作る（大体　O(N) らしい）								
    boolean[] arr = new boolean[500100];		// 初期値 false(素数なら false となる)			
    void eratosthenes(long n){								
    	arr[0] = arr[1] = true;					// 素数でない			
    	for(int i = 2; i * i <= n; i++){							
   		 if( !arr[i] ) {						
    			for(int j = i + i; j <= n; j += i){		// j は i の倍数となる		
   				 arr[j] = true;					// i の倍数は素数でない		
    			}					
    		}						
    	}							
    }
    Set<Integer> setR = new TreeSet<>();
    Set<Integer> setP = new TreeSet<>();
    public void solve() {
    	long N = Integer.parseInt(sc.next());
    	eratosthenes(N);
    	for (int i = 0; i * i <= 2*N; i++) {
			if(!arr[i])		setR.add(i);
		}
    	for (int i = 0; i <= N; i++) {
			if(!arr[i])		setP.add(i);
		}
    	int cnt = 0;
    	long wk;
    	for (int r : setR) {
			wk = (long)r * r;
			for (int p : setP) {
				if(wk - p > N)		continue;
				if(wk - p < p)		break;
				if(wk - p == p)							cnt++;
				else if(setP.contains((int)(wk - p))) 	cnt += 2;
			}
		}
    	out.println(cnt);
    }
    // ==================================================================
}
0