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

• 問題: No.36 素数が嫌い！
• ユーザー: tsuchinaga
• 提出日時: 2019-03-15 15:30:11
• 言語: Go (1.22.1)
• 結果: WA
• 実行時間: -
• コード長: 691 bytes
• コンパイル時間: 10,999 ms
• コンパイル使用メモリ: 213,960 KB
• 実行使用メモリ: 80,736 KB
• 最終ジャッジ日時: 2023-09-14 12:17:56
• 合計ジャッジ時間: 16,151 ms
• ジャッジサーバーID （参考情報）: judge12 / judge13

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	154 ms; 39,092 KB
testcase_01	AC	281 ms; 62,016 KB
testcase_02	AC	1 ms; 4,380 KB
testcase_03	AC	2 ms; 4,376 KB
testcase_04	AC	1 ms; 4,380 KB
testcase_05	WA	-
testcase_06	WA	-
testcase_07	WA	-
testcase_08	AC	1 ms; 4,380 KB
testcase_09	AC	1 ms; 4,376 KB
testcase_10	AC	1 ms; 4,376 KB
testcase_11	WA	-
testcase_12	WA	-
testcase_13	AC	362 ms; 80,736 KB
testcase_14	AC	162 ms; 51,576 KB
testcase_15	AC	1 ms; 4,380 KB
testcase_16	WA	-
testcase_17	AC	1 ms; 4,376 KB
testcase_18	AC	2 ms; 4,376 KB
testcase_19	AC	110 ms; 32,848 KB
testcase_20	AC	353 ms; 80,736 KB
testcase_21	AC	252 ms; 62,000 KB
testcase_22	AC	267 ms; 64,080 KB
testcase_23	AC	138 ms; 39,104 KB
testcase_24	AC	156 ms; 43,264 KB
testcase_25	AC	167 ms; 43,252 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")
	}
}