CodeIQ：バスの料金を計算しよう（ややリアル編）

問題の概要

私のプログラム

Pythonで解答しています。

import fileinput
import math

#特別料金を計算する
def calcSpecialPrice(i):
	return 10 * math.ceil(i*56/1000)

# 子供料金を計算する
def CalcChildPrice(lst):
	ret = []
	for i in lst:
		n = 10 * math.ceil((i // 10) / 2)
		ret.append(n)

	return ret

# 入力文字列をパースする
def parseLine(str):
	ret = []
	price, passenger = str.split(":")
	ret.append(passenger.split(","))
	ret.append(list(map(int, price.split(","))))
	ret.append(CalcChildPrice(ret[1]))
	return ret

# 乗客の大人、子供、幼児の数を数える
def countPassenger(lst):
	p = {"An":0, "Ax":0, "Ap":0, "Cn":0, "Cx":0, "Cp":0, "In":0, "Ix":0, "Ip":0}
	for k,v in p.items():
		p[k] = len(list(filter(lambda a:a==k, lst)))
	return p

# 1日乗車券を考慮して大人、子供、幼児の値段表を計算する
def calcPriceDic(ap, cp):
	price_dic = {
		"An": sum(ap),
		"Ax": sum(list(map(calcSpecialPrice, ap))),
		"Ap": 0,
		"Cn": sum(cp),
		"Cx": sum(list(map(calcSpecialPrice, cp))),
		"Cp": 0,
		"In": sum(cp),
		"Ix": sum(list(map(calcSpecialPrice, cp))),
		"Ip": 0,
	}

	oneday_pass = {
		"An": 910,
		"Ax": 510,
		"Ap": 0,
		"Cn": 460,
		"Cx": 260,
		"Cp": 0,
		"In": 460,
		"Ix": 260,
		"Ip": 0,
	}

	for k,v in price_dic.items():
		if price_dic[k] > oneday_pass[k]:
			price_dic[k] = oneday_pass[k]

	return price_dic

# 金額計算
# p 乗客を数えたもの
# ap 大人の金額
# cp 子供の金額
def calcTotal(p, ap, cp):
	price_dic = calcPriceDic(ap, cp)

	# 通常料金の幼児を相殺する
	cnt_i =  p["In"]
	cnt_a = 2*(p["An"] + p["Ax"] + p["Ap"])

	if (cnt_i - cnt_a) < 0:
		cnt_i = p["In"] + p["Ix"]
		p["In"] = 0

		if (cnt_i - cnt_a)<0:
			p["Ix"] = 0
		else:
			p["Ix"] = cnt_i - cnt_a
	else:
		p["In"] = (cnt_i - cnt_a)

	ret = 0
	for k, v in p.items():
		ret += v * price_dic[k]

	return ret

#==============================================================================
# main
#------------------------------------------------------------------------------

for line in fileinput.input():
	if not line.strip():
		continue

	PassengerList, PliceListA, PliceListC = parseLine(line.strip())
	PassengerCount = countPassenger(PassengerList)
	total = calcTotal(PassengerCount, PliceListA, PliceListC)
	print(total)

解説

初級編の問題があるのでそちらをやっていれば基本的な処理の手順は同じです。ただし、条件がかなりややこしくなっています。これをいかにシンプルに扱えるかがポイントになります。
もう一点、浮動小数点の誤差もポイントになります。

雑感

実際の業務にありそうな問題とのことでしたが、確かに条件によってこういうこまごまとした処理が必要なことはよくあります。こういう時は前準備でデータを整えて可能な限り単純なループで処理できるようにするのがポイントと思っています。
このプログラムの場合は最終的な計算は88〜89行目のループですが非常に単純です。もし、ここに条件文が含まれていたりするとデバッグが大変ですし、修正も大変です。この問題程度の複雑さなら大したことはないかもしれませんが、実際の業務で作るプログラムの場合、何か修正したら別のところに問題が出ることにすらなり得ます。