#include <bits/stdc++.h>
using namespace std;
#define LIFESUCKS ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0)
#define ll long long
#define ld long double
#define ar array
#include<cstdio>
#define vt vector
#include<fstream>
//ifstream fin("QUANMA.INP");
//ofstream fout("QUANMA.OUT");
#include<fstream>
#define pb push_back
#define all(c) (c).begin(), (c).end()
//#define length(x) (int)(x).size()
#define fi first
#define se second
#define vt vector
const int mxn = 1e6;
ll d[mxn + 3];
ll p[mxn + 3];
int main()
{
ll n, s, k; cin >> n >> s >> k;
ll crpos = 0;
vt<ll>prefd; prefd.pb(0);
vt<ll>prefp; prefp.pb(0);
for(int i = 0; i < n - 1; i++){
cin >> d[i];
prefd.pb(prefd[i] + d[i]);
}
for(int i = 0; i < n; i++){
cin >> p[i];
prefp.pb(prefp[i] + p[i]);
}
ll left[n], right[n];
llx id[n];
left[0] = 0;
for(int i = 1; i < n; i++){
int l = 0, r = i - 1, ans = i;
while(l <= r){
//cout << "aa";
int mid = (l + r) / 2;
if(prefd[i] - prefd[mid] <= k){
ans = mid; r = mid - 1;
}else{
l = mid + 1;
}
}
left[i] = ans;
}
right[n - 1] = n - 1;
for(int i = n - 2; i >= 0; i--){
int l = i + 1, r = n - 1, ans = i;
while(l <= r){
int mid = (l + r) / 2;
if(prefd[mid] - prefd[i] <= k){
ans = mid; l = mid + 1;
}else{
r = mid - 1;
}
}
right[i] = ans;
}
int cr = n - 1;
int mx[n];
for(int i = n - 1; i >= 0; i--){
while(cr >= left[i]){
mx[cr] = right[i];
id[cr] = i;
cr--;
}
}
ll ans = 0, idd = -1;
for(int i = 0; i < n; i++){
ll curr = p[i], pos = i;
for(int j = 0; j < s; j++){
if(pos == n - 1)break;
curr += prefp[mx[pos] + 1] - prefp[pos + 1];
pos = mx[pos];
}
if(curr > ans){
idd = i;
ans = curr;
}
}
vt<int>anss; anss.pb(id[idd]);
int pos = idd;
for(int i = 0; i < s - 1; i++){
if(pos == n - 1)break;
pos = mx[pos] + 1;
if(pos == n)break;
anss.pb(id[pos]);
}
cout << anss.size() << "\n";
for(auto i: anss)cout << i + 1 << " ";
return 0;
}
Compilation message
SolarStorm.cpp: In function 'int main()':
SolarStorm.cpp:40:5: error: 'llx' was not declared in this scope; did you mean 'll'?
40 | llx id[n];
| ^~~
| ll
SolarStorm.cpp:75:13: error: 'id' was not declared in this scope; did you mean 'i'?
75 | id[cr] = i;
| ^~
| i
SolarStorm.cpp:97:26: error: 'id' was not declared in this scope; did you mean 'idd'?
97 | vt<int>anss; anss.pb(id[idd]);
| ^~
| idd
SolarStorm.cpp:25:8: warning: unused variable 'crpos' [-Wunused-variable]
25 | ll crpos = 0;
| ^~~~~
