ARC092 E - Both Sides Merger
問題ページ
D - Two Sequences
考えたこと
- 構築ゲーっぽいのでとりあえず実験
- どうがんばっても隣の要素を足すのは不可能そう
- 位置の偶奇が違う要素はどうがんばっても足せなさそう
- 全部正の数なら奇数番目の要素と偶数番目の要素の和で大きい方が答えになりそう
- マイナスをどうやって消すか?
- マイナスの要素を選択するとその両隣の2つが消える
- 1番目 + 3番目 + … + n番目 と奇数番目を足し合わせたいときに両隣の偶数番目が消えてもどうでもいい
- マイナスは消しされることがわかった
- 奇数番目の正の数の和 と 偶数番目の正の数の和 のうち大きいほうが答えになりそう
- 構成をどうしようか考える
- とりあえずマイナスを全部消す
- そのあと奇数番目の和がほしいなら偶数番目、偶数番目がほしいなら奇数番目を全部消せばよい
- 前から消していくとその分ずれるので消す場所は全部2か1,2,2,2,2,…のどっちかになる
- バグらせそうなのを頑張って実装すると通る
- 全部マイナスのときに0を出力したりしないように注意
ソースコード
#define __USE_MINGW_ANSI_STDIO 0 #include <bits/stdc++.h> using namespace std; using ll = long long; #define int ll using VI = vector<int>; using VVI = vector<VI>; using PII = pair<int, int>; #define FOR(i, a, n) for (ll i = (ll)a; i < (ll)n; ++i) #define REP(i, n) FOR(i, 0, n) #define ALL(x) x.begin(), x.end() #define PB push_back const ll LLINF = (1LL<<60); const int INF = (1LL<<30); const int MOD = 1000000007; template <typename T> T &chmin(T &a, const T &b) { return a = min(a, b); } template <typename T> T &chmax(T &a, const T &b) { return a = max(a, b); } template <typename T> bool IN(T a, T b, T x) { return a<=x&&x<b; } template<typename T> T ceil(T a, T b) { return a/b + !!(a%b); } template<class S,class T> ostream &operator <<(ostream& out,const pair<S,T>& a){ out<<'('<<a.first<<','<<a.second<<')'; return out; } template<class T> ostream &operator <<(ostream& out,const vector<T>& a){ out<<'['; REP(i, a.size()) {out<<a[i];if(i!=a.size()-1)out<<',';} out<<']'; return out; } int dx[] = {0, 1, 0, -1}, dy[] = {1, 0, -1, 0}; signed main(void) { cin.tie(0); ios::sync_with_stdio(false); int n; cin >> n; VI a(n); REP(i, n) cin >> a[i]; int suma = 0, sumb = 0; bool flaga = true, flagb = true; REP(i, n) { if(i%2==0 && a[i]>=0) suma += a[i], flaga = false; if(i%2 && a[i]>=0) sumb += a[i], flagb = false; } if(flaga && flagb) { int ma = -LLINF, idx = -1; REP(i, n) { if(ma < a[i]) { ma = a[i]; idx = i; } } cout << ma << endl; cout << n-1 << endl; REP(i, idx) cout << 1 << endl; for(int i=n-idx; i>1; --i) cout << i << endl; return 0; } if(suma > sumb) { cout << suma << endl; bool flag = true; int num1 = 0, num2 = 0; VI ans; REP(i, n) { if(i%2==0 && a[i]<0) { ans.PB(i-num1); if(i==0||i==n-1) { num1++; if(flag) num2++; } else { num1+=2; if(flag) num2+=2; } } if(i%2==0 && a[i]>=0) { flag = false; } } if(num2%2) { ans.PB(0); REP(i, n-num1) { if(i%2==0) ans.PB(1); } ans.pop_back(); } else { REP(i, n-num1) { if(i%2) ans.PB(1); } } cout << ans.size() << endl; REP(i, ans.size()) cout << ans[i]+1 << endl; } else { cout << sumb << endl; bool flag = true; int num1 = 0, num2 = 0; VI ans; REP(i, n) { if(i%2 && a[i]<0) { ans.PB(i-num1); if(i==0 || i==n-1) { num1++; if(flag) num2++; } else { num1+=2; if(flag) num2+=2; } } if(i%2 && a[i]>=0) flag = false; } if(num2%2) { REP(i, n-num1) { if(i%2) ans.PB(1); } } else { ans.PB(0); REP(i, n-num1) { if(i%2==0) ans.PB(1); } ans.pop_back(); } cout << ans.size() << endl; REP(i, ans.size()) cout << ans[i]+1 << endl; } return 0; }