Count the Triplets

Task: Given an array of distinct elements, Find the number of triplets. Triplet means two elements sum is equals to third element.

Example:

Input:

1

4

1 5 3 2

Output:

2 // 1 + 2 = 3 and 2 + 3 = 5.

---

Here is the Code:

Program in C++:

1. #include<iostream>
2. #include<bits/stdc++.h>
3. int countTheTriplets(int *a,int n)
4. {
5.     sort(a,a+n);
6.     //1 2 3 5
7.     int j,k,c=0;
8.     for(int i=0;i<n-2;i++)
9.     {
10.         j=i+1;
11.         k=j+1;
12.         while(k<n)
13.         {
14.             if(a[i]+a[j]==a[k])
15.             {
16.                 c++;
17.                 j++;
18.                 //k--;
19.             }
20.             else if(a[i]+a[j]<a[k])
21.             {
22.                 j++;
23.                 k--;
24.             }
25.             k++; 
26.         }
27.     }
28.     if(c==0)
29.     {
30.         return -1;
31.     }
32.     else
33.     {
34.         return c;
35.     }
36. }
37. int main()
38.  {
39. int t;
40. cin>>t;
41. while(t--)
42. {
43.     int n;
44.     cin>>n;
45.     int a[n];
46.     for(int i=0;i<n;i++)
47.     {
48.         cin>>a[i];
49.     }
50.     cout<<countTheTriplets(a,n)<<endl;
51. }
52. return 0;
53. }
54. //Updated Version Code
55. class Solution{
56. public:
58. int countTriplet(int a[], int n)
59. {
60.     // Your code goes here
61.     sort(a,a+n);
62.     //1 2 3 5
63.     int j,k,c=0;
64.     for(int i=0;i<n-2;i++)
65.     {
66.         j=i+1;
67.         k=j+1;
68.         while(k<n)
69.         {
70.             if(a[i]+a[j]==a[k])
71.             {
72.                 c++;
73.                 j++;
74.                 //k--;
75.             }
76.             else if(a[i]+a[j]<a[k])
77.             {
78.                 j++;
79.                 k--;
80.             }
81.             k++; 
82.         }
83.     }
84.     return c;
85. }
86. };

---

Here is the Video:

---