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;
}
};