Ahmed_Negm

Untitled

Apr 23rd, 2022
89
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.44 KB | None | 0 0
  1. #include<iostream>
  2. #include<cmath>
  3. #include<iomanip>
  4. #include<algorithm>
  5. #include<cstdlib>
  6. #include<cstring>
  7. #include<vector>
  8. #include<utility>
  9.  
  10. #define ll long long
  11. #define sz(x) int(x.size())
  12. #define all(x) x.begin(),x.end()
  13. #define rall(s)  s.rbegin(), s.rend()
  14. #define ceill(n, m) (((n) / (m)) + ((n) % (m) ? 1 : 0))
  15. using namespace std;
  16.  
  17. void Fast_IO(){
  18.     ios_base::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
  19.     #ifndef ONLINE_JUDGE
  20.         freopen("input.txt", "r", stdin), freopen("output.txt", "w", stdout);
  21.     #endif
  22. }
  23.  
  24.  
  25.  
  26.  
  27.  
  28. void solve(){
  29.     ll n,k; cin>>n>>k;
  30.     vector<ll>v(n+1);
  31.     vector<ll>slept(n+1);
  32.     vector<ll>prefix(n+1);
  33.      ll res =0;
  34.     ll maxi = -__LONG_LONG_MAX__;
  35.     for(int i=1; i<=n;i++){
  36.         cin>>v[i];
  37.     }
  38.     for(int i=1; i<=n; i++){
  39.         cin>>slept[i];
  40.         if(slept[i] == 1 ){
  41.             res += v[i];
  42.         }
  43.         if(slept[i]==0){
  44.             prefix[i]=prefix[i-1] + v[i];
  45.         }else prefix[i]=prefix[i-1];
  46.     }
  47.  
  48.     // for(int i=1; i<=n;i++) cout<<prefix[i]<<' ';
  49.     // cout<<'\n';
  50.     for(int i=1; i<=n-k+1; i++){
  51.             if(prefix[i+k-1]-prefix[i-1]>maxi){
  52.             maxi = prefix[i+k-1]-prefix[i-1];
  53.             }
  54.             // cout<<prefix[i+k-1]-prefix[i-1]<<'\n';
  55.         // cout<<maxi<<'\n';
  56.  
  57.     }
  58.    
  59.    
  60.     cout<<res+maxi;
  61.    
  62.  
  63. }
  64.  
  65. int main(){
  66.     Fast_IO();
  67. int t =1;
  68. //cin>>t;
  69. while(t--){
  70. solve();
  71. }
  72. return 0;
  73. }  
Advertisement
Add Comment
Please, Sign In to add comment