#9344 (Go) No.112 ややこしい鶴亀算

• 問題: No.112 ややこしい鶴亀算
• ユーザー: tyochiai
• 提出日時: 2015-01-20 03:16:48
• 言語: Go (1.22.1)
• 結果: AC
• 実行時間: 2 ms / 5,000 ms
• コード長: 855 bytes
• コンパイル時間: 10,409 ms
• コンパイル使用メモリ: 238,172 KB
• 実行使用メモリ: 6,944 KB
• 最終ジャッジ日時: 2024-04-18 09:47:59
• 合計ジャッジ時間: 11,511 ms
• ジャッジサーバーID （参考情報）: judge3 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 6,812 KB
testcase_01	AC	1 ms; 6,816 KB
testcase_02	AC	2 ms; 6,940 KB
testcase_03	AC	2 ms; 6,940 KB
testcase_04	AC	2 ms; 6,940 KB
testcase_05	AC	2 ms; 6,940 KB
testcase_06	AC	1 ms; 6,940 KB
testcase_07	AC	1 ms; 6,940 KB
testcase_08	AC	2 ms; 6,944 KB
testcase_09	AC	1 ms; 6,944 KB
testcase_10	AC	1 ms; 6,944 KB
testcase_11	AC	1 ms; 6,940 KB
testcase_12	AC	2 ms; 6,940 KB
testcase_13	AC	1 ms; 6,940 KB
testcase_14	AC	1 ms; 6,944 KB
testcase_15	AC	1 ms; 6,944 KB
testcase_16	AC	1 ms; 6,940 KB
testcase_17	AC	2 ms; 6,944 KB
testcase_18	AC	2 ms; 6,944 KB
testcase_19	AC	2 ms; 6,944 KB
testcase_20	AC	1 ms; 6,940 KB
testcase_21	AC	2 ms; 6,944 KB
testcase_22	AC	1 ms; 6,940 KB
testcase_23	AC	2 ms; 6,944 KB
testcase_24	AC	2 ms; 6,944 KB

ソースコード

package main

import (
	"fmt"
	"reflect"
)

func Sum(a []int) int {
	ret := 0
	for i := 0; i < len(a); i++ {
		ret += a[i]
	}
	return ret
}

func resolve(N int, A []int) string {
	total := 0
	B := make([]int, N)
	C := make([]int, N)
	for _, a := range []int{2, 4} {
		t := A[0] + a
		for i := 0; i < N; i++ {
			B[i] = t - A[i]
		}
		for i := 0; i < N; i++ {
			C[i] = Sum(B[:i]) + Sum(B[i+1:])
		}
		if reflect.DeepEqual(A, C) == false {
			continue
		}
		total = t
		break
	}
	if total == 0 {
		return "???"
	}

	turu := 0
	kame := 0
	for i := 0; i < N; i++ {
		switch total - A[i] {
		case 2:
			turu += 1
		case 4:
			kame += 1
		}
	}
	return fmt.Sprintf("%d %d", turu, kame)
}

func main() {
	var N int

	fmt.Scanf("%d\n", &N)
	A := make([]int, N)
	for i := 0; i < N; i++ {
		fmt.Scanf("%d", &A[i])
	}
	fmt.Scanf("\n")

	fmt.Println(resolve(N, A))
}