题目
我的思路
看到这题首先想到的就是直接遍历所有元素,简单暴力且不会出错
/**
* Note: The returned array must be malloced, assume caller calls free().
*/
int* twoSum(int* nums, int numsSize, int target, int* returnSize){
int* rec=(int*)malloc(sizeof(int)*2);
int i,j;
for(i=0;i<numsSize-1;i++)
{
for(j=i+1;j<numsSize;j++)
{
if(nums[i]+nums[j]==target)
{
rec[0]=i;
rec[1]=j;
break;
}
}
}
* returnSize=2;
return rec;
}
结果
优化解法二分内二分
int bisect_right(int *nums, int *index, int left, int right, int target, bool isright) {
while (left < right) {
int mid = left + (right - left) / 2;
if (nums[index[mid]] > target) {
right = mid;
} else if (nums[index[mid]] == target) {
return mid;
} else left = mid + 1;
}
if (isright) return left - 1; // 没有退出就是左边小一,否则就是left情况
return left;
}
int* twoSum(int* nums, int numsSize, int target, int* returnSize){
int *index = (int *)malloc(sizeof(int) * numsSize);
for (int i = 0; i < numsSize; i++) index[i] = i;
int cmp(int *a, int *b) {
return nums[*a] - nums[*b];
}
qsort(index, numsSize, sizeof(index[0]), cmp);
int left = 0, right = numsSize - 1;
while (left < right) { // 闭区间,因为存在[2,3,3],target=6
if (nums[index[left]] + nums[index[right]] < target) {
// 比target小,因为index升序,因此一定是左边小!
left = bisect_right(nums, index, left, right, target - nums[index[right]], false);
} else if (nums[index[left]] + nums[index[right]] > target) {
// 这时有可能死循环,[1,2,5,7,9],target=9,上界始终是9
right = bisect_right(nums, index, left, right, target - nums[index[left]], true);
} else {
int *res = (int *)malloc(sizeof(int) * 2);
res[0] = index[left];
res[1] = index[right];
*returnSize = 2;
return res;
}
}
return NULL;
}
每天进步一点点,每天收获一点点.