"第30回オフラインリアルタイムどう書くの問題" 解いてみた

概要

Nabetaniさんの「とある世界のタクシー料金」問題をRubyで解いてみました。
元記事はこちらから
http://qiita.com/Nabetani/items/c70417d384720a3339d6

実装

Rubyで解いてますが、解き方自体はどの言語でも良さそうな感じの実装です。
(どちらの都市でも200m毎に料金加算なので使ってない定数は放置)

taxi_fee.rb

# encoding: sjis
# 
# とある世界のタクシー料金を計算する
# Author:: lithtle
# Created_At:: 2015-04-21
# Link:: http://nabetani.sakura.ne.jp/hena/ord30taxi/

# 都市の地点と経路
CITY = {
	enlightened: [:A, :B, :C, :AB, :AC, :BG, :CF, :BC],
	registance: [:D, :E, :F, :G, :AD, :CD, :DE, :DF, :EG, :FG]
}.freeze

# 各地点間の距離
DISTANTCE = {
	AB: 1090,
	AC: 180,
	AD: 540,
	BC: 960,
	BG: 1270,
	CD: 400,
	CF: 200,
	DE: 720,
	DF: 510,
	EG: 1050,
	FG: 230
}.freeze

# 各都市でのタクシー設定
TAXI_SETTING = {
	enlightened: {min_dist: 995, min_fee: 400, per_meter: 200, per_fee: 60},
	registance: {min_dist: 845, min_fee: 350, per_meter: 200, per_fee: 50}
}.freeze

# 経路から都市を判定する
# == Args
# _route_: 経路または地点、文字列
# == Returns
# 覚醒派か開放派か
def get_city(route)
	if CITY[:enlightened].any?{ |e| e == route.to_sym || e == route.reverse.to_sym }
		:enlightened
	elsif CITY[:registance].any?{ |e| e == route.to_sym || e == route.reverse.to_sym }
		:registance
	else
		raise "未定義の経路指定"
	end
end

# とある世界のタクシー料金を求める
# == Args
# _route_: 料金を計算したいルート
# == Returns
# 料金
def get_taxi_fee(route)
	n = 2
	start_point = route[0]

	# 各地点の累積距離を求める
	route = route.scan(/\w/).each_cons(n).map(&:join) # => ["AB", "BC", "CD"]
	each_dist = route.map{ |e| DISTANTCE[e.to_sym] || DISTANTCE[e.reverse.to_sym] } # => [100, 200, 300]
	integrated_distantce =  each_dist.collect.with_index{ |e, i| each_dist[0..i].reduce(:+) } # => [100, 300, 600]

	# 初期設定(初乗り運賃、初乗り料金)を求める
	running_disatnce, total_fee = [
		TAXI_SETTING[get_city(start_point)][:min_dist],
		TAXI_SETTING[get_city(start_point)][:min_fee]
	]

	# タクシー⊂（＾ω＾）⊃ﾌﾞｰﾝ
	total_distance = integrated_distantce[-1]
	while running_disatnce < total_distance
		# 到着したら終了
		if running_disatnce > total_distance
			break
		end

		# 現在位置からどちらの都市かを判定し運賃を加算
		current_pos = route[integrated_distantce.find_index{ |e| e > running_disatnce }]
		total_fee += TAXI_SETTING[get_city(current_pos)][:per_fee]

		# 200m すすむ
		running_disatnce += 200
	end

	total_fee
end

# てすと
def test
	DATA.read.split("\n").map{ |e|
		e.gsub(/\s+/, ",").split(",")
	}.each do |item|
		i, route, expect = item
		actual = get_taxi_fee(route)
		ok = actual == expect.to_i ? "ok" : "ng"
		puts "%2d: %-10s => res: %5s == %-5s : exp => %s" % [i.to_i, route, actual, expect, ok]
	end
end

if $0 == __FILE__
	test
end

__END__
0 	ADFC 	510
1 	CFDA 	500
2 	AB 	460
3 	BA 	460
4 	CD 	400
5 	DC 	350
6 	BG 	520
7 	GB 	530
8 	FDA 	450
9 	ADF 	450
10 	FDACB 	750
11 	BCADF 	710
12 	EDACB 	800
13 	BCADE 	810
14 	EGFCADE 	920
15 	EDACFGE 	910
16 	ABCDA 	960
17 	ADCBA 	1000
18 	BADCFGB 	1180
19 	BGFCDAB 	1180
20 	CDFC 	460
21 	CFDC 	450
22 	ABGEDA 	1420
23 	ADEGBA 	1470
24 	CFGB 	640
25 	BGFC 	630
26 	ABGEDFC 	1480
27 	CFDEGBA 	1520
28 	CDFGEDABG 	1770
29 	GBADEGFDC 	1680

その他

都市名がingress由来ということにさっぱり気づきませんでした。書いてる途中にふと気になって調べてみたらingressのページがでてきたため、いそいでコードに修正を加えました。