文章目錄

• • 1. 題目
  • 2. 解題

1. 題目

鏈接：https://ac.nowcoder.com/acm/contest/10325/A
來源：牛客網

現在給定一個數組arr，和a,b兩個數字，你要做的就是找到（i，j，k）。且滿足
1. 0 <= i < j < k < arr.size()
2. |arr[i] - arr[j]| <= a
3. |arr[j] - arr[k]| <= b
統計滿足條件的個數并返回(最后結果可能很大，請取1000000007的余數)。

arr.size() <= 5000
其余變量均<=1e9

2. 解題

• 枚舉中間點，時間復雜度 $O(n^2)$

class Solution { public:/*** 代碼中的類名、方法名、參數名已經指定，請勿修改，直接返回方法規定的值即可* * @param arr int整型vector * @param a int整型 * @param b int整型 * @return int整型*/int countTriplets(vector<int>& arr, int a, int b) {// write code hereint n = arr.size();if(n < 3) return 0;long long ans = 0, mod = 1e9+7;for(int j = 1; j < n-1; j++) {long long ct1 = 0, ct2 = 0;for(int i = 0; i < j; i++){if(abs(arr[i]-arr[j]) <= a)ct1++;}for(int k = j+1; k < n; k++){if(abs(arr[j]-arr[k]) <= b)ct2++;}ans = (ans+ct1*ct2)%mod;}return ans;} };

---

我的CSDN博客地址 https://michael.blog.csdn.net/

長按或掃碼關注我的公眾號（Michael阿明），一起加油、一起學習進步！

總結

以上是生活随笔為你收集整理的牛客 统计好元组的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。