CF230B

Author: EndlessCheng

main/200-299/230B.go

@@ -0,0 +1,37 @@
+package main
+
+import (
+	"bufio"
+	. "fmt"
+	"io"
+	"math"
+)
+
+// github.com/EndlessCheng/codeforces-go
+func CF230B(_r io.Reader, _w io.Writer) {
+	in := bufio.NewReader(_r)
+	out := bufio.NewWriter(_w)
+	defer out.Flush()
+	const mx int = 1e6
+	np := [mx + 1]bool{1: true}
+	for i := 2; i <= mx; i++ {
+		if !np[i] {
+			for j := 2 * i; j <= mx; j += i {
+				np[j] = true
+			}
+		}
+	}
+
+	var n, v int64
+	for Fscan(in, &n); n > 0; n-- {
+		Fscan(in, &v)
+		rt := int64(math.Sqrt(float64(v)))
+		if rt*rt == v && !np[rt] {
+			Fprintln(out, "YES")
+		} else {
+			Fprintln(out, "NO")
+		}
+	}
+}
+
+//func main() { CF230B(os.Stdin, os.Stdout) }

main/200-299/230B_test.go

@@ -0,0 +1,21 @@
+package main
+
+import (
+	"github.com/EndlessCheng/codeforces-go/main/testutil"
+	"testing"
+)
+
+// https://codeforces.com/problemset/problem/230/B
+// https://codeforces.com/problemset/status/230/problem/B
+func TestCF230B(t *testing.T) {
+	// just copy from website
+	rawText := `
+inputCopy
+4
+4 5 6
+outputCopy
+YES
+NO
+NO`
+	testutil.AssertEqualCase(t, rawText, 0, CF230B)
+}