結果

問題 No.308 素数は通れません
ユーザー yosupotyosupot
提出日時 2015-12-01 23:35:24
言語 Go
(1.22.1)
結果
AC  
実行時間 7 ms / 1,000 ms
コード長 2,374 bytes
コンパイル時間 12,575 ms
コンパイル使用メモリ 223,088 KB
実行使用メモリ 6,016 KB
最終ジャッジ日時 2024-05-05 18:30:46
合計ジャッジ時間 14,680 ms
ジャッジサーバーID
(参考情報)
judge5 / judge3
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 1 ms
5,376 KB
testcase_02 AC 1 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 1 ms
5,376 KB
testcase_05 AC 1 ms
5,376 KB
testcase_06 AC 1 ms
5,376 KB
testcase_07 AC 1 ms
5,376 KB
testcase_08 AC 1 ms
5,376 KB
testcase_09 AC 1 ms
5,376 KB
testcase_10 AC 1 ms
5,376 KB
testcase_11 AC 1 ms
5,376 KB
testcase_12 AC 1 ms
5,376 KB
testcase_13 AC 1 ms
5,376 KB
testcase_14 AC 2 ms
5,376 KB
testcase_15 AC 1 ms
5,376 KB
testcase_16 AC 1 ms
5,376 KB
testcase_17 AC 1 ms
5,376 KB
testcase_18 AC 1 ms
5,376 KB
testcase_19 AC 1 ms
5,376 KB
testcase_20 AC 1 ms
5,376 KB
testcase_21 AC 1 ms
5,376 KB
testcase_22 AC 1 ms
5,376 KB
testcase_23 AC 1 ms
5,376 KB
testcase_24 AC 1 ms
5,376 KB
testcase_25 AC 1 ms
5,376 KB
testcase_26 AC 2 ms
5,376 KB
testcase_27 AC 2 ms
5,376 KB
testcase_28 AC 2 ms
5,376 KB
testcase_29 AC 1 ms
5,376 KB
testcase_30 AC 2 ms
5,376 KB
testcase_31 AC 1 ms
5,376 KB
testcase_32 AC 2 ms
5,376 KB
testcase_33 AC 1 ms
5,376 KB
testcase_34 AC 1 ms
5,376 KB
testcase_35 AC 1 ms
5,376 KB
testcase_36 AC 2 ms
5,376 KB
testcase_37 AC 2 ms
5,376 KB
testcase_38 AC 2 ms
5,376 KB
testcase_39 AC 2 ms
5,376 KB
testcase_40 AC 3 ms
5,376 KB
testcase_41 AC 3 ms
5,376 KB
testcase_42 AC 2 ms
5,376 KB
testcase_43 AC 2 ms
5,376 KB
testcase_44 AC 3 ms
5,376 KB
testcase_45 AC 3 ms
5,376 KB
testcase_46 AC 3 ms
5,376 KB
testcase_47 AC 3 ms
5,376 KB
testcase_48 AC 4 ms
5,376 KB
testcase_49 AC 4 ms
5,376 KB
testcase_50 AC 4 ms
5,376 KB
testcase_51 AC 5 ms
5,376 KB
testcase_52 AC 6 ms
5,760 KB
testcase_53 AC 7 ms
5,760 KB
testcase_54 AC 7 ms
5,888 KB
testcase_55 AC 7 ms
6,016 KB
testcase_56 AC 2 ms
5,376 KB
testcase_57 AC 1 ms
5,376 KB
testcase_58 AC 2 ms
5,376 KB
testcase_59 AC 2 ms
5,376 KB
testcase_60 AC 1 ms
5,376 KB
testcase_61 AC 1 ms
5,376 KB
testcase_62 AC 1 ms
5,376 KB
testcase_63 AC 2 ms
5,376 KB
testcase_64 AC 1 ms
5,376 KB
testcase_65 AC 2 ms
5,376 KB
testcase_66 AC 2 ms
5,376 KB
testcase_67 AC 1 ms
5,376 KB
testcase_68 AC 2 ms
5,376 KB
testcase_69 AC 1 ms
5,376 KB
testcase_70 AC 2 ms
5,376 KB
testcase_71 AC 1 ms
5,376 KB
testcase_72 AC 1 ms
5,376 KB
testcase_73 AC 2 ms
5,376 KB
testcase_74 AC 2 ms
5,376 KB
testcase_75 AC 2 ms
5,376 KB
testcase_76 AC 2 ms
5,376 KB
testcase_77 AC 2 ms
5,376 KB
testcase_78 AC 2 ms
5,376 KB
testcase_79 AC 1 ms
5,376 KB
testcase_80 AC 2 ms
5,376 KB
testcase_81 AC 2 ms
5,376 KB
testcase_82 AC 2 ms
5,376 KB
testcase_83 AC 1 ms
5,376 KB
testcase_84 AC 2 ms
5,376 KB
testcase_85 AC 2 ms
5,376 KB
testcase_86 AC 2 ms
5,376 KB
testcase_87 AC 3 ms
5,376 KB
testcase_88 AC 2 ms
5,376 KB
testcase_89 AC 2 ms
5,376 KB
testcase_90 AC 2 ms
5,376 KB
testcase_91 AC 3 ms
5,376 KB
testcase_92 AC 2 ms
5,376 KB
testcase_93 AC 2 ms
5,376 KB
testcase_94 AC 2 ms
5,376 KB
testcase_95 AC 2 ms
5,376 KB
testcase_96 AC 2 ms
5,376 KB
testcase_97 AC 1 ms
5,376 KB
testcase_98 AC 2 ms
5,376 KB
testcase_99 AC 2 ms
5,376 KB
testcase_100 AC 2 ms
5,376 KB
testcase_101 AC 5 ms
5,376 KB
testcase_102 AC 2 ms
5,376 KB
testcase_103 AC 2 ms
5,376 KB
testcase_104 AC 2 ms
5,376 KB
testcase_105 AC 2 ms
5,376 KB
testcase_106 AC 1 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

