回答編集履歴

3 追記

LouiS0616

LouiS0616 score 25007

2018/06/30 13:37  投稿

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

LouiS0616

LouiS0616 score 25007

2018/06/29 21:00  投稿

個人的には、辞書を使わない方が書きやすい気がします。
**解法の方針**
0. 3個ずつ組にしたタプルのリストを作る
0. 最も合計値が小さいタプルのインデックスを調べる
```Python
n, span = map(int, input().split())
nums = [
   int(input().split()[-1]) for _ in range(n)
]
groups = zip(*[nums[s:] for s in range(span)])
indice = [
   index
   for index, group in sorted(
       enumerate(groups, start=1),
       key=lambda ig: sum(ig[-1])  # (index, num_group)
   )
]
print(indice[0])
```
[Wandbox](https://wandbox.org/permlink/FtTATIMNeMWGuNMJ)
---  
hayataka2049さんのご指摘を受けて修正したバージョン。  
```Python  
n, span = map(int, input().split())  
 
nums = [  
   int(input().split()[-1]) for _ in range(n)  
]  
 
groups = zip(*[nums[s:] for s in range(span)])  
index, _ = min(  
   enumerate(groups, start=1),  
   key=lambda ig: sum(ig[-1])  # (index, num_group)  
)  
 
print(index)  
```  
 
[Wandbox](https://wandbox.org/permlink/1nzQntpOahKrbzns)  
 
せっかくなので
---
辞書を使った方法も書いてみた。
```Python
n, span = map(int, input().split())
nums = [
   int(input().split()[-1]) for _ in range(n)
]
sum_dict = {
   i: 0 for i in range(n-span+1)
}
for key in sum_dict:
   for i in range(span):
       sum_dict[key] += nums[key+i]
       
index = sorted(sum_dict.items(), key=lambda item: item[-1])[0][0]
print(index+1)
```
[Wandbox](https://wandbox.org/permlink/TxopW8SeqDztgdHd)
1 追記

LouiS0616

LouiS0616 score 25007

2018/06/29 20:30  投稿

個人的には、辞書を使わない方が書きやすい気がします。
**解法の方針**
0. 3個ずつ組にしたタプルのリストを作る
0. 最も合計値が小さいタプルのインデックスを調べる
```Python
n, span = map(int, input().split())
nums = [
   int(input().split()[-1]) for _ in range(n)
]
groups = zip(*[nums[s:] for s in range(span)])
indice = [
   index
   for index, group in sorted(
       enumerate(groups, start=1),
       key=lambda ig: sum(ig[-1])  # (index, num_group)
   )
]
print(indice[0])
```
[Wandbox](https://wandbox.org/permlink/FtTATIMNeMWGuNMJ)
[Wandbox](https://wandbox.org/permlink/FtTATIMNeMWGuNMJ)
せっかくなので
---
辞書を使った方法も書いてみた。
```Python
n, span = map(int, input().split())
nums = [
   int(input().split()[-1]) for _ in range(n)
]
sum_dict = {
   i: 0 for i in range(n-span+1)
}
for key in sum_dict:
   for i in range(span):
       sum_dict[key] += nums[key+i]
       
index = sorted(sum_dict.items(), key=lambda item: item[-1])[0][0]
print(index+1)
```
[Wandbox](https://wandbox.org/permlink/TxopW8SeqDztgdHd)

思考するエンジニアのためのQ&Aサイト「teratail」について詳しく知る