Skip to content

Commit d097dce

Browse files
authored
Merge pull request Sunchit#1054 from samueldenzil/master
Java & C++ Solution of Day 20 - June Challenge | K Radius Subarray Averages
2 parents de9dbff + 83815fb commit d097dce

File tree

2 files changed

+44
-0
lines changed

2 files changed

+44
-0
lines changed
Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
1+
class Solution {
2+
public:
3+
vector<int> getAverages(vector<int>& nums, int k) {
4+
int n = nums.size();
5+
6+
vector<long> prefix(n + 1, 0);
7+
for (int i = 0; i < n; ++i) {
8+
prefix[i + 1] = prefix[i] + nums[i];
9+
}
10+
11+
vector<int> ans(n, -1);
12+
13+
for (int i = k; i < (n - k); ++i) {
14+
int leftBound = i - k, rightBound = i + k;
15+
long sum = prefix[rightBound + 1] - prefix[leftBound];
16+
int avg = (int) (sum / (2 * k + 1));
17+
ans[i] = avg;
18+
}
19+
20+
return ans;
21+
}
22+
};
Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
1+
class Solution {
2+
public int[] getAverages(int[] nums, int k) {
3+
int n = nums.length;
4+
5+
long[] prefix = new long[n + 1];
6+
for (int i = 0; i < n; ++i) {
7+
prefix[i + 1] = prefix[i] + nums[i];
8+
}
9+
10+
int[] ans = new int[n];
11+
Arrays.fill(ans, -1);
12+
13+
for (int i = k; i < (n - k); ++i) {
14+
int leftBound = i - k, rightBound = i + k;
15+
long sum = prefix[rightBound + 1] - prefix[leftBound];
16+
int avg = (int) (sum / (2 * k + 1));
17+
ans[i] = avg;
18+
}
19+
20+
return ans;
21+
}
22+
}

0 commit comments

Comments
 (0)