#774627 (Go) No.1747 Many Formulae 2

• 問題: No.1747 Many Formulae 2
• ユーザー: Hima
• 提出日時: 2022-06-30 21:58:18
• 言語: Go (1.22.1)
• 結果: AC
• 実行時間: 2 ms / 2,000 ms
• コード長: 1,229 bytes
• コンパイル時間: 12,650 ms
• コンパイル使用メモリ: 224,260 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-05-03 18:28:39
• 合計ジャッジ時間: 12,407 ms
• ジャッジサーバーID （参考情報）: judge3 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 5,248 KB
testcase_01	AC	1 ms; 5,248 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	2 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	2 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	1 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

ソースコード

package main

import (
	"bufio"
	"fmt"
	"os"
	"strconv"
	"unsafe"
)

var sc = bufio.NewScanner(os.Stdin)
var out = bufio.NewWriter(os.Stdout)

func isPrime(x int) bool {
	if x == 1 {
		return false
	}
	for i := 2; i*i <= x; i++ {
		if x%i == 0 {
			return false
		}
	}
	return true
}

func clone(s string) string {
	b := make([]byte, len(s))
	copy(b, s)
	return *(*string)(unsafe.Pointer(&b))
}

func solve(s string) int {
	n := len(s)
	var ans int
	//answers := make(map[int]struct{})
	for mask := 0; mask < 1<<(n-1); mask++ {
		//fmt.Printf("%b ", mask)
		cs := clone(s)
		cm := mask
		sum := 0
		for i := 0; i < len(cs)-1; i++ {
			if cm>>i&1 > 0 {
				//fmt.Printf("%s %s ", cs[:i+1], cs[i+1:])
				x, _ := strconv.Atoi(cs[:i+1])
				sum += x
				cs = cs[i+1:]
				cm >>= i + 1
				i = -1
			}
		}
		x, _ := strconv.Atoi(cs)
		sum += x
		//fmt.Println(mask, sum)
		if isPrime(sum) {
			ans++
		}
		//fmt.Println()
	}
	return ans
}

func main() {
	buf := make([]byte, 1024*1024)
	sc.Buffer(buf, bufio.MaxScanTokenSize)
	sc.Split(bufio.ScanWords)

	s := nextString()
	ans := solve(s)
	PrintInt(ans)
}

func nextString() string {
	sc.Scan()
	return sc.Text()
}

func PrintInt(x int) {
	defer out.Flush()
	fmt.Fprintln(out, x)
}