結果

問題 No.258 回転寿司(2)
ユーザー sekiya9311
提出日時 2016-10-19 09:18:26
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 3 ms / 2,000 ms
コード長 2,704 bytes
コンパイル時間 1,102 ms
コンパイル使用メモリ 118,984 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-06 19:09:15
合計ジャッジ時間 4,561 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 67
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <string>
#include <queue>
#include <stack>
#include <algorithm>
#include <list>
#include <vector>
#include <complex>
#include <utility>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <climits>
#include <bitset>
#include <ctime>
#include <map>
#include <unordered_map>
#include <set>
#include <unordered_set>
#include <cassert>
#include <cstddef>
#include <iomanip>
#include <numeric>
#include <tuple>
#include <sstream>
#include <fstream>
#include <random>

#define REP(i, n) for (int (i) = 0; (i) < (n); (i)++)
#define FOR(i, a, b) for (int (i) = (a); (i) < (b); (i)++)
#define RREP(i, a) for (int (i) = (a) - 1; (i) >= 0; (i)--)
#define FORR(i, a, b) for (int (i) = (a) - 1; (i) >= (b); (i)--)
#define PI acos(-1.0)
#define DEBUG(C) cerr << C << endl;
#define VI vector <int>
#define VII vector <VI>
#define VL vector <LL>
#define VLL vector <VL>
#define VD vector <double>
#define VDD vector <VD>
#define PII pair <int, int>
#define PDD pair <double, double>
#define PLL pair <LL, LL>
#define VPII vector <PII>
#define ALL(a) (a).begin(), (a).end()
#define SORT(a) sort(ALL(a))
#define REVERSE(a) reverse(ALL(a))
#define MP make_pair
#define EB emplace_back
#define FORE(a, b) for (auto &&a : b)
#define FIND(s, n) (s.find(n) != s.end())

using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
const int INF = 1e9;
const int MOD = INF + 7;
const LL LLINF = 1e18;

int dp[1010][2];
//dp[i][j] = i皿目まで来ていてi皿目をとったとき(j = 1),取らなかった時(j = 0)の最大値

int main(void) {
    int N; cin >> N;
    VI V(N + 1);
    REP(i, N) cin >> V[i + 1];
    dp[0][0] = dp[0][1] = 0;
    FOR(i, 1, N + 1) {
        dp[i][0] = max(dp[i - 1][0], dp[i - 1][1]);
        dp[i][1] = max({dp[i - 1][0], dp[i - 2][0], dp[i - 2][1]}) + V[i];
    }
    int ans = max(dp[N][0], dp[N][1]);
    cout << ans << endl;
    int jj;
    VI ansv;
    if (dp[N][0] >= dp[N][1]) {
        jj = 0;
    } else {
        jj = 1;
        ansv.EB(N);
    }
    FORR(i, N + 1, 1) {
        int buf = dp[i][jj];
        if (jj == 0) {
            if (buf == dp[i - 1][0]) {
                jj = 0;
            } else {
                ansv.EB(i - 1);
                jj = 1;
            }
        } else {
            buf -= V[i];
            if (buf == dp[i - 1][0]) {
                jj = 0;
            } else if (i >= 2 && buf == dp[i - 2][0]) {
                jj = 0; i--;
            } else if (i >= 2 && buf == dp[i - 2][1]) {
                ansv.EB(i - 2);
                jj = 1; i--;
            }
        }
    }
    REVERSE(ansv);
    FORE(el, ansv) {
        cout << el << " ";
    }
    cout << endl;
}
0