#142667 (Go) No.115 遠足のおやつ

• 問題: No.115 遠足のおやつ
• ユーザー: yuki2006
• 提出日時: 2017-01-03 16:09:00
• 言語: Go (1.22.1)
• 結果: WA
• 実行時間: -
• コード長: 1,070 bytes
• コンパイル時間: 10,081 ms
• コンパイル使用メモリ: 237,788 KB
• 実行使用メモリ: 6,944 KB
• 最終ジャッジ日時: 2024-05-09 16:01:36
• 合計ジャッジ時間: 11,109 ms
• ジャッジサーバーID （参考情報）: judge4 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	WA	-
testcase_01	AC	1 ms; 5,376 KB
testcase_02	AC	1 ms; 5,376 KB
testcase_03	AC	1 ms; 5,376 KB
testcase_04	AC	1 ms; 5,376 KB
testcase_05	AC	1 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	1 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	WA	-
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
testcase_19	AC	1 ms; 5,376 KB
testcase_20	AC	1 ms; 5,376 KB
testcase_21	AC	1 ms; 5,376 KB
testcase_22	AC	1 ms; 5,376 KB
testcase_23	AC	1 ms; 5,376 KB
testcase_24	AC	1 ms; 5,376 KB
testcase_25	AC	1 ms; 5,376 KB
testcase_26	WA	-
testcase_27	AC	1 ms; 5,376 KB
testcase_28	WA	-
testcase_29	AC	1 ms; 5,376 KB
testcase_30	AC	1 ms; 5,376 KB
testcase_31	AC	1 ms; 5,376 KB
testcase_32	AC	1 ms; 5,376 KB
testcase_33	AC	1 ms; 5,376 KB
testcase_34	AC	1 ms; 5,376 KB
testcase_35	AC	1 ms; 5,376 KB
testcase_36	AC	1 ms; 5,376 KB
testcase_37	AC	1 ms; 5,376 KB
testcase_38	AC	1 ms; 5,376 KB
testcase_39	AC	1 ms; 5,376 KB
testcase_40	AC	1 ms; 5,376 KB
testcase_41	AC	1 ms; 5,376 KB
testcase_42	AC	1 ms; 5,376 KB

ソースコード

package main

import (
	"fmt"
	"strconv"
	"strings"
)

func main() {
	var N, D, K int
	fmt.Scan(&N, &D, &K)
	total := 0
	lastChoice := 0
	for k := 1; k <= K && lastChoice <= N; {
		total += (lastChoice + 1)
		if total > D {
			break
		}
		lastChoice++
		k++
	}
	diff := D - total
	if diff < 0 {
		fmt.Println(-1)
		return
	}
	endNum := N + 1
	midChoice := 0
	for i := 0; i < K; i++ {
		if diff - (endNum - 1) >= 0 {
			diff -= (endNum - 1) - lastChoice
			lastChoice--
			endNum--
		} else if diff > 0 {
			midChoice = lastChoice + diff
			lastChoice--
			diff = 0
		}
	}
	if diff > 0 {
		fmt.Println(-1)
		return
	}

	ans := make([]int, 0)
	for i := 1; i <= lastChoice; i++ {
		ans = append(ans, i)
	}
	if midChoice > 0 {
		ans = append(ans, midChoice)
	}
	for k := endNum; k <= N; k++ {
		ans = append(ans, k)
	}
	PrintI(ans...)
}

func PrintI(args ...int) {
	fmt.Println(strings.Join(mapToString(args), " "))
}

func mapToString(arr []int) []string {
	ret := make([]string, len(arr))
	for i := 0; i < len(arr); i++ {
		ret[i] = strconv.Itoa(arr[i])
	}
	return ret
}