📜  数组中总和大于 0 的对数

📅  最后修改于: 2021-10-26 06:43:32             🧑  作者: Mango

给定一个大小为N的数组arr[] ,任务是找到数组中总和 > 0的不同对的数量。

例子:

朴素的方法:这个问题的朴素的方法是考虑数组中所有唯一的元素对。对于每一对,检查总和是否为正。
时间复杂度: O(N 2 )

有效的方法:

  • 这个想法是使用排序的概念和两个指针技术。
  • 对于这个问题,使用排序是因为对于总和arr[i] + arr[j] > 0其中 i, j 是数组中的一些随机索引, arr[i] > 0arr[j] > 0或两者兼而有之arr[i] 和 arr[j] > 0
  • 因此,一旦数组被排序,因为我们需要找到唯一的对。对于每个满足arr[i] > 0 的‘i’,我们需要找到满足arr[j] + arr[j] > 0的 j 的数量。
  • 在这里,由于数组已排序,因此使用双指针技术很容易找到对的计数。我们只需要找到条件成立的 ‘j’ 最左边的位置。这是使用-arr[i] + 1lower_bound找到的。
  • 例如,让数组 arr[] = {-4, 4, -5, 5, 3, -2, -3, -1, 2, 1}。此数组已排序。因此,数组变为,{-5, -4, -3, -2, -1, 1, 2, 3, 4, 5}。对于一些随机 i,假设 arr[i] = 4。因此,在数组中找到 -3 的索引,即 2。现在,我们可以确定对于索引 2 和 8 之间的所有值,该值arr[i] + arr[j] > 0。

下面是上述方法的实现:

CPP
// C++ program to find the
// number of pairs in the
// array with the sum > 0
 
#include 
using namespace std;
 
// Function to find the number
// of pairs in the array with
// sum > 0
int findNumOfPair(int* a, int n)
{
 
    // Sorting the given array
    sort(a, a + n);
 
    // Variable to store the count of pairs
    int ans = 0;
 
    // Loop to iterate through the array
    for (int i = 0; i < n; ++i) {
 
        // Ignore if the value is negative
        if (a[i] <= 0)
            continue;
 
        // Finding the index using lower_bound
        int j = lower_bound(a, a + n, -a[i] + 1) - a;
 
        // Finding the number of pairs between
        // two indices i and j
        ans += i - j;
    }
    return ans;
}
 
// Driver code
int main()
{
    int a[] = { 3, -2, 1 };
    int n = sizeof(a) / sizeof(a[0]);
 
    int ans = findNumOfPair(a, n);
    cout << ans << endl;
 
    return 0;
}


Java
// Java program to find the
// number of pairs in the
// array with the sum > 0
import java.util.*;
 
class GFG {
 
    // Function to find the number
    // of pairs in the array with
    // sum > 0
    static int findNumOfPair(int arr[], int n)
    {
 
        // Sorting the given array
        Arrays.sort(arr);
 
        // Variable to store the count of pairs
        int ans = 0;
 
        // Loop to iterate through the array
        for (int i = 0; i < n; ++i) {
 
            // Ignore if the value is negative
            if (arr[i] <= 0)
                continue;
           
            /*
            minReqVal val is the min value ,which will
            give >=1 after adding with the arr[i]
            */
            int minReqVal = -arr[i] + 1;
            int j = lower_bound(arr, minReqVal);
 
            if (j >= 0)
                ans += i - j;
        }
        return ans;
    }
 
    /*
         it return the index of a minimum Number in the
         array which is just >= val
      */
    static int lower_bound(int arr[], int val)
    {
        int start = 0, end = arr.length;
 
        /*
          using the Binary search technique , since our
          array is sorted
        */
        while (start < end) {
            int mid = (start + end) >> 1;
 
            if (val > arr[mid])
                start = mid + 1;
            else
                end = mid;
        }
 
        // when we dont find the answer return -1
        if (start == arr.length)
            return -1;
 
        return start;
    }
 
    // Driver code
    public static void main(String[] args)
    {
        int a[] = {-2,-1,-1,-1,-1,0 ,1,2,3};
        int n = a.length;
 
        int ans = findNumOfPair(a, n);
        System.out.println(ans);
    }
}
 
// This code is contributed by Pradeep Mondal P


Python3
# Python3 program to find the
# number of pairs in the
# array with the sum > 0
from bisect import bisect_left as lower_bound
 
# Function to find the number
# of pairs in the array with
# sum > 0
 
 
def findNumOfPair(a, n):
 
    # Sorting the given array
    a = sorted(a)
 
    # Variable to store the count of pairs
    ans = 0
 
    # Loop to iterate through the array
    for i in range(n):
 
        # Ignore if the value is negative
        if (a[i] <= 0):
            continue
 
        # Finding the index using lower_bound
        j = lower_bound(a, -a[i] + 1)
 
        # Finding the number of pairs between
        # two indices i and j
        ans += i - j
    return ans
 
 
# Driver code
if __name__ == '__main__':
    a = [3, -2, 1]
    n = len(a)
 
    ans = findNumOfPair(a, n)
    print(ans)
 
# This code is contributed by mohit kumar 29


C#
// C# program to find the
// number of pairs in the
// array with the sum > 0
using System;
 
class GFG {
 
    // Function to find the number
    // of pairs in the array with
    // sum > 0
    static int findNumOfPair(int[] arr, int n)
    {
 
        // Sorting the given array
        Array.Sort(arr);
 
        // Variable to store the count of pairs
        int ans = 0;
 
        // Loop to iterate through the array
        for (int i = 0; i < n; ++i) {
 
            // Ignore if the value is negative
            if (arr[i] <= 0)
                continue;
 
            /*
            minReqVal val is the min value ,which will
            give >=1 after adding with the arr[i]
            */
            int minReqVal = -arr[i] + 1;
            int j = lower_bound(arr, minReqVal);
 
            if (j >= 0)
                ans += i - j;
        }
        return ans;
    }
 
    /*
           it return the index of a minimum Number in the
           array which is just >= val
        */
    static int lower_bound(int[] arr, int val)
    {
        int start = 0, end = arr.Length;
 
        /*
          using the Binary search technique , since our
          array is sorted
        */
        while (start < end) {
            int mid = (start + end) >> 1;
 
            if (val > arr[mid])
                start = mid + 1;
            else
                end = mid;
        }
 
        // when we dont find the answer return -1
        if (start == arr.Length)
            return -1;
 
        return start;
    }
 
    // Driver code
    public static void Main()
    {
        int[] a = { -2, 1, 3 };
        int n = a.Length;
 
        int ans = findNumOfPair(a, n);
        Console.Write(ans);
    }
}
 
// This code is contributed by Pradeep Mondal P


Javascript


输出
2

时间复杂度: O(N * log(N))

如果您希望与专家一起参加现场课程,请参阅DSA 现场工作专业课程学生竞争性编程现场课程