mirror of
https://github.com/autistic-symposium/master-algorithms-py.git
synced 2025-04-30 12:46:11 -04:00
22 lines
417 B
Python
22 lines
417 B
Python
#!/usr/bin/python3
|
|
# mari von steinkirch @2013
|
|
# steinkirch at gmail
|
|
|
|
|
|
def sum_square_diff(n):
|
|
sq_sum, sum_sq = 0, 0
|
|
for i in range(1, n+1):
|
|
sum_sq += i**2
|
|
sq_sum += i
|
|
sq_sum = sq_sum **2
|
|
return sq_sum - sum_sq
|
|
|
|
def main():
|
|
assert(sum_square_diff(10) == 2640)
|
|
print(sum_square_diff(100))
|
|
print('Tests Passed!')
|
|
|
|
if __name__ == '__main__':
|
|
main()
|
|
|