2364. 统计坏数对的数目-数学推导转化+快速排序-搜索相同元素集合题解

2364. 统计坏数对的数目-力扣数学推导+快速排序题解

给你一个下标从 0 开始的整数数组 nums 。如果 i < j 且 j - i != nums[j] - nums[i] ，那么我们称 (i, j) 是一个 坏数对 。

请你返回 nums 中 坏数对 的总数目。

示例 1：

输入：nums = [4,1,3,3]
输出：5
解释：数对 (0, 1) 是坏数对，因为 1 - 0 != 1 - 4 。
数对 (0, 2) 是坏数对，因为 2 - 0 != 3 - 4, 2 != -1 。
数对 (0, 3) 是坏数对，因为 3 - 0 != 3 - 4, 3 != -1 。
数对 (1, 2) 是坏数对，因为 2 - 1 != 3 - 1, 1 != 2 。
数对 (2, 3) 是坏数对，因为 3 - 2 != 3 - 3, 1 != 0 。
总共有 5 个坏数对，所以我们返回 5 。

示例 2：

输入：nums = [1,2,3,4,5]
输出：0
解释：没有坏数对。

对于这一题，个人一开始还写写了一个二分查找发现不是能用得到，这一题我们需要做一个推导
我举个例子 a - b = c - d => a - c = b - d

那么我们本题进行一个转化，我们把nums[i] -> nums[i]-i

然后，就是如果nums[i]==nums[j] 说明他们之间不能组成坏数，我们求解所有相同元素得集合，然后求解他们二元子集合差值，之后，和总的二子集合数差值进行相减，即可得到最终答案

一个集合得二元子集合数量公式计算如下 ：
size*(size-1)/2,size 为集合大小

这一题，目前leedcodeC语言没有题解，这个还是很有趣的一个解法，感兴趣可以学习一下，解题代码如下：

void quick(int *a,int low,int high){
    if(low<high){
        int l=low,h=high,p=a[low];
        while(low<high){
            while(low<high&&a[high]>=p){
                high--;
            }
            a[low]=a[high];
           
            while(low<high&&a[low]<=p){
                low++;
            }
            a[high]=a[low];
           
        }
        a[low]=p;
       
        quick(a,l,low-1);
        quick(a,low+1,h);
    }
}

int find_index(int *nums,int val,int numsSize){
    int low=0,high=numsSize-1,mid;
    while(high<=low){
       
         mid=(low+high)/2;
        if(nums[mid]<val){
            low=mid+1;
        }
        else{
            high=mid-1;
        }

    }
    if(low>=0&&low<numsSize){
        if(nums[low]==val){
             return low;

        }
        else{
            return -1;
        }
       
    }
    else{
        return -1;
    }
}

int cmp(const void* a, const void* b)
 {
     return *(int*)a - *(int*)b;
 }



long long countBadPairs(int* nums, int numsSize){
   
    for(int i=0;i<numsSize;i++){
      
        nums[i]=nums[i]-i;
        
    }
    long long p=numsSize;
    int count=0;
    long long sum=0;
     qsort(nums,numsSize,sizeof(int),cmp);
    //quick(nums,0,numsSize-1);

     for(int i=0;i<numsSize;){
       int index=i;
       for(;i<numsSize;){
           if(nums[i]==nums[index]){
               i++;
           }
           else{
               break;

           }
       }
     long long count=i-index;
     sum=sum+count*(count-1)/2;

      
    }
  //  printf("-- %d ",count);
return p*(p-1)/2-sum;
    

    

}