LC2367. 算术三元组的数目

题目链接：2367. 算术三元组的数目 - 力扣（LeetCode）

题解

• 三层 for 循环暴力，也可以优化，枚举 i 和 j，然后查找 a[k] 是否存在。

参考代码

class Solution {
public:
    int arithmeticTriplets(vector<int>& a, int diff) {
        int ans = 0;
        for(int i=0; i<a.size(); ++i) {
            for(int j=i+1; j<a.size(); j++) {
                if(a[j]-a[i]>diff) break;
                for(int k=j+1;k<a.size(); ++k) {
                    if(a[j]-a[i]==a[k]-a[j] && a[j]-a[i]==diff) {
                        ans += 1;
                    }
                    if (a[k]-a[j]>a[j]-a[i]) break;
                }
            }
        }
        return ans;
    }
};