1512. 好数对的数目
思路:先用排序函数将数组进行排序,我们可以发现规律
举个栗子 相同数字 为 1
1 1 相同数字个数为 2 好数对为 1
1 1 1 相同数字个数为 2 好数对为 1+2
1 1 1 相同数字个数为 3 好数对为 1+2+3
因此我们可以通过统计相同数字的个数来计算好数对的个数
计算好数对的函数
int cmp(void const *a, void const *b) {
return *(int *)a - *(int *)b;
}
int sum(int n){//计算好数对函数
if(n==0) return 0;
return n+sum(n-1);
}
int numIdenticalPairs(int* nums, int numsSize){
qsort(nums,numsSize, sizeof(int), cmp);//排序
int num=0,pre=0,ans=0;
for(int i=0;i<numsSize-1;i++){
if(nums[i]==nums[i+1]){//注意i+1 越界问题
num++; //记录相同数字的个数
}else{
if(num!=0){
ans+=sum(num);
num=0;
}
}
if(i==numsSize-2&&num!=0){//特殊情况当nums[numsSize-2]和nums[numsSize-1]相等
ans+=sum(num);
}
}
return ans;
}
1711. 大餐计数
采用传统的双重遍历方法会出现超时
受英雄哥思路启发,我们可以通过排序寻找最大数maxi,那么数组中任意数字的两数之和必定小于 max = 2 * maxi。
直接双重遍历时间太长,我们可以反着来
通过2的对应次幂减去当前数并在type数组中直接查看是否有这个数即type[ 2的n次幂-当前数组的元素 ] 是否大于0。大于0说明数字存在那么我们加上统计总和
for(sum=1;sum<=max;sum<<=1) {
int temp=sum - deliciousness[i];
if (temp < 0) {
continue;
}
result+=type[temp];
result%=1000000007;
}
边遍历边记录数组中每个元素出现的个数记录在type数组中便于我们直接通过下标索引我们需要寻找的值避免重复遍历增加时间。
int type[ (1<<21) + 1 ];//记录数据出现次数,
//0 <= deliciousness[i] <= 2的20次方 所以开辟空间2的21次方加一
int cmp(const void* a,const void* b){
return *(int*)a - *(int*)b;
}
int countPairs(int* deliciousness, int deliciousnessSize){
int i, sum,max = 0;
int result = 0;
memset (type, 0, sizeof(type)); //将数组初始化为0
qsort(deliciousness,deliciousnessSize,sizeof(int),cmp);
max=deliciousness[deliciousnessSize-1]*2;
for(i = 0; i < deliciousnessSize;i++) {
for(sum=1;sum<=max;sum<<=1) {
int temp=sum - deliciousness[i];
if (temp < 0) {
continue;
}
result+=type[temp];
result%=1000000007;
}
++ type[ deliciousness[i] ];
}
return result;
}