aboutsummaryrefslogtreecommitdiff
path: root/div_non_div_sum_diff.py
diff options
context:
space:
mode:
Diffstat (limited to 'div_non_div_sum_diff.py')
-rw-r--r--div_non_div_sum_diff.py25
1 files changed, 25 insertions, 0 deletions
diff --git a/div_non_div_sum_diff.py b/div_non_div_sum_diff.py
new file mode 100644
index 0000000..51d7cac
--- /dev/null
+++ b/div_non_div_sum_diff.py
@@ -0,0 +1,25 @@
+# 2894. Divisible and Non-divisible Sums Difference
+
+"""
+you are given a positive integers 'n' and 'm'. define two integers as
+follows: 'num1': the sum of all integers in the range '[1, n]' that are not
+divisible by 'm'. 'num2': the sum of all integers in the range '[1, n]' that
+are divisible by 'm'. return the integer 'num1 - num2'
+"""
+
+
+class Solution(object):
+ def differenceOfSums(self, n, m):
+ """
+ :type n: int
+ :type m: int
+ :rtype: int
+ """
+ return (1 + n) * n // 2 - (1 + n // m) * (n // m) * m
+
+
+if __name__ == "__main__":
+ obj = Solution()
+ print(obj.differenceOfSums(n=10, m=3))
+ print(obj.differenceOfSums(n=5, m=6))
+ print(obj.differenceOfSums(n=5, m=1))