AtCoder Beginner Contest 139 D. ModSum【Python】

https://atcoder.jp/contests/abc139/tasks/abc139_d
AtCoder ProblemsのRecommendationで Difficulty: 357、Solve Probability: 49%でした。

N = int(input())
ans = (1 + N-1) * (N-1) // 2
print(ans)

WAとなったコード

N = int(input())
ans = int((1 + N-1) * (N-1) / 2)
print(ans)