AtCoder Beginner Contest 044AをPythonで解く

今回は、AtCoder Beginner Contest 044Aを解いていきたいと思います。

問題

1軒のホテルがあります。 このホテルの宿泊費は、次のようになっています。
最初のK泊までは、1泊あたり X円K+1泊目以降は、1泊あたり Y円高橋君は、このホテルにN泊連続で宿泊することにしました。 高橋君の宿泊費は合計で何円になるか求めてください。

こう考えた

• N<=Kでif分岐させて出力を分ける。
• if N<=K:はprint(X*N)
• else:はprint(X*K+Y*(N-K))

書いたコード

N=int(input())
K=int(input())
X=int(input())
Y=int(input())

if N<=K:
  print(X*N)
else:
  print(X*K+Y*(N-K))

🐍これでいけんちゃうかな？入力のところは改善できるはず。

結果

AC！！

🐍おし！

入力を改善したコード

n,k,x,y=[int(input()) for i in [0]*4]
print(n*x-(x-y)*max([n-k,0]))

🐍リスト内包表記便利や