Submission #1136806

#TimeUsernameProblemLanguageResultExecution timeMemory
1136806Roumak77Solar Storm (NOI20_solarstorm)C++20
28 / 100
170 ms37056 KiB
#pragma GCC optimize ("O3")
#pragma GCC optimize ("unroll-loops")
#pragma GCC optimize("-Ofast")
#include <bits/stdc++.h>
#include <algorithm>
#include <iostream>
#include <vector>
#include <limits>
#include <cmath>
#include <stack>
#include <queue>
#include <map>
#include <math.h>
using namespace std;


/*
 * Subtask Marks Additional Constraints

    6) S ≤ 50 -> 16 points
    7) No constraints -> 30 points


    Total = 54

 */


using ll = long long;

void solve(){


    ll n, s, k;
    cin >> n >> s >> k;
    vector<ll> list_n(n, 0);

    for(ll i = 1; i < n; i++){
        cin >> list_n[i];
        list_n[i] += list_n[i - 1];
    }

    vector<ll> list_k(n);
    vector<ll> pref(n + 1, 0);
    for(ll i = 0; i < n; i++){
        cin >> list_k[i];
        pref[i + 1] = pref[i] + list_k[i];
    }

    ll curr_max = pref[s];
    vector<ll> fin(s);
    for(ll i = 0; i < s; i++){
        fin[i] = i;
    }
    ll Ubound = n;
    for(ll i = 0; i < n; i++){
        if(i >= Ubound){
            break;
        }
        ll rem = s;
        vector<ll> temp;
        ll l1 = -1;
        ll r1 = i;

        while(l1 + 1 < r1){
            ll mid = (l1 + r1)/2;

            if(list_n[i] - list_n[mid] <= k){
                r1 = mid;
            }else{
                l1 = mid;
            }
        }

        rem--;
        temp.push_back(i);


        ll l2 = i;
        ll r2 = n;
        while(l2 + 1 < r2){
            ll mid = (l2 + r2)/2;
            if(list_n[mid] - list_n[i] <= k){
                l2 = mid;
            }else{
                r2 = mid;
            }
        }

        ll start = r1;
        ll end = l2;

        while(rem > 0){
            ll new_idx = end + 1;
            if(new_idx >= n){
                break;
            }
            ll l = new_idx;
            ll r = n;
            while(l + 1 < r){
                ll mid = (l + r)/2;
                if(list_n[mid] - list_n[new_idx] <= k){
                    l = mid;
                }else{
                    r = mid;
                }
            }
            rem--;
            ll new_add = l;
            temp.push_back(l);
            r = n;
            while(l + 1 < r){
                ll mid = (l + r)/2;
                if(list_n[mid] - list_n[new_add] <= k){
                    l = mid;
                }else{
                    r = mid;
                }
            }
            end =  l;
        }
        if(r1 > 0 && Ubound == n && fin.size() >= 2){
            Ubound = fin[1];
        }
        ll val = pref[end + 1] - pref[start];
        if(val > curr_max){
            fin = temp;
            curr_max = val;
        }


    }

    cout << fin.size() << endl;
    for(ll i : fin){
        cout << i + 1 << " ";

    }

    //cout << curr_max << endl;





}

bool single = true;

signed main(){

    ios_base::sync_with_stdio(false);
    cout.tie(0);
    cin.tie(0);

    ll t = 1;
    if(!single) cin >> t;

    while(t--){
        solve();
    }
}
#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...