问题描述:
给定一组数字a[N], 一个常数(比如 int target) ,要求在这一堆数里面找到K个数字,使得这K个数字的和等于target。
注意这一组数字可能有重复项:比如 1 1 2 3 , 求3sum, 然后 target = 6, 你搜的时候可能会得到 两组1 2 3, 1 2 3,1 来自第一个1或者第二个1, 但是结果其实只有一组,所以最后结果要去重。
求解方法:
方法一:
暴力穷举,就是枚举所有的K个数相加的情况(写K个for循环), 那么这样的复杂度就是 从N选出K个,复杂度是O(N^K)
方法二:
排序,这个算法可以考虑最简单的case, 2sum,这是个经典问题,方法就是先排序,然后利用头尾指针找到两个数使得他们的和,由于找数的过程是线性的,所以复杂度是排序的复杂度。
同样的,3个数字就是 先取出一个数,只要在剩下的数字里面找到两个数字使得他们的和等于(target – 那个取出的数)就可以了,这样退化为2个数的问题。 利用这个思路K-sum问题能做到O(n^(K-1))复杂度
下面是两个数的核心代码:
#include <stdio.h>
//找到a[i]+a[j] = sum,先排序,再利用下面的算法处理,时间复杂度是排序的复杂度。
int main()
{
int a[6] = {-1, 1, 1, 2, 4, 7};
int i, j;
int sum = -1;
i = 0;
j = 5;
while (i<j)
{
if (a[i]+a[j] == sum)
break;
else if (a[i]+a[j] > sum){
j--;
}
else{
i++;
}
}
printf("%d, %d", i, j);
return 0;
}