| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 693228 | saayan007 | Solar Storm (NOI20_solarstorm) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using pi = pair<int, int>;
using pl = pair<ll, ll>;
using vi = vector<int>;
using vl = vector<ll>;
using vpi = vector<pi>;
using vpl = vector<pl>;
#define fur(i, a, b) for(ll i = a; i <= (ll) b; ++i)
#define ruf(i, a, b) for(ll i = a; i >= (ll) b; --i)
#define fr first
#define sc second
#define mp make_pair
#define pb push_back
#define eb emplace_back
#define all(x) (x).begin(),(x).end()
#define rall(x) (x).rbegin(),(x).rend()
#define nl "\n"
const ll mxn = 1e4L + 1;
#warning mxn is according to subtask
ll n, s, k;
ll d[mxn];
ll v[mxn];
ll x[mxn];
bool has[mxn] = {};
int main() {
ios_base::sync_with_stdio(0);
cin.tie(0);
cin >> n >> s >> k;
fur(i, 1, n - 1) {
cin >> d[i];
}
fur(i, 1, n) {
cin >> v[i];
}
x[1] = 0;
fur(i, 2, n) {
x[i] = x[i - 1] + d[i - 1];
}
ll res = 0;
vl ans;
vl tmp;
fur(i, 1, n) {
tmp.clear();
ll lt = i;
ll cur = i;
tmp.pb(cur);
fur(j, 2, s) {
ll nex = upper_bound(x + cur, x + n + 1, x[cur] + k) - x;
nex = upper_bound(x + nex, x + n + 1, x[nex] + k) - x - 1;
cur = nex;
tmp.pb(nex);
}
ll rt = cur;
ll opt = 0;
fur(j, 1, n) {
if(j < lt) {
if(x[j] + k >= x[lt])
opt += v[j];
} else if(j > rt) {
if(x[rt] + k >= x[j])
opt += v[j];
} else {
opt += v[j];
}
}
if(res < opt) {
res = opt;
swap(tmp, ans);
}
}
cout << ans.size() << nl;
for(auto u : ans)
cout << u << ' ';
cout << nl;
}