結果

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

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
#define rep(i,n) for(int (i)=0;(i)<(int)(n);++(i))
#define each(itr,c) for(__typeof(c.begin()) itr=c.begin(); itr!=c.end(); ++itr)
#define all(x) (x).begin(),(x).end()
#define mp make_pair
#define pb push_back
#define fi first
#define se second

typedef pair<int,int> pi;

int main()
{
    int n;
    cin >>n;
    vector<int> v(n);
    rep(i,n) cin >>v[i];

    int dp[1001][2]={0};
    pi p[1001][2];
    for(int i=1; i<=n; ++i)
    {
        dp[i][0]=dp[i-1][1]+v[i-1];
        p[i][0]=pi(i-1,1);

        if(dp[i-1][1]>=dp[i-1][0])
        {
            dp[i][1]=dp[i-1][1];
            p[i][1]=pi(i-1,1);
        }
        else
        {
            dp[i][1]=dp[i-1][0];
            p[i][1]=pi(i-1,0);
        }
    }

    //rep(i,n+1) printf("%d %d\n", dp[i][0],dp[i][1]);

    int ans=max(dp[n][0],dp[n][1]);
    vector<int> a;

    //経路復元
    pi st;
    if(dp[n][0]>dp[n][1]) st=pi(n,0);
    else st=pi(n,1);

    while(1)
    {
        if(st.fi==0) break;

        pi nst=p[st.fi][st.se];
        if(st.se==0 && nst.se==1) a.pb(st.fi);

        st=nst;
    }

    reverse(all(a));

    //output
    cout << ans << endl;
    rep(i,a.size())
    {
        if(i) printf(" ");
        printf("%d", a[i]);
    }
    printf("\n");

    return 0;
}
0