結果

問題 No.258 回転寿司(2)
コンテスト
ユーザー @abcde
提出日時 2019-06-08 21:28:07
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 3 ms / 2,000 ms
コード長 2,035 bytes
コンパイル時間 1,738 ms
コンパイル使用メモリ 167,372 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-11-06 19:25:54
合計ジャッジ時間 5,301 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 67
権限があれば一括ダウンロードができます

ソースコード

diff # 

// TODO: ロジック検討中.
#include <bits/stdc++.h>
using namespace std;

int dp[1001][2]; // dp[i][0]: i皿目を食べない場合を保存, dp[i][1]: i皿目を食べた場合を保存.
int v[1001];
int memo[1001];

// 消費状況メモの更新.
// @param ans: 得られる美味しさの最大値.
// @param N: お寿司の数.
// @return : 特に無し.
void update(int ans, int N){

    if(dp[N][0] < dp[N][1]) memo[N] = 2;
    else                    memo[N] = 1;
    for(int i = N - 1; i >= 0; i--){
        if(memo[i + 1] == 2) memo[i] = 1;
        if(memo[i + 1] == 1){
            if(dp[i][0] >= dp[i][1]) memo[i] = 1;
            if(dp[i][0] <  dp[i][1]) memo[i] = 2;
        }
    }
    return;

}

int main() {
    
    // 1. 入力情報取得.
    int N;
    cin >> N;
    for(int i = 1; i < N + 1; i++) cin >> v[i];
    
    // 2. dp[i] を i皿目までの消費状況を保存する.
    dp[1][0] = 0;    // 1皿目を食べない場合.
    dp[1][1] = v[1]; // 1皿目を食べた場合.
    for(int i = 2; i < N + 1; i++){
        // 2-1. i皿目 を 食べない場合は, (i - 1)皿目 を 食べた場合, 食べない場合の大きい方を保存.
        dp[i][0] = max(dp[i - 1][0], dp[i - 1][1]);
        // 2-2. i皿目 を 食べる場合は, (i - 1)皿目 を 食べてない状態から計算.
        dp[i][1] = dp[i - 1][0] + v[i];
    }
    
    // 3. 美味しさの最大値を計算.
    // ex.
    // [入力例]
    // 7
    // 1 2 9 10 3 1 4
    // -> 17 の はず.
    // for(int i = 0; i < N + 1; i++) cout << dp[i][0] << " " << dp[i][1] << endl;
    int ans = max(dp[N][0], dp[N][1]);
    cout << ans << endl;
    
    // 4. 取ったお皿を確認.
    update(ans, N);
    
    // 5. 出力.
    // for(int i = 1; i < N + 1; i++) cout << memo[i] << " ";
    // cout << endl;
    // for(int i = 1; i < N + 1; i++) if(memo[i] == 2) cout << v[i] << " ";
    // cout << endl;
    for(int i = 1; i < N + 1; i++) if(memo[i] == 2) cout << i << " ";
    cout << endl;
    return 0;
    
}
0