-
Notifications
You must be signed in to change notification settings - Fork 0
/
CountTriplets.cpp
60 lines (52 loc) · 1.1 KB
/
CountTriplets.cpp
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
//https://practice.geeksforgeeks.org/problems/count-the-triplets/0
#include<bits/stdc++.h>
using namespace std;
#define ll long long int
int main()
{
//code
ll test,size,*arr,count,val;
cin>>test;
while(test--)
{
cin>>size;
count=0;
arr =new ll[size];
ll hsh[100007] = {0};
for(ll i=0;i<size;i++)
{
cin>>arr[i];
hsh[arr[i]] = 1;
}
if(size<=2)
{
cout<<-1<<endl;
}
else
{
sort(arr,arr+size);
for(ll j=size-1;j>=2;j--)
{
for(ll k=j-1,l=0;k>l;)
{
if(arr[k]+arr[l]==arr[j])
{
k--;l++;
count++;
}
else if(arr[k]+arr[l]>arr[j]){
k--;
}
else{
l++;
}
}
}
if(count>0)
cout<<count<<endl;
else
cout<<-1<<endl;
}
}
return 0;
}