forked from zhangwg1990/leetcode
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy path303.c
More file actions
64 lines (56 loc) · 1.69 KB
/
303.c
File metadata and controls
64 lines (56 loc) · 1.69 KB
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
#include <stdio.h>
#include <stdlib.h>
struct NumArray {
int *nums;
int *curSum;
int size;
};
/** Initialize your data structure here. */
struct NumArray* NumArrayCreate(int* nums, int numsSize) {
int i;
struct NumArray *res;
res = (struct NumArray *)malloc(sizeof(struct NumArray));
res->nums = (int *)malloc(sizeof(int) * numsSize);
res->curSum = (int *)malloc(sizeof(int) * numsSize + 1);
res->size = numsSize;
res->nums[0] = nums[0];
res->curSum[0] = 0;
for (i = 1; i < numsSize; ++i) {
res->nums[i] = nums[i];
res->curSum[i] = res->curSum[i - 1] + nums[i - 1];
}
res->curSum[i] = res->curSum[i - 1] + nums[i - 1];
return res;
}
int sumRange(struct NumArray* numArray, int i, int j) {
return (numArray->curSum[j + 1] - numArray->curSum[i]);
}
/** Deallocates memory previously allocated for the data structure. */
void NumArrayFree(struct NumArray* numArray) {
if (numArray) {
if (numArray->nums) free(numArray->nums);
free(numArray);
}
}
// Your NumArray object will be instantiated and called as such:
// struct NumArray* numArray = NumArrayCreate(nums, numsSize);
// sumRange(numArray, 0, 1);
// sumRange(numArray, 1, 2);
// NumArrayFree(numArray);
int main()
{
int sum, i;
int a[] = {-2, 0, 3, -5, 2, -1};
struct NumArray *numArray;
numArray = NumArrayCreate(a, 6);
printf("numArray->size = %d\n", numArray->size);
for (i = 0; i < numArray->size; ++i) {
printf("%d\n", numArray->nums[i]);
}
sum = sumRange(numArray, 0, 1);
printf("sum1 = %d\n", sum);
sum = sumRange(numArray, 1, 2);
printf("sum2 = %d\n", sum);
NumArrayFree(numArray);
return 0;
}