-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathpairSumInAnArray.java
93 lines (77 loc) · 1.93 KB
/
pairSumInAnArray.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
/*
You have been given an integer array/list(ARR) and a number 'num'. Find and return the total number of pairs in the array/list which sum to 'num'.
Note:
Given array/list can contain duplicate elements.
Input format :
The first line contains an Integer 't' which denotes the number of test cases or queries to be run. Then the test cases follow.
First line of each test case or query contains an integer 'N' representing the size of the first array/list.
Second line contains 'N' single space separated integers representing the elements in the array/list.
Third line contains an integer 'num'.
Output format :
For each test case, print the total number of pairs present in the array/list.
Output for every test case will be printed in a separate line.
Constraints :
1 <= t <= 10^2
0 <= N <= 10^4
0 <= num <= 10^9
Time Limit: 1 sec
Sample Input 1:
1
9
1 3 6 2 5 4 3 2 4
7
Sample Output 1:
7
Sample Input 2:
2
9
1 3 6 2 5 4 3 2 4
12
6
2 8 10 5 -2 5
10
Sample Output 2:
0
2
Explanation for Input 2:
Since there doesn't exist any pair with sum equal to 12 for the first query, we print 0.
For the second query, we have 2 pairs in total that sum up to 10. They are, (2, 8) and (5, 5).*/
import java.util.Arrays;
public class Solution {
public static int pairSum(int[] arr, int num) {
//Your code goes here
Arrays.sort(arr);
int n = arr.length;
int count = 0;
int low = 0;
int high = arr.length - 1;
while (low < high) {
if (arr[low] + arr[high] == num) {
int a = arr[low];
int b = arr[high];
if (a == b) {
int temp = high - low + 1;
count = count + (temp * (temp - 1)) / 2;
break;
} else {
int temp_a = 0, temp_b = 0;
while (arr[low] == a) {
temp_a++;
low++;
}
while (arr[high] == b) {
temp_b++;
high--;
}
count += (temp_a * temp_b);
}
}
else if (arr[low] + arr[high] > num) {
high--;
} else {
low++;
}
}
return count;
}
}