mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-11-27 15:01:08 +00:00
00fc53de97
* Create sumset.py * updating DIRECTORY.md * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Add descriptive var names * Update maths/sumset.py Co-authored-by: Caeden <caedenperelliharris@gmail.com> * Update sumset.py * updating DIRECTORY.md Co-authored-by: github-actions <${GITHUB_ACTOR}@users.noreply.github.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Caeden <caedenperelliharris@gmail.com> Co-authored-by: Christian Clauss <cclauss@me.com>
38 lines
870 B
Python
38 lines
870 B
Python
"""
|
|
|
|
Calculates the SumSet of two sets of numbers (A and B)
|
|
|
|
Source:
|
|
https://en.wikipedia.org/wiki/Sumset
|
|
|
|
"""
|
|
|
|
|
|
def sumset(set_a: set, set_b: set) -> set:
|
|
"""
|
|
:param first set: a set of numbers
|
|
:param second set: a set of numbers
|
|
:return: the nth number in Sylvester's sequence
|
|
|
|
>>> sumset({1, 2, 3}, {4, 5, 6})
|
|
{5, 6, 7, 8, 9}
|
|
|
|
>>> sumset({1, 2, 3}, {4, 5, 6, 7})
|
|
{5, 6, 7, 8, 9, 10}
|
|
|
|
>>> sumset({1, 2, 3, 4}, 3)
|
|
Traceback (most recent call last):
|
|
...
|
|
AssertionError: The input value of [set_b=3] is not a set
|
|
"""
|
|
assert isinstance(set_a, set), f"The input value of [set_a={set_a}] is not a set"
|
|
assert isinstance(set_b, set), f"The input value of [set_b={set_b}] is not a set"
|
|
|
|
return {a + b for a in set_a for b in set_b}
|
|
|
|
|
|
if __name__ == "__main__":
|
|
from doctest import testmod
|
|
|
|
testmod()
|