Submission #1292237

#TimeUsernameProblemLanguageResultExecution timeMemory
1292237chaitanyamehtaSolar Storm (NOI20_solarstorm)C++20
28 / 100
519 ms39432 KiB
// 
#include<bits/stdc++.h>
using namespace std;
#define int long long
#define pii pair<int ,int>
#define v vector
#define all(a) a.begin() , a.end()

signed main(){
    int n , s , k;
    cin>>n>>s>>k;
    vector<int> d(n + 1 , 0) , di(n + 1, 0);
    for(int i = 1 ; i <= n-1; i++){
        cin>>di[i];
    }
    v<int> v(n +1, 0) , impact(n + 1 , 0) , p(n + 1 , 0);
    for(int i = 1 ;i <= n ; i++)cin>>v[i];
    for(int i = 1; i<=n;i++){
        p[i] = p[i-1]+v[i];
    }
    // cout << d[0] << " ";
    d[1] = 0;
    // cout << d[1] <<" ";
    for(int i = 2 ; i <= n ; i++){
        d[i] = di[i - 1] + d[i - 1];
        // cout << d[i] << " ";
    }
    // cout << "\n";
     for(int i = 1 ;i<=n;i++){
         
        int left = max(0LL , d[i] - k);
        int right = min(d[n] , d[i] + k);
        int left_idx = max(0LL ,( (long long)(lower_bound(all(d) , left) - d.begin()) - 1LL));
        int right_idx = ((long long)(upper_bound(all(d) , right) - d.begin())) - 1LL;
        // cout << left_idx << " "<<right_idx << "\n";
         
        int sum = 0;
        sum += p[i] - p[left_idx];
        sum += p[right_idx] - p[i - 1];
        impact[i] = sum - v[i];
        // cout << impact[i] << " ";
    }
    cout << 1 << "\n";
    int ans = max_element(impact.begin() , impact.end()) - impact.begin();
    cout << ans;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...