AtCoder Beginner Contest 044AをPythonで解く

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

問題

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

こう考えた

  • N<=Kif分岐させて出力を分ける。
  • 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]))

🐍リスト内包表記便利や

Last Updated: 6/13/2019, 10:52:02 PM