2367. 算术三元组的数目

最新推荐文章于 2024-05-18 14:09:43 发布

飞哥不鸽

最新推荐文章于 2024-05-18 14:09:43 发布

阅读量256

点赞数

文章标签：算法 leetcode 数据结构

本文链接：https://blog.csdn.net/qq_62835094/article/details/128511885

版权

重新开启每日算法！！！！奥里给，新年新气象！！！
代码实现：

class Solution {
public:
    int arithmeticTriplets(vector<int>& nums, int diff) {
        int len = nums.size(), ans = 0;
        for (int i = 0; i < len; ++i) {
            bool flag = find(i, diff, nums);
            if (flag) ++ans;
        }    
        return ans;
    }
    // 查找并判断以nums[i]为左端点，diff为差值是否能形成三元组
    bool find(int i, int diff, vector<int>& nums) {
        int R = nums.size() - 1, L = i + 1, tar1 = nums[i] + diff, tar2 = tar1 + diff;
        return findBinary(nums, L, R, tar1) && findBinary(nums, L, R, tar2); 
    }
    // 二分查找tar
    bool findBinary(vector<int>& nums, int L, int R, int tar) {
        int lef = L, rig = R;
        while(lef <= rig) {
            int mid = (lef + rig) / 2;
            if (tar < nums[mid]) {
                rig = mid - 1; 
            }else if (tar > nums[mid]) {
                lef = mid + 1;
            }else {
                return true;
            }
        }
        return false;
    }
};

解题思路：
遍历所有的nums中的元素，并以其为左端点，判断是否存在另外两个数满足三元组条件。
具体的判断逻辑就是（i, j, k）
$n u m s [k] - n u m s [j] = n u m s [j] - n u m s [i] = d i ff$
为了寻找 $n u m s [k]$ 和 $n u m s [j]$ ，我们可以使用二分查找