aboutsummaryrefslogtreecommitdiff
path: root/count_num_fair_pairs.py
diff options
context:
space:
mode:
Diffstat (limited to 'count_num_fair_pairs.py')
-rw-r--r--count_num_fair_pairs.py5
1 files changed, 2 insertions, 3 deletions
diff --git a/count_num_fair_pairs.py b/count_num_fair_pairs.py
index 5933dbd..8d02562 100644
--- a/count_num_fair_pairs.py
+++ b/count_num_fair_pairs.py
@@ -19,9 +19,8 @@ class Solution(object):
nums.sort()
ans = 0
for i in range(len(nums) - 1, -1, -1):
- v = nums[i]
- a = bisect.bisect_left(nums, lower - v, lo=0, hi=i)
- b = bisect.bisect_right(nums, upper - v, lo=0, hi=i)
+ a = bisect.bisect_left(nums, lower - nums[i], i + 1)
+ b = bisect.bisect_right(nums, upper - nums[i], i + 1)
ans += b - a
return ans