ある条件を除く合計数の求め方

Question

def sum13(nums):
    if not nums:
        return 0
    total = 0
    for i in range(len(nums)):
        if nums[i] != 13:
            if nums[i-1] != 13:
                total += nums[i]
        else:
            total += 0
            
    return total

パラメーターのnumsにいくつかの数字が入るときに合計を求めたいです。
その時、数字が１３だった場合は１３とその次の数は合計にカウントしないという条件があります。
私がこのプログラムを実行したら下のようなエラーが出ました。

File "C:\Users\Miyu Oyama\CS-1410\cs1410_week01_review\cs1410_week01_review_07\tests\test_problems.py", line 32, in test_04
    self.assertEqual(ans, correct_answer, "You returned {0}. You should have returned {1}. Check your logic and try again.".format(ans, correct_answer))
AssertionError: 3 != 4 : You returned 3. You should have returned 4. Check your logic and try again.

このメッセージによるともしnumsパラメーターに[1, 2, 13, 2, 1, 13]の順に数字が入っているときに私のプログラムだと合計が４になるべきなのに３になってしまうといわれました。
何度試しても同じようなエラーが出てしまうのですが、どうして１３の前の１がカウントされないというエラーが起こるのか教えてほしいです。
よろしくお願いします。

Answer 1

このメッセージによるともしnumsパラメーターに[1, 2, 13, 2, 1, 13]の順に数字が入っているときに私のプログラムだと合計が４になるべきなのに３になってしまうといわれました。

nums = [1, 2, 13, 2, 1, 13] のとき、len(nums) は 6。
range(len(nums) は range(6) だから、i は 0～5。

nums[0] は 1 で 13 ではなく、nums[-1] が 13 だから total は 0 のまま。
nums[1] は 2 で 13 ではなく、nums[0] が 13 ではないから total は 2。
nums[2] は 13 で else に行き、total は 2 のまま。
nums[3] は 2 で 13 ではなく、nums[2] が 13 だから total は 2 のまま。
nums[4] は 1 で 13 ではなく、nums[3] が 13 ではないから total は 3。
nums[5] は 13 で else に行き、total は 3 のまま。

追記
なぜ、3 となるかは説明しましたが、条件通りのコードとするには、

def sum13(nums):
	if not nums:
		return 0
	total = 0
	for i in range(len(nums)):
		if nums[i] != 13 and (i == 0 or nums[i-1] != 13):
			total += nums[i]
	return total

print(sum13([1, 2, 13, 2, 1, 13]))

条件は、「13ではない、かつ 13の次ではない」であり、
「13の次ではない」は「先頭であるか、または 1つ前が 13ではない」。

追記2
条件は、「13ではない、かつ 1つ前が 13ではない」ですから、
1つ前の値(previous value) を憶えておく変数を用意すると、

def sum13(nums):
	if not nums:
		return 0
	prev = total = 0
	for i in range(len(nums)):
		if nums[i] != 13 and prev != 13:
			total += nums[i]
		prev = nums[i]
	return total

print(sum13([1, 2, 13, 2, 1, 13]))

とにかく、書き方はいくつもあります。

追記3
range も len も [i] による要素アクセスも全部なくせます。

def sum13(nums):
	if not nums:
		return 0
	prev = total = 0
	for val in nums:
		if val != 13 and prev != 13:
			total += val
		prev = val
	return total

print(sum13([1, 2, 13, 2, 1, 13]))

Comments

[Miyu_Oyama]

返信ありがとうございます。
なるほど！！　
数え方が1からスタートと０からスタートで違うからエラーになるんですね。
長さを図るときは１からスタートだと覚えたいです！
一つずづどのように私のプログラムが作用するか教えてくださりありがとうございました(*^^*)

[Miyu_Oyama]

ありがとうございます！
そういった書き方も出来るんですね。
先頭であるかどうかという区別の仕方勉強になりました✿

[Miyu_Oyama]

すごいですね
[i]も省略できるんですね！
たくさん例を教えてくれてとってもありがたいです(*・v・*

Answer 2

def sum13(nums):
    if not nums:
        return 0
    total = 0
    for i in range(len(nums)):
        print('loop',i,'回目')
        if nums[i] != 13:
            if nums[i-1] != 13:
                print('足し算')
                total += nums[i]
        else:
            total += 0

    return total

sum13([1, 2, 13, 2, 1, 13])

"""
loop 0 回目
loop 1 回目
足し算
loop 2 回目
loop 3 回目
loop 4 回目
足し算
loop 5 回目
3
"""

途中でデバックコードprintを入れてみると動作がこのようになっています。
rangeが0からはじまっているので最初のi-1が判定されていないなど、
動作が規定通りではないですね。

自分だとflag=Falseを用意しておいて、Falseのとき足し算する。Trueになったらflagをfalseにするだけといったロジックにすると思います。

Comments

[kazuma-s]

> rangeが0からはじまっているので最初のi-1が判定されていないなど、
動作が規定通りではないですね。

Python では nums[-1] は最後の要素で 13 なので、判定されています。
動作はコードの通りです。問題の条件通りではないですが。

[Miyu_Oyama]

返信ありがとうございます。
私のコードのどこがダメだったのか試してくださったんですね。
参考になりました。
私の解決した方法は確かにコードが長いのでアドバイスいただいた方法も試してみます。
新しい知識もありがとうございました(^-^)

Answer 3

def sum13(nums):
    if not nums:
        return 0
    total = 0
    for i in range(len(nums)):
        if nums[i] != 13 and i == 0:
            total += nums[i]
        if nums[i] != 13 and i >= 1:
            if nums[i-1] != 13 and i >= 1:
                total += nums[i]
        else:
            total += 0
            
    return total

アドバイスをいただいたようにiが０の時もうまくカウントできるようにできました。
でもToshikawaさんの方法ではないので、その方法もできたら追加で上げたいです。
ありがとうございました!!