forked from neetcode-gh/leetcode
- Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path0015-3sum.c
72 lines (64 loc) · 2.19 KB
/
0015-3sum.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
intcompareInt(constvoid*a, constvoid*b) {
return*(int*)a-*(int*)b;
}
/**
* Return an array of arrays of size *returnSize.
* The sizes of the arrays are returned as *returnColumnSizes array.
* Note: Both returned array and *columnSizes array must be malloced, assume caller calls free().
*/
int**threeSum(int*nums, intnumsSize, int*returnSize, int**returnColumnSizes) {
intcap=32;
int**result= (int**)malloc(sizeof(int*) *cap);
int*cols= (int*)malloc(sizeof(int) *cap);
int*triplet;
intsum, target, count;
inti, left, right;
qsort(nums, numsSize, sizeof(int), compareInt);
count=0;
for (i=0; i<numsSize-2; i++) {
// The array is sorted, there is no possible tripet when this happen.
if (nums[i] >0) {
break;
}
// skip element with same value to avoid duplicate triplets.
if (i>0&&nums[i] ==nums[i-1]) {
continue;
}
target=0-nums[i];
left=i+1;
right=numsSize-1;
while (left<right) {
sum=nums[left] +nums[right];
if (sum>target) {
right--;
} elseif (sum<target) {
left++;
} else {
triplet= (int*)malloc(sizeof(int) *3);
triplet[0] =nums[i];
triplet[1] =nums[left];
triplet[2] =nums[right];
result[count] =triplet;
cols[count] =3;
count++;
if (count==cap) {
cap *= 2;
result= (int**)realloc(result, sizeof(int*) *cap);
cols= (int*)realloc(cols, sizeof(int) *cap);
}
left++;
right--;
// skip element with same value to avoid duplicate triplets.
while(left<right&&nums[left] ==nums[left-1]) {
left++;
}
while(left<right&&nums[right] ==nums[right+1]) {
right--;
}
}
}
}
*returnSize=count;
*returnColumnSizes=cols;
returnresult;
}