File tree Expand file tree Collapse file tree 2 files changed +44
-0
lines changed
Expand file tree Collapse file tree 2 files changed +44
-0
lines changed Original file line number Diff line number Diff line change 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+ };
Original file line number Diff line number Diff line change 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+ }
You can’t perform that action at this time.
0 commit comments