#543186 (Go) No.1200 お菓子配り-3

• 問題: No.1200 お菓子配り-3
• ユーザー: ccppjsrb
• 提出日時: 2020-08-29 17:21:22
• 言語: Go (1.22.1)
• 結果: WA
• 実行時間: -
• コード長: 2,358 bytes
• コンパイル時間: 13,682 ms
• コンパイル使用メモリ: 221,392 KB
• 実行使用メモリ: 79,304 KB
• 最終ジャッジ日時: 2024-11-14 17:36:42
• 合計ジャッジ時間: 26,875 ms
• ジャッジサーバーID （参考情報）: judge2 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	WA	-
testcase_01	AC	3 ms; 6,816 KB
testcase_02	AC	3 ms; 6,816 KB
testcase_03	AC	3 ms; 6,816 KB
testcase_04	WA	-
testcase_05	WA	-
testcase_06	WA	-
testcase_07	WA	-
testcase_08	WA	-
testcase_09	WA	-
testcase_10	WA	-
testcase_11	WA	-
testcase_12	WA	-
testcase_13	WA	-
testcase_14	WA	-
testcase_15	WA	-
testcase_16	WA	-
testcase_17	WA	-
testcase_18	WA	-
testcase_19	WA	-
testcase_20	WA	-
testcase_21	WA	-
testcase_22	WA	-
testcase_23	WA	-
testcase_24	WA	-
testcase_25	WA	-
testcase_26	WA	-
testcase_27	AC	3 ms; 6,820 KB
testcase_28	AC	1,116 ms; 6,820 KB
testcase_29	AC	1,129 ms; 7,712 KB
testcase_30	AC	1,125 ms; 7,700 KB
testcase_31	WA	-
testcase_32	WA	-

ソースコード

package main

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

func configure(scanner *bufio.Scanner) {
	scanner.Split(bufio.ScanWords)
	scanner.Buffer(make([]byte, 1000005), 1000005)
}
func getNextString(scanner *bufio.Scanner) string {
	scanned := scanner.Scan()
	if !scanned {
		panic("scan failed")
	}
	return scanner.Text()
}
func getNextInt(scanner *bufio.Scanner) int {
	i, _ := strconv.Atoi(getNextString(scanner))
	return i
}
func getNextInt64(scanner *bufio.Scanner) int64 {
	i, _ := strconv.ParseInt(getNextString(scanner), 10, 64)
	return i
}
func getNextFloat64(scanner *bufio.Scanner) float64 {
	i, _ := strconv.ParseFloat(getNextString(scanner), 64)
	return i
}
func main() {
	fp := os.Stdin
	wfp := os.Stdout
	extra := 0
	if os.Getenv("I") == "IronMan" {
		fp, _ = os.Open(os.Getenv("END_GAME"))
		extra = 100
	}
	scanner := bufio.NewScanner(fp)
	configure(scanner)
	writer := bufio.NewWriter(wfp)
	defer func() {
		r := recover()
		if r != nil {
			fmt.Fprintln(writer, r)
		}
		writer.Flush()
	}()
	solve(scanner, writer)
	for i := 0; i < extra; i++ {
		fmt.Fprintln(writer, "-----------------------------------")
		solve(scanner, writer)
	}
}
func solve(scanner *bufio.Scanner, writer *bufio.Writer) {
	eras := make([]int, 100000)
	usa := make([]int, 0)
	for i := 2; i < 100000; i++ {
		if eras[i] == 0 {
			usa = append(usa, i)
			for j := i; j < 100000; j += i {
				eras[j] = i
			}
		}
	}
	var testcase func()
	testcase = func() {
		var ans int64
		x := getNextInt(scanner)
		y := getNextInt(scanner)
		if x < y {
			x, y = y, x
		}
		p := x + y
		m := x - y
		dd := make([]int, 0)
		dd = append(dd, 1)
		for i := 0; i < len(usa); i++ {
			pp := p
			for pp > 1 && pp%usa[i] == 0 {
				ddd := make([]int, 0)
				for _, d := range dd {
					ddd = append(ddd, d*usa[i])
				}
				for _, d := range ddd {
					dd = append(dd, d)
				}
				pp /= usa[i]
			}
		}
		for _, d := range dd {
			ans += calc(p, m, d)
		}
		fmt.Fprintln(writer, ans)
	}
	s := getNextInt(scanner)
	for s > 0 {
		testcase()
		s--
	}
}

func calc(p, m, a int) int64 {
	if a == 1 {
		return 0
	}
	if a == 2 {
		if m == 0 {
			return int64(p / a)
		}
		return 0
	}
	bpc := p / a
	if m%(a-2) != 0 {
		return 0
	}
	bmc := m / (a - 2)
	b2 := bpc + bmc
	if b2 < 2 {
		return 0
	}
	if b2&1 == 1 {
		return 0
	}
	b := b2 >> 1
	c := b - bmc
	if c < 1 {
		return 0
	}
	return 1
}