#61701 (C++11) No.308 素数は通れません

提出ソース

結果

問題	No.308 素数は通れません
ユーザー	zerokugi
提出日時	2015-12-01 08:38:29
言語	C++11 (gcc 11.4.0)
結果	AC
実行時間	6 ms / 1,000 ms
コード長	2,550 bytes
コンパイル時間	1,179 ms
コンパイル使用メモリ	77,100 KB
実行使用メモリ	6,824 KB
最終ジャッジ日時	2024-11-27 18:08:34
合計ジャッジ時間	3,399 ms
ジャッジサーバーID （参考情報）	judge5 / judge1

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,816 KB
testcase_01	AC	2 ms 6,820 KB
testcase_02	AC	2 ms 6,816 KB
testcase_03	AC	2 ms 6,816 KB
testcase_04	AC	2 ms 6,820 KB
testcase_05	AC	2 ms 6,816 KB
testcase_06	AC	2 ms 6,820 KB
testcase_07	AC	2 ms 6,816 KB
testcase_08	AC	1 ms 6,820 KB
testcase_09	AC	2 ms 6,820 KB
testcase_10	AC	2 ms 6,820 KB
testcase_11	AC	2 ms 6,816 KB
testcase_12	AC	2 ms 6,820 KB
testcase_13	AC	2 ms 6,820 KB
testcase_14	AC	1 ms 6,816 KB
testcase_15	AC	1 ms 6,816 KB
testcase_16	AC	2 ms 6,820 KB
testcase_17	AC	2 ms 6,816 KB
testcase_18	AC	1 ms 6,820 KB
testcase_19	AC	2 ms 6,820 KB
testcase_20	AC	1 ms 6,816 KB
testcase_21	AC	1 ms 6,816 KB
testcase_22	AC	2 ms 6,816 KB
testcase_23	AC	2 ms 6,816 KB
testcase_24	AC	2 ms 6,820 KB
testcase_25	AC	2 ms 6,820 KB
testcase_26	AC	2 ms 6,820 KB
testcase_27	AC	1 ms 6,820 KB
testcase_28	AC	2 ms 6,820 KB
testcase_29	AC	2 ms 6,816 KB
testcase_30	AC	2 ms 6,816 KB
testcase_31	AC	2 ms 6,816 KB
testcase_32	AC	2 ms 6,816 KB
testcase_33	AC	2 ms 6,820 KB
testcase_34	AC	2 ms 6,820 KB
testcase_35	AC	2 ms 6,816 KB
testcase_36	AC	2 ms 6,816 KB
testcase_37	AC	2 ms 6,816 KB
testcase_38	AC	1 ms 6,816 KB
testcase_39	AC	2 ms 6,816 KB
testcase_40	AC	2 ms 6,816 KB
testcase_41	AC	2 ms 6,816 KB
testcase_42	AC	2 ms 6,816 KB
testcase_43	AC	2 ms 6,820 KB
testcase_44	AC	2 ms 6,820 KB
testcase_45	AC	1 ms 6,816 KB
testcase_46	AC	2 ms 6,816 KB
testcase_47	AC	2 ms 6,816 KB
testcase_48	AC	2 ms 6,820 KB
testcase_49	AC	1 ms 6,816 KB
testcase_50	AC	2 ms 6,816 KB
testcase_51	AC	2 ms 6,820 KB
testcase_52	AC	2 ms 6,816 KB
testcase_53	AC	2 ms 6,816 KB
testcase_54	AC	2 ms 6,820 KB
testcase_55	AC	2 ms 6,816 KB
testcase_56	AC	2 ms 6,820 KB
testcase_57	AC	2 ms 6,820 KB
testcase_58	AC	2 ms 6,820 KB
testcase_59	AC	2 ms 6,816 KB
testcase_60	AC	2 ms 6,816 KB
testcase_61	AC	2 ms 6,820 KB
testcase_62	AC	2 ms 6,816 KB
testcase_63	AC	2 ms 6,816 KB
testcase_64	AC	2 ms 6,816 KB
testcase_65	AC	2 ms 6,820 KB
testcase_66	AC	2 ms 6,820 KB
testcase_67	AC	2 ms 6,820 KB
testcase_68	AC	2 ms 6,816 KB
testcase_69	AC	1 ms 6,816 KB
testcase_70	AC	2 ms 6,816 KB
testcase_71	AC	2 ms 6,820 KB
testcase_72	AC	2 ms 6,820 KB
testcase_73	AC	2 ms 6,820 KB
testcase_74	AC	2 ms 6,816 KB
testcase_75	AC	2 ms 6,816 KB
testcase_76	AC	2 ms 6,820 KB
testcase_77	AC	2 ms 6,816 KB
testcase_78	AC	2 ms 6,820 KB
testcase_79	AC	2 ms 6,816 KB
testcase_80	AC	3 ms 6,816 KB
testcase_81	AC	2 ms 6,816 KB
testcase_82	AC	2 ms 6,820 KB
testcase_83	AC	2 ms 6,820 KB
testcase_84	AC	2 ms 6,820 KB
testcase_85	AC	2 ms 6,820 KB
testcase_86	AC	2 ms 6,824 KB
testcase_87	AC	3 ms 6,816 KB
testcase_88	AC	2 ms 6,816 KB
testcase_89	AC	2 ms 6,816 KB
testcase_90	AC	4 ms 6,816 KB
testcase_91	AC	2 ms 6,816 KB
testcase_92	AC	2 ms 6,816 KB
testcase_93	AC	2 ms 6,816 KB
testcase_94	AC	1 ms 6,820 KB
testcase_95	AC	2 ms 6,816 KB
testcase_96	AC	2 ms 6,816 KB
testcase_97	AC	2 ms 6,820 KB
testcase_98	AC	2 ms 6,816 KB
testcase_99	AC	3 ms 6,816 KB
testcase_100	AC	2 ms 6,816 KB
testcase_101	AC	6 ms 6,816 KB
testcase_102	AC	1 ms 6,816 KB
testcase_103	AC	2 ms 6,820 KB
testcase_104	AC	2 ms 6,816 KB
testcase_105	AC	2 ms 6,820 KB
testcase_106	AC	2 ms 6,820 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <iostream>
#include <set>
#include <vector>
#include <cassert>
using namespace std;

