#322807 (Go) No.36 素数が嫌い！

• 問題: No.36 素数が嫌い！
• ユーザー: tsuchinaga
• 提出日時: 2019-03-15 15:30:11
• 言語: Go (1.22.1)
• 結果: WA
• 実行時間: -
• コード長: 691 bytes
• コンパイル時間: 11,292 ms
• コンパイル使用メモリ: 220,220 KB
• 実行使用メモリ: 77,440 KB
• 最終ジャッジ日時: 2024-07-01 19:44:08
• 合計ジャッジ時間: 16,874 ms
• ジャッジサーバーID （参考情報）: judge5 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	160 ms; 36,864 KB
testcase_01	AC	287 ms; 60,672 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	1 ms; 5,376 KB
testcase_04	AC	1 ms; 5,376 KB
testcase_05	WA	-
testcase_06	WA	-
testcase_07	WA	-
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	WA	-
testcase_12	WA	-
testcase_13	AC	393 ms; 77,184 KB
testcase_14	AC	185 ms; 48,768 KB
testcase_15	AC	1 ms; 5,376 KB
testcase_16	WA	-
testcase_17	AC	1 ms; 5,376 KB
testcase_18	AC	1 ms; 5,376 KB
testcase_19	AC	128 ms; 31,360 KB
testcase_20	AC	392 ms; 77,184 KB
testcase_21	AC	293 ms; 59,520 KB
testcase_22	AC	306 ms; 61,312 KB
testcase_23	AC	171 ms; 37,504 KB
testcase_24	AC	176 ms; 41,600 KB
testcase_25	AC	190 ms; 40,960 KB
testcase_26	WA	-
testcase_27	WA	-
testcase_28	WA	-
testcase_29	WA	-

ソースコード

package main

import (
	"fmt"
	"math"
)

func main() {
	var n int
	_, _ = fmt.Scan(&n)
	m := n

	ans := 0
	p := make(map[int]int, 0)
	nums := make([]int, int(math.Ceil(math.Sqrt(float64(n))))+1)
	nums[0], nums[1] = 1, 1

	for i, v := range nums {
		if i == 0 || i == 1 || v != 0 {
			continue
		}

		for {
			if n%i != 0 {
				break
			}
			p[i]++
			ans++
			n /= i
		}

		if n == 1 {
			break
		}

		for j := i + i; j < len(nums)-1; j += i {
			nums[j] = 1
		}
	}

	// fmt.Println(p)
	if ans >= 3 {
		fmt.Println("YES")
	} else {
		a := 1
		for k, v := range p {
			a *= int(math.Pow(float64(k), float64(v)))
		}
		if a != m {
			fmt.Println("YES")
			return
		}

		fmt.Println("NO")
	}
}