OpenSoure_Repos/TheAlgorithms-Python
1
0
Fork 0
mirror of https://hub.njuu.cf/TheAlgorithms/Python.git synced 2023-10-11 13:06:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

updated type hints

Browse Source
This commit is contained in:
siddwarr 2023-10-03 20:55:50 +05:30
parent a3ee9ebf0b
commit f6f790fdbd
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
data_structures/arrays/pairs_with_given_sum.py
View File

@ -16,9 +16,9 @@ to avoid the same pair getting counted twice
"""
def pairs_with_sum(arr, k) -> int:
def pairs_with_sum(arr: list, sum: int) -> int:
"""
Return the no. of pairs with sum k
Return the no. of pairs with sum "sum"
>>> pairs_with_sum([1,5,7,1],6)
2
@ -36,8 +36,8 @@ def pairs_with_sum(arr, k) -> int:
ans = 0
for i in arr:
d[i] -= 1
if k - i in d and d[k - i] != 0:
ans += d[k - i]
if sum - i in d and d[sum - i] != 0:
ans += d[sum - i]
d[i] += 1
return ans // 2