My issue
Why this code is not working
My code
def maximum_iteration(maxT,maxN,sumN):
ans = 0
temp = sumN//maxN
if temp < maxT:
ans = temp * (maxN*maxN)
else:
ans = maxT * (maxN*maxN)
mod = sumN%maxN
ans += (mod*mod)
return ans
T = int(input().strip())
for i in range(T):
maxT , maxN , sumN = map(int,input().strip().split())
ans = maximum_iteration(maxT,maxN,sumN)
print(ans)
Problem Link: SLOWSOLN Problem - CodeChef