📜  给定总和的 Count 对的Java程序

📅  最后修改于: 2022-05-13 01:55:25.592000             🧑  作者: Mango

给定总和的 Count 对的Java程序

给定一个整数数组和一个数字“sum”,找出数组中总和等于“sum”的整数对的数量。

例子:

Input  :  arr[] = {1, 5, 7, -1}, 
          sum = 6
Output :  2
Pairs with sum 6 are (1, 5) and (7, -1)

Input  :  arr[] = {1, 5, 7, -1, 5}, 
          sum = 6
Output :  3
Pairs with sum 6 are (1, 5), (7, -1) &
                     (1, 5)         

Input  :  arr[] = {1, 1, 1, 1}, 
          sum = 2
Output :  6
There are 3! pairs with sum 2.

Input  :  arr[] = {10, 12, 10, 15, -1, 7, 6, 
                   5, 4, 2, 1, 1, 1}, 
          sum = 11
Output :  9

预期时间复杂度 O(n)

朴素的解决方案——一个简单的解决方案是遍历每个元素并检查数组中是否有另一个数字可以添加到它以得出总和。

Java
// Java implementation of simple method to find count of
// pairs with given sum.
public class find {
    public static void main(String args[])
    {
        int[] arr = { 1, 5, 7, -1, 5 };
        int sum = 6;
        getPairsCount(arr, sum);
    }
  
    // Prints number of pairs in arr[0..n-1] with sum equal
    // to 'sum'
    public static void getPairsCount(int[] arr, int sum)
    {
  
        int count = 0; // Initialize result
  
        // Consider all possible pairs and check their sums
        for (int i = 0; i < arr.length; i++)
            for (int j = i + 1; j < arr.length; j++)
                if ((arr[i] + arr[j]) == sum)
                    count++;
  
        System.out.printf("Count of pairs is %d", count);
    }
}
// This program is contributed by Jyotsna


Java
/* Java implementation of simple method to find count of
pairs with given sum*/
  
import java.util.HashMap;
  
class Test {
    static int arr[] = new int[] { 1, 5, 7, -1, 5 };
  
    // Returns number of pairs in arr[0..n-1] with sum equal
    // to 'sum'
    static int getPairsCount(int n, int sum)
    {
        HashMap hm = new HashMap<>();
  
        // Store counts of all elements in map hm
        for (int i = 0; i < n; i++) {
  
            // initializing value to 0, if key not found
            if (!hm.containsKey(arr[i]))
                hm.put(arr[i], 0);
  
            hm.put(arr[i], hm.get(arr[i]) + 1);
        }
        int twice_count = 0;
  
        // iterate through each element and increment the
        // count (Notice that every pair is counted twice)
        for (int i = 0; i < n; i++) {
            if (hm.get(sum - arr[i]) != null)
                twice_count += hm.get(sum - arr[i]);
  
            // if (arr[i], arr[i]) pair satisfies the
            // condition, then we need to ensure that the
            // count is decreased by one such that the
            // (arr[i], arr[i]) pair is not considered
            if (sum - arr[i] == arr[i])
                twice_count--;
        }
  
        // return the half of twice_count
        return twice_count / 2;
    }
  
    // Driver method to test the above function
    public static void main(String[] args)
    {
  
        int sum = 6;
        System.out.println(
            "Count of pairs is "
            + getPairsCount(arr.length, sum));
    }
}
// This code is contributed by Gaurav Miglani


输出
Count of pairs is 3

时间复杂度: O(n 2 )
辅助空间: O(1)

高效的解决方案——
在 O(n) 时间内可能会有更好的解决方案。下面是算法——

  1. 创建一个映射来存储数组中每个数字的频率。 (需要单次遍历)
  2. 在下一次遍历中,对于每个元素,检查它是否可以与任何其他元素(除了它自己!)组合以给出所需的总和。相应地增加计数器。
  3. 在完成第二次遍历后,我们会在 counter 中存储两倍的所需值,因为每一对都被计算了两次。因此将计数除以 2 并返回。

以下是上述想法的实现:

Java

/* Java implementation of simple method to find count of
pairs with given sum*/
  
import java.util.HashMap;
  
class Test {
    static int arr[] = new int[] { 1, 5, 7, -1, 5 };
  
    // Returns number of pairs in arr[0..n-1] with sum equal
    // to 'sum'
    static int getPairsCount(int n, int sum)
    {
        HashMap hm = new HashMap<>();
  
        // Store counts of all elements in map hm
        for (int i = 0; i < n; i++) {
  
            // initializing value to 0, if key not found
            if (!hm.containsKey(arr[i]))
                hm.put(arr[i], 0);
  
            hm.put(arr[i], hm.get(arr[i]) + 1);
        }
        int twice_count = 0;
  
        // iterate through each element and increment the
        // count (Notice that every pair is counted twice)
        for (int i = 0; i < n; i++) {
            if (hm.get(sum - arr[i]) != null)
                twice_count += hm.get(sum - arr[i]);
  
            // if (arr[i], arr[i]) pair satisfies the
            // condition, then we need to ensure that the
            // count is decreased by one such that the
            // (arr[i], arr[i]) pair is not considered
            if (sum - arr[i] == arr[i])
                twice_count--;
        }
  
        // return the half of twice_count
        return twice_count / 2;
    }
  
    // Driver method to test the above function
    public static void main(String[] args)
    {
  
        int sum = 6;
        System.out.println(
            "Count of pairs is "
            + getPairsCount(arr.length, sum));
    }
}
// This code is contributed by Gaurav Miglani
输出
Count of pairs is 3

有关详细信息,请参阅有关给定总和的计数对的完整文章!