CountGoodTriplets_1534.cpp

/*
~ Author    : https://leetcode.com/tridib_2003/
~ Problem   : 1534. Count Good Triplets
~ Link      : https://leetcode.com/problems/count-good-triplets/
*/

class Solution {
public:
    int countGoodTriplets(vector<int>& arr, int a, int b, int c) {

        int n = arr.size(), tripletCount = 0;

        for (int i = 0; i < n - 2; ++i) {

            for (int j = i + 1; j < n - 1; ++j) {

                if (abs(arr[i] - arr[j]) <= a) {

                    for (int k = j + 1; k < n; ++k) {

                        if (abs(arr[j] - arr[k]) <= b && abs(arr[i] - arr[k]) <= c)
                            ++tripletCount;
                    }
                }
            }
        }

        return tripletCount;
    }
};