結果
問題 |
No.258 回転寿司(2)
|
ユーザー |
|
提出日時 | 2023-04-28 17:17:36 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 3 ms / 2,000 ms |
コード長 | 1,306 bytes |
コンパイル時間 | 3,918 ms |
コンパイル使用メモリ | 250,060 KB |
実行使用メモリ | 6,820 KB |
最終ジャッジ日時 | 2024-11-17 16:55:48 |
合計ジャッジ時間 | 7,188 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 67 |
ソースコード
#include <bits/stdc++.h> using namespace std; int main() { ios::sync_with_stdio(false); cin.tie(nullptr); int n; cin >> n; vector<int> a(n); for (int i = 0; i < n; i++) { cin >> a[i]; } vector<array<int, 2>> dp(n + 1); vector<array<int, 2>> pre(n + 1); for (int i = 0; i < n; i++) { dp[i][0] = dp[i][1] = 0; } for (int i = 0; i < n; i++) { if (dp[i][0] >= dp[i][1]) { dp[i + 1][0] = dp[i][0]; pre[i + 1][0] = 0; } else { dp[i + 1][0] = dp[i][1]; pre[i + 1][0] = 1; } dp[i + 1][1] = max(dp[i + 1][1], dp[i][0] + a[i]); pre[i + 1][1] = 0; } vector<int> ans; if (dp[n][0] >= dp[n][1]) { int j = 0; for (int i = n; i >= 1; i--) { if (j == 1) { ans.push_back(i); } j = pre[i][j]; } } else { int j = 1; for (int i = n; i >= 1; i--) { if (j == 1) { ans.push_back(i); } j = pre[i][j]; } } reverse(ans.begin(), ans.end()); cout << max(dp[n][0], dp[n][1]) << '\n'; for (int i = 0; i < (int) ans.size(); i++) { cout << ans[i] << " \n"[i + 1 == (int) ans.size()]; } }