package main

import (
	"fmt"
	"math/big"
	"os"
)

type UnionFind struct {
	n  int
	ig []int
	gi [][]int
}

func MakeUF(n int) *UnionFind {
	uf := new(UnionFind)
	uf.n = n
	uf.ig = make([]int, n)
	uf.gi = make([][]int, n)
	for i := 0; i < n; i++ {
		uf.ig[i] = i
		uf.gi[i] = []int{i}
	}
	return uf
}

func (uf *UnionFind) merge(a, b int) {
	a = uf.ig[a]
	b = uf.ig[b]
	if a == b {
		return
	}
	if len(uf.gi[a]) < len(uf.gi[b]) {
		tmp := a
		a = b
		b = tmp
	}
	for _, c := range uf.gi[b] {

		uf.ig[c] = a
	}
	uf.gi[a] = append(uf.gi[a], uf.gi[b]...)
}

func (uf *UnionFind) same(a, b int) bool {
	return uf.ig[a] == uf.ig[b]
}

const PS = 50000

var pr [PS]bool

func init() {
	for i := 2; i < PS; i++ {
		pr[i] = true
	}
	for i := 2; i < PS; i++ {
		if pr[i] == false {
			continue
		}
		for j := 2 * i; j < PS; j += i {
			pr[j] = false
		}
	}
}

func solve1(n int) int {
	for i := 2; i <= n; i++ {
		uf := MakeUF(n + 1)
		for j := 1; j <= n; j++ {
			if j%i != 1 {
				if !pr[j] && !pr[j-1] {
					uf.merge(j, j-1)
				}
			}
			if i < j {
				if !pr[j] && !pr[j-i] {
					uf.merge(j, j-i)
				}
			}
		}
		if uf.same(1, n) {
			return i
		}
	}
	os.Exit(1)
	return -1
}

func solve2(n *big.Int) int {
	for i := 2; i < 1000; i++ {
		//		const B = 1000
		B := i * 5
		{
			uf := MakeUF(B + 1)
			for j := 1; j <= B; j++ {
				if j%i != 1 {
					if !pr[j] && !pr[j-1] {
						uf.merge(j, j-1)
					}
				}
				if i < j {
					if !pr[j] && !pr[j-i] {
						uf.merge(j, j-i)
					}
				}
			}
			if !uf.same(1, B/i*i) {
				continue
			}
		}
		{
			uf := MakeUF(B + 1)
			ii := big.NewInt(int64(i))
			bpr := make([]bool, B+1)
			for jj := 0; jj <= B; jj++ {
				j := new(big.Int)
				j.Set(n).Sub(j, big.NewInt(int64(B-jj)))
				bpr[jj] = j.ProbablyPrime(30)
			}
			k := new(big.Int)
			jm := int(k.Set(n).Mod(k, ii).Int64())
			for jj := 1; jj <= B; jj++ {
				if ((jm-(B-jj))%i+i)%i != 1 {
					if !bpr[jj] && !bpr[jj-1] {
						uf.merge(jj, jj-1)
					}
				}
				if i <= jj {
					if !bpr[jj] && !bpr[jj-i] {
						uf.merge(jj, jj-i)
					}
				}
			}
			nn := new(big.Int)
			nn.Set(n)
			if uf.same(B-int(nn.Mod(nn, big.NewInt(int64(i))).Int64()), B) {
				return i
			}
		}
	}
	os.Exit(1)
	return -1
}

func main() {
	b := new(big.Int)
	fmt.Scan(b)
	if b.Cmp(big.NewInt(20000)) == -1 {
		fmt.Println(solve1(int(b.Int64())))
	} else {
		fmt.Println(solve2(b))
	}
}
0