Project Euler problems 06, 20

project_euler/problem_06/sol4.py

@@ -0,0 +1,40 @@
+# -*- coding: utf-8 -*-
+"""
+Problem:
+
+The sum of the squares of the first ten natural numbers is,
+            1^2 + 2^2 + ... + 10^2 = 385
+
+The square of the sum of the first ten natural numbers is,
+            (1 + 2 + ... + 10)^2 = 552 = 3025
+
+Hence the difference between the sum of the squares of the first ten natural
+numbers and the square of the sum is 3025 − 385 = 2640.
+
+Find the difference between the sum of the squares of the first N natural
+numbers and the square of the sum.
+"""
+
+
+def solution(n):
+    """Returns the difference between the sum of the squares of the first n
+    natural numbers and the square of the sum.
+
+    >>> solution(10)
+    2640
+    >>> solution(15)
+    13160
+    >>> solution(20)
+    41230
+    >>> solution(50)
+    1582700
+    >>> solution(100)
+    25164150
+    """
+    sum_of_squares = n * (n + 1) * (2 * n + 1) / 6
+    square_of_sum = (n * (n + 1) / 2) ** 2
+    return int(square_of_sum - sum_of_squares)
+
+
+if __name__ == "__main__":
+    print(solution(int(input("Enter a number: ").strip())))

project_euler/problem_20/sol3.py

@@ -0,0 +1,39 @@
+"""
+n! means n × (n − 1) × ... × 3 × 2 × 1
+
+For example, 10! = 10 × 9 × ... × 3 × 2 × 1 = 3628800,
+and the sum of the digits in the number 10! is 3 + 6 + 2 + 8 + 8 + 0 + 0 = 27.
+
+Find the sum of the digits in the number 100!
+"""
+from math import factorial
+
+
+def solution(n):
+    """Returns the sum of the digits in the number 100!
+    >>> solution(1000)
+    10539
+    >>> solution(200)
+    1404
+    >>> solution(100)
+    648
+    >>> solution(50)
+    216
+    >>> solution(10)
+    27
+    >>> solution(5)
+    3
+    >>> solution(3)
+    6
+    >>> solution(2)
+    2
+    >>> solution(1)
+    1
+    >>> solution(0)
+    1
+    """
+    return sum(map(int, str(factorial(n))))
+
+
+if __name__ == "__main__":
+    print(solution(int(input("Enter the Number: ").strip())))