原题链接 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 }