Pairs Having Similar Elements

Given an array, A, having N integers A1, A2,..., AN. Two elements of the array Ai and  Aj are called similar iff  Ai=Aj+1 or Aj=Ai+1. Also, the similarity follows transitivity. If 

$A_i$ and $A_j$ are similar and $A_j$ and $A_k$ are similar, then $A_i$ and $A_k$ are also similar. 
Note: $i$, $j$, and $k$ are all distinct.

You need to find the number of pairs of indices $(i,j)$ such that $i<j$ and $A_i$ is similar to $A_j$.

Example:

Input:  n = 8, a = [1,3,5,7,8,2,5,7]
Output: 6

Approach

C++

#include <bits/stdc++.h>
using namespace std;

long long pairsHavingSimiElements(long long n, long long a[])
{

    sort(a, a + n);
    long long cnt = 0, ans = 0;
    long long y = 0, i = 0;
    while (i < n)
    {
        if (i < n - 1 && a[i] + 1 == a[i + 1])
        {
            cnt++;
        }
        else if (i < n - 1 && a[i] == a[i + 1])
        {
            cnt++;
            y++;
        }
        else
        {
            if (y != cnt)
                ans += cnt * (cnt + 1) / 2;
            cnt = 0;
            y = 0;
        }
        i++;
    }
    return ans;
}
int main()
{

    long long n = 8;
    long long a[n] = {1, 3, 5, 7, 8, 2, 5, 7};

    cout << pairsHavingSimiElements(n, a) << "\n";

    return 0;
}