【练习题】三数之和（有序数组）

【练习题】三数之和（有序数组）

总体思路：

将求三数和的问题变成two_sum（求两数之和）

题目：

给定一个从小到大排列的有序数组，在数组中选三个下标不同的元素使得他们的和等于给定的target值

• 实现three_sum函数，传入的nums是数组，length是元素个数，target表示三数值和。

int* three_sum(int *nums, int length, int target) {
    int *ans = malloc(sizeof(int) * 3);
    
    return ans;
}

思路：

时间复杂度：O(n^2)

• 在数组中按序（从头到尾或从尾到头）遍历生成新的和new_target = target - nums[i]
• 在剩下的数中进行two_sum——双指针查找，若new_target 小于双指针所指两数的和，头指针向后走，否则尾指针向前走（因为有序数组从小到大排列）

int* three_sum(int *nums, int length, int target) {
    int *ans = (int *)malloc(sizeof(int) * 3);
	int sum = target;
    for (int i = length - 1; i >= 2; i--) {
        sum = target - 	nums[i];
        int l = 0, r = i - 1;
        while (l < r) {
            if (nums[l] + nums[r] == sum) {
                ans[0] = l; ans[1] = r; ans[2] = i;
            	return ans;
            }
            if (nums[l] + nums[r] < sum) l++;
            else r--;
        }
    }
    return ans;
}