結果
問題 | No.258 回転寿司(2) |
ユーザー |
![]() |
提出日時 | 2016-05-06 10:35:51 |
言語 | C++11(廃止可能性あり) (gcc 13.3.0) |
結果 |
AC
|
実行時間 | 2 ms / 2,000 ms |
コード長 | 1,430 bytes |
コンパイル時間 | 1,743 ms |
コンパイル使用メモリ | 164,712 KB |
実行使用メモリ | 5,248 KB |
最終ジャッジ日時 | 2024-11-06 19:01:53 |
合計ジャッジ時間 | 4,781 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 67 |
ソースコード
#include <bits/stdc++.h>typedef long long ll;typedef unsigned long long ull;#define FOR(i,a,b) for(int (i)=(a);i<(b);i++)#define REP(i,n) FOR(i,0,n)#define RANGE(vec) (vec).begin(),(vec).end()using namespace std;class ConveyorBeltSushi2{public:void solve(void){int N;cin>>N;vector<int> vi(N);REP(i,N)cin>>vi[i];vector<vector<int>> dp(N+1,vector<int>(2,0));dp[1][0] = 0;dp[1][1] = vi[0];FOR(i,1,N){dp[i+1][0] = max(dp[i][0], dp[i][1]);dp[i+1][1] = max({dp[i][0]+vi[i], dp[i-1][0]+vi[i], dp[i-1][1]+vi[i]});}vector<int> ret;int n = N;int w = max(dp[n][0],dp[n][1]);while (w){if ( dp[n][1] == w ){ret.push_back(n);w -= vi[n-1];if ( n >= 2 ){if (dp[n-2][0] == w || dp[n-2][1] == w)--n;}}--n;}cout<<max(dp[N][0],dp[N][1])<<endl;reverse(RANGE(ret));for (auto j : ret)cout<<j<<" ";cout<<endl;}};#if 1int main(int argc, char *argv[]){ios::sync_with_stdio(false);auto obj = new ConveyorBeltSushi2();obj->solve();delete obj;return 0;}#endif