-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path2jan_kadane.cpp
102 lines (84 loc) · 2.25 KB
/
2jan_kadane.cpp
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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
/*
Author : Manas Rawat
Date : 02/01/2024
Problem : Largest Sum Subarray of Size at least K
Difficulty : Medium
Problem Link : https://www.geeksforgeeks.org/problems/largest-sum-subarray-of-size-at-least-k3121/1
Video Solution : https://youtu.be/u_hxfNXqb3g
*/
//{ Driver Code Starts
// C++ program to find largest subarray sum with
// at-least k elements in it.
#include <bits/stdc++.h>
using namespace std;
// } Driver Code Ends
class Solution{
public:
// long long int maxSumWithK(long long int a[], long long int n, long long int k)
// {
// vector<long long> dp(n + 1, 0);
// for(int i = n - 1; i > -1; i--){
// dp[i] = a[i] + dp[i + 1];
// dp[i] = max(dp[i], 0LL);
// }
// long long ans = -1e18;
// long long cur_sum = 0;
// for(int i = 0; i < k; i++)
// cur_sum += a[i];
// for(int i = k; i < n; i++){
// long long sum = cur_sum + dp[i];
// ans = max(ans, sum);
// cur_sum += a[i];
// cur_sum -= a[i - k];
// }
// return max(ans, cur_sum);
// }
// };
long long int maxSumWithK(long long int a[], long long int n, long long int k)
{
int ans=INT_MIN;
int c_sum=a[0];
vector<int> dp(n+1,0);
dp[0]=c_sum;
for(int i=1;i<n;i++)
{
if(c_sum>0)
c_sum+=a[i];
else
c_sum=a[i];
dp[i]=c_sum;
}
int exactK=0;
for(int i=0;i<k;i++)
{exactK+=a[i];
}
ans=max(ans,exactK);
for(int i=k;i<n;i++)
{
exactK+=a[i]-a[i-k];
ans=max(ans,exactK);
int moreThanK=dp[i-k]+exactK;
ans=max(ans,moreThanK);
}
return ans;
}
};
//{ Driver Code Starts.
// Driver code
int main() {
long long int t;
cin >> t;
while (t--) {
long long int n, k, i;
cin >> n;
long long int a[n];
for (i = 0; i < n; i++) {
cin >> a[i];
}
cin >> k;
Solution ob;
cout << ob.maxSumWithK(a, n, k) << endl;
}
return 0;
}
// } Driver Code Ends