B.统计好元组(巅峰赛S2第12场)

https://ac.nowcoder.com/acm/contest/10324/B

我当时的做法是从i入手，对于每个i去从i+1开始到尾去找k，再对于每个j从j+1开始到尾去找k…三重循环，每次都是头扫到尾，超时，卒.

正解是从j入手，因为j与i跟k都有关联，对于每个j，从j-1到头去找i，从j+1到尾去找k，分别计数，然后相乘即可。n^2的复杂度。

    int countTriplets(vector<int>& arr, int a, int b) {
        // write code here
        long long ans=0;
        long long x=0,y=0;
        int k=0;
        for(int j=1;j<arr.size()-1;j++){
            x=0,y=0;
            for(int i=j-1;i>=0;i--){
               if(abs(arr[j]-arr[i])<=a){
                   y++;
               }
            }
            for(int k=j+1;k<arr.size();k++){
               if(abs(arr[j]-arr[k])<=b){
                   x++;
               }
            }
            ans+=x*y;
        }
        ans=ans%1000000007;
        return ans;
    }
};