Gym-100676E Time Limit Exceeded?-优快云博客

本文介绍了一种解决计数问题的有效算法，通过标记数组元素并查找每个元素前后31个数的方法来避免暴力枚举超时，利用前缀和技巧快速统计符合条件的数对数量。

原题链接 https://odzkskevi.qnssl.com/1110bec98ca57b5ce6aec79b210d2849?v=1491063604

**********************************************************************************************************************

题意:选择v[i]里的两个数相减的绝对值小于32，问你有多少对这样的数

解题思路：看题目就知道暴力枚举肯定会超时XD;

　　　　　所以就标记一下v[i]，每一次找v[i]前后31个数，看看有多少个, 减去重复的，就是答案。

**********************************************************************************************************************

 1 #include <iostream>
 2 #include <algorithm>
 3 #include <cstring>
 4 using namespace std;
 5 const int maxn = 10000+10;
 6 int num[maxn];
 7 int a[maxn];
 8 int sum[maxn];
 9 int vis[maxn];
10 int main()
11 {
12     int T;
13     cin >> T;
14     while(T--){
15         memset(a, 0, sizeof(a));
16         memset(num, 0, sizeof(num));
17         memset(vis, 0, sizeof(vis));
18         int n, i, ans=0;
19         cin >> n;
20 
21         for(i=0;i<n;i++){
22             cin >> a[i];
23             num[a[i]]++;
24         }
25         sum[0]=0;
26         for(i=1;i<maxn;i++)//通过前缀和的方法求出有多少个符合
27             sum[i]=sum[i-1]+num[i];
28 //        for(i=1;i< 150 ;i++)
29 //            cout << i << " " << sum[i] << endl;
30         for(i=0;i<n;i++){
31             int x = a[i];
32             int rightn=min(10000, x+31);//只往右边跑，避免左右一起跑出现重复
33             if(!vis[x]){
34                 ans+=num[x] * (sum[rightn] - sum[x]);//加上[x,x+31]的数的个数
35                 ans+=num[x]*(num[x]-1)/2;//如果同一个数出现了多次,C(2, n)个
36                 vis[x]=1;
37             }
38         }
39         cout << ans  << endl;
40     }
41     return 0;
42 }