個人的には、辞書を使わない方が書きやすい気がします。
解法の方針
0. 3個ずつ組にしたタプルのリストを作る
0. 最も合計値が小さいタプルのインデックスを調べる
Python
1n, span = map(int, input().split())
2
3nums = [
4 int(input().split()[-1]) for _ in range(n)
5]
6
7groups = zip(*[nums[s:] for s in range(span)])
8indice = [
9 index
10 for index, group in sorted(
11 enumerate(groups, start=1),
12 key=lambda ig: sum(ig[-1]) # (index, num_group)
13 )
14]
15
16print(indice[0])
Wandbox
hayataka2049さんのご指摘を受けて修正したバージョン。
Python
1n, span = map(int, input().split())
2
3nums = [
4 int(input().split()[-1]) for _ in range(n)
5]
6
7groups = zip(*[nums[s:] for s in range(span)])
8index, _ = min(
9 enumerate(groups, start=1),
10 key=lambda ig: sum(ig[-1]) # (index, num_group)
11)
12
13print(index)
Wandbox
せっかくなので
辞書を使った方法も書いてみた。
Python
1n, span = map(int, input().split())
2
3nums = [
4 int(input().split()[-1]) for _ in range(n)
5]
6
7sum_dict = {
8 i: 0 for i in range(n-span+1)
9}
10for key in sum_dict:
11 for i in range(span):
12 sum_dict[key] += nums[key+i]
13
14index = sorted(sum_dict.items(), key=lambda item: item[-1])[0][0]
15print(index+1)
Wandbox
コメントを受けて
enumerateなんですが、for文の際に使うもので、リストのインデックスをとるときに使用するものだと認識しておりますが、for文がありません。
ここでのenumerateは何をしているのでしょうか。。？
また、groupsは上記でタプルにされているデータをさしているのはわかるのですが、stard=1は何なのでしょうか？
for文で良く用いられるのは確かですが、その実態はジェネレータを返しているだけなのです。
また、キーワード引数startは、どの数値から順に付与するかを示します。
Python
1>>> abc = list('abcdefg')
2>>> abc
3['a', 'b', 'c', 'd', 'e', 'f', 'g']
4>>>
5>>> enumerate(abc)
6<enumerate object at 0x00000270D21ED438>
7>>>
8>>> list(enumerate(abc))
9[(0, 'a'), (1, 'b'), (2, 'c'), (3, 'd'), (4, 'e'), (5, 'f'), (6, 'g')]
10>>>
11>>> list(enumerate(abc, start=5))
12[(5, 'a'), (6, 'b'), (7, 'c'), (8, 'd'), (9, 'e'), (10, 'f'), (11, 'g')]
key=lambda ig: sum(ig[-1])
lambdaは、引数、:の後に処理を記述すると認識しております。
min(enumerate(groups, start=1)この処理結果がigにはいって、sumして、keyに入って、それをindexと_に戻り値として渡される？という認識でよろしいでしょうか？
enumerate(groups, start=1)が返した各要素がigに入ります。
例えば一回目は(1, (0, 20, 10))が、二回目は(2, (20, 10, 80))がigに与えられるわけです。
keyは関数が『何をもって要素を比べるか』指定するために使います。
今回はsum(ig[-1])、つまり件の三個組の要素の合計を指標にしています。
minした結果をラムダで処理しているのではなく、minを求めるのにラムダを用いています。
minがどんな処理をしているか覗いてみたら、少しイメージが湧くかもしれません。
Python
1def my_min(it, key=None):
2 print('-' * 42)
3 if key is None:
4 key = lambda x: x # 恒等関数
5
6 ret = next(it)
7 ret_score = key(ret)
8
9 for e in it:
10 e_score = key(e)
11 print(f'{ret}[score={ret_score}] vs {e}[score={e_score}]', end='\t')
12
13 if ret_score > e_score:
14 print('latter wins')
15 ret, ret_score = e, e_score
16 else:
17 print('former wins')
18
19 print(f'smallest element is {ret}')
20 print('-' * 42)
21 return ret
実行結果 (minの代わりにmy_minを使用) Wandbox
plain
1------------------------------------------
2(1, (0, 20, 10))[score=30] vs (2, (20, 10, 80))[score=110] former wins
3(1, (0, 20, 10))[score=30] vs (3, (10, 80, 10))[score=100] former wins
4(1, (0, 20, 10))[score=30] vs (4, (80, 10, 0))[score=90] former wins
5(1, (0, 20, 10))[score=30] vs (5, (10, 0, 10))[score=20] latter wins
6smallest element is (5, (10, 0, 10))
7------------------------------------------
85
註： 説明のために書いたコードであり、実際のminの実装を踏襲したものではありません。
前半部分「sorted(sum_dict.items()」は、辞書をソートされている。ここまではわかるのですが、
やはり、lambdaの処理が完璧に理解できません。（ソートした辞書をitemに渡す？）
辞書の各要素をタプルにしたシーケンスを作り、そいつをソートしています。
ソートした結果をラムダで処理しているのではなく、ソートするのにラムダを用いています。
Python
1>>> tmp_dict = {i: i**2 for i in range(10)}
2>>> tmp_dict
3{0: 0, 1: 1, 2: 4, 3: 9, 4: 16, 5: 25, 6: 36, 7: 49, 8: 64, 9: 81}
4>>>
5>>> tmp_dict.items()
6dict_items([(0, 0), (1, 1), (2, 4), (3, 9), (4, 16), (5, 25), (6, 36), (7, 49), (8, 64), (9, 81)])
回答いただいたのに、それに対しての質問ばかりでごめんなさい。
いえいえ、お気遣いなく。
Q&A
ベストアンサー
2018/06/29 16:31