typedef __int128 Int;

Int mulMod(Int x, Int k, Int m) {
  if (k == 0)     return 0;
  if (k % 2 == 0) return mulMod((x+x) % m, k/2, m);
  else            return (x+mulMod(x, k-1, m)) % m;
}
Int powMod(Int x, Int k, Int m) {
  if (k == 0)     return 1;
  if (k % 2 == 0) return powMod(mulMod(x, x, m) % m, k/2, m);
  else            return mulMod(x, powMod(x, k-1, m), m);
}
bool suspect(Int a, int s, Int d, Int n) {
  Int x = powMod(a, d, n);
  if (x == 1) return true;
  for (int r = 0; r < s; ++r) {
    if (x == n - 1) return true;
    x = mulMod(x, x, n);
  }
  return false;
}
bool isPrime(Int n) {
  if (n <= 1 || (n > 2 && n % 2 == 0)) return false;
  int test[] = {2,3,5,7,11,13,17,19,23,27,29,31,-1};
  Int d = n - 1, s = 0;
  while (d % 2 == 0) ++s, d /= 2;
  for (int i = 0; test[i] < n && test[i] != -1; ++i)
    if (!suspect(test[i], s, d, n)) return false;
  return true;
}

vector<int> eratosthenes_sieve(int n){
	vector<int> res(n+1);
	if(n<2) return res;
	res[2]=1;
	for(int i=3;i<=n;i+=2){
		if(i*i<=n && !res[i]) for(int j=i+i+i;j<=n;j+=i+i) res[j]=1;
		res[i]=!res[i];
	}
	return res;
}

string s;
Int n;
vector<int> p;
int solve_naive(Int n){
    for(int i=3;;i++){
        vector<int> ok(2*n);
        ok[0] = 1;
        for(int _=9;_--;)for(int j=0;j<n;j++)if(ok[j] && !p[j+1]){
            if(j/i == (j+1)/i) ok[j+1] = 1;
            if(j>=i) ok[j-i] = 1;
            if(j%i) ok[j-1] = 1;
            ok[j+i] = 1;
        }
        if(ok[n-1]){
            return i;
        }
    }
}

int solve(Int n){
	for(int i=3;;i++){
		set<int> visited({1}), col({1}), q({1});
		int f = 0;
		while(!q.empty() && (col.size() < i || i % 2)){
			int j = *q.begin(); q.erase(q.begin());
			if(j == n){
				f = 1;
				break;
			}
			vector<int> d;
			if(j%i != 0) d.push_back(j+1);
			if(i < j) d.push_back(j-i);
			if(j%i != 1) d.push_back(j-1);
			d.push_back(j+i);
			for(int v : d) if(v <= n && !visited.count(v) && !p[v]){
				visited.insert(v);
				col.insert(v%i);
				q.insert(v);
			}
		}
		
		if(!f && col.size() == i && i % 2 == 0){
			f = (n % i != 1) || !isPrime(n - i);
		}
		if(f){
//			if(i != solve_naive(n)){
//				cout << (int)n << ": " << i << ", " << solve_naive(n) << endl;
//				exit(0);
//			}
			return i;
		}
	}
}

int main(int argc, char *argv[]){
	p = eratosthenes_sieve(100000);
/*	for(int i=4;;i++)if(!p[i]){
		solve(i);
	}
*/
	cin >> s;
	for(char c : s) n = n*10 + c - '0';
	cout << solve(n) << endl;
	return 0;
}

yukicoder

結果

テストケース

ソースコード