diff options
Diffstat (limited to 'div_non_div_sum_diff.py')
-rw-r--r-- | div_non_div_sum_diff.py | 25 |
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)) |