SuitNdtie

Guk

Apr 24th, 2019
147
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 0.65 KB | None | 0 0
  1. #include<stdio.h>
  2. #include<algorithm>
  3. using namespace std;
  4. typedef long long int ll;
  5.  
  6. int main(){
  7.     ll n;
  8.     ll d;
  9.     scanf("%lld %lld",&n,&d);
  10.     ll arr[n+1];
  11.     for(ll i = 1 ; i <= n ; i ++){
  12.         scanf("%lld",&arr[i]);
  13.     }
  14.     sort(arr+1,arr+n+1);
  15. /*  for(int i = 1 ; i <= n ; i++){
  16.         printf("%d ",arr[i]);
  17.     }printf("\n");*/
  18.     ll i = 1;
  19.     ll ans = 0;
  20.     while(i <= n){
  21.         ll j = i + 1;
  22.         while(j <= n && arr[j] - arr[i] <= d){
  23.             j++;
  24.         }
  25.         ll cnt = j - i - 1;
  26.     //  printf("Test (%d,%d) -> %d : %d\n",i,j,cnt,(int)((double)(cnt+1)*((double)cnt/2)));
  27.         ans += (ll)((double)(cnt+1)*((double)cnt/2));
  28.         i = j - 1;
  29.         i++;
  30.     }
  31.     printf("%lld",ans);
  32.     return 0;
  33. }
Advertisement
Add Comment
Please, Sign In to add comment