mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-11-23 21:11:08 +00:00
Count pairs with given sum (#10282)
* added power_of_4 * deleted power_of_4 * added pairs_with_given_sum * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * updated the comment * updated return hint * updated type hints * updated the variable * updated annotation * updated code * updated code * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * added the problem link and used defaultdict * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * corrected import formatting * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update pairs_with_given_sum.py * Update data_structures/arrays/pairs_with_given_sum.py --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Christian Clauss <cclauss@me.com>
This commit is contained in:
parent
92fbe60082
commit
09ce6b23d7
28
data_structures/arrays/pairs_with_given_sum.py
Normal file
28
data_structures/arrays/pairs_with_given_sum.py
Normal file
|
@ -0,0 +1,28 @@
|
|||
#!/usr/bin/env python3
|
||||
|
||||
"""
|
||||
Given an array of integers and an integer req_sum, find the number of pairs of array
|
||||
elements whose sum is equal to req_sum.
|
||||
|
||||
https://practice.geeksforgeeks.org/problems/count-pairs-with-given-sum5022/0
|
||||
"""
|
||||
from itertools import combinations
|
||||
|
||||
|
||||
def pairs_with_sum(arr: list, req_sum: int) -> int:
|
||||
"""
|
||||
Return the no. of pairs with sum "sum"
|
||||
>>> pairs_with_sum([1, 5, 7, 1], 6)
|
||||
2
|
||||
>>> pairs_with_sum([1, 1, 1, 1, 1, 1, 1, 1], 2)
|
||||
28
|
||||
>>> pairs_with_sum([1, 7, 6, 2, 5, 4, 3, 1, 9, 8], 7)
|
||||
4
|
||||
"""
|
||||
return len([1 for a, b in combinations(arr, 2) if a + b == req_sum])
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
from doctest import testmod
|
||||
|
||||
testmod()
|
Loading…
Reference in New Issue
Block a user