競技プログラミング コード終了 python3

Question

前提・実現したいこと

競技プログラミングで、for文内のifが一度真になったらコード終了したいのです。
しかし、exit()では、コードを終了できませんでした。
sys.exit()では標準出力にはなりませんでした。
なにか良い方法はありませんか？

該当のソースコード

10000円札i枚と5000円札j枚と1000円札(num-i-j)枚の合計額がtotalを探す。

num,total = map(int,input().split(" "))
for i in range(num+1):
    for j in range(num+1):
        if i*10000 + j*5000 + (num-i-j)*1000 == total:
            print(i,j,num-i-j)
            exit()
print("not found")


または  +++import sys

       ---exit()
       +++sys.exit()

Comments

[LouiS0616]

『標準出力にはならない』という意味がいまいち分かりません。もうちょっと詳しく、期待する挙動と実際の挙動を書いてください。

[tachikoma]

exitで終了させたいのは、終了ステータスをいじりたいとかですか？

Answer 1

for文内のifが一度真になったらコード終了したいのです。

ループを抜ければよいです。以下のようにelseを利用すればできます。
参考：Pythonで多重ループ（ネストしたforループ）からbreak

for i in range(2):
    for j in range(3):
        for k in range(4):
            print(i,j,k)
            if i == 0 and j == 1 and k == 2:
                break
        else:
            continue
        break
    else:
        continue
    break

Answer 2

処理部分を関数にして、exitをreturnにすればいいのでは。

def myfunc(num, total):
    for i in range(num+1):
        for j in range(num+1):
            if i*10000 + j*5000 + (num-i-j)*1000 == total:
                print(i,j,num-i-j)
                return True
    return False

num, total = map(int,input().split(" "))
found = myfunc(num, total)

if not found:
    print("not found")

Comments

[jacknash]

ありがとうございました。やっぱ関数のほうが扱いやすいですね