結果

問題 No.12 限定された素数
ユーザー uafr_cs
提出日時 2015-12-06 21:16:49
言語 Java
(openjdk 23)
結果
WA  
実行時間 -
コード長 2,124 bytes
コンパイル時間 2,947 ms
コンパイル使用メモリ 80,368 KB
実行使用メモリ 91,740 KB
最終ジャッジ日時 2024-09-14 16:03:29
合計ジャッジ時間 22,329 ms
ジャッジサーバーID
(参考情報)
judge1 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 10 WA * 14 RE * 2
権限があれば一括ダウンロードができます

ソースコード

diff #

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.List;
import java.util.Scanner;
import java.util.Set;
import java.util.TreeSet;

public class Main {
	
	public static void update(int[] count, String str, boolean add){
		for(final char ch : str.toCharArray()){
			count[Character.getNumericValue(ch)] += (add ? 1 : -1);
		}
	}
	
	public static boolean valid_range(int[] count, Set<Integer> valid){
		for(int i = 0; i < count.length; i++){
			if(valid.contains(i)  && count[i] <= 0){ return false; }
			if(!valid.contains(i) && count[i]  > 0){ return false; }
		}
		
		return true;
	}
	
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		
		final int N = sc.nextInt();
		Set<Integer> valid_set = new HashSet<Integer>();
		for(int i = 0; i < N; i++){
			valid_set.add(sc.nextInt());
		}
		
		final int MAX = 5000000;
		TreeSet<Integer> prime_tree = new TreeSet<Integer>();
		boolean[] is_prime = new boolean[MAX + 1];
		Arrays.fill(is_prime, true);
		is_prime[0] = is_prime[1] = false;
		
		for(int i = 2; i <= MAX; i++){
			if(!is_prime[i]){ continue; }
			
			prime_tree.add(i);
			for(int j = i * 2; j <= MAX; j += i){
				is_prime[j] = false;
			}
		}
		//System.out.println(valid_set);
		
		int lower = 1, upper = 1; // [lower, upper]
		int[] count = new int[10];
		int max = Integer.MIN_VALUE;
		while(lower <= upper){
			//System.out.println("[" + lower + ", " + upper + "] + " + Arrays.toString(count));
			
			if(valid_range(count, valid_set) && upper <= MAX){
				max = Math.max(max, (prime_tree.ceiling(upper) - 1) - (prime_tree.floor(lower - 1) + 1));
				//System.out.println(max);
				upper++;
				if(is_prime[upper]){ update(count, Integer.toString(upper), true); }
			}else if(lower < MAX){
				if(is_prime[lower]){ update(count, Integer.toString(lower), false); }
				lower++;
				
				if(lower > upper){
					if(is_prime[lower]){ update(count, Integer.toString(lower), true); }
					upper = lower;
				}
			}else{
				break;
			}
		}
		
		System.out.println(max == Integer.MIN_VALUE ? -1 : max);
	}

}
0