結果

問題 No.258 回転寿司(2)
コンテスト
ユーザー tkmst201
提出日時 2017-05-17 20:23:45
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 1,518 bytes
コンパイル時間 1,881 ms
コンパイル使用メモリ 164,376 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-06 19:11:57
合計ジャッジ時間 5,328 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 67
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:29:24: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   29 |         REP(i, n) scanf("%d", v + i);
      |                   ~~~~~^~~~~~~~~~~~~

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;

#define FOR(i,a,b) for(int i=(a);i<(b);i++)
#define REP(i,n) FOR(i,0,n)
#define ALL(v) (v).begin(),(v).end()

template<typename A, typename B> inline bool chmax(A &a, B b) { if (a<b) { a=b; return 1; } return 0; }
template<typename A, typename B> inline bool chmin(A &a, B b) { if (a>b) { a=b; return 1; } return 0; }

typedef unsigned long long ull;
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef pair<int, pii> P;

const ll INF = 1ll<<29;
const ll MOD = 1000000007;
const double EPS  = 1e-10;

int n;
int v[1000];
int dp[1001][2];
int prevv[1001][2],  prevs[1001][2];

int main() {
	cin >> n;
	REP(i, n) scanf("%d", v + i);
	
	memset(prevv, -1, sizeof(prevv));
	memset(prevs, -1, sizeof(prevs));
	
	REP(i, n) {
		if (chmax(dp[i + 1][1], dp[i][0] + v[i])) {
			prevv[i + 1][1] = i;
			prevs[i + 1][1] = i;
		}
		if (chmax(dp[i + 1][0], dp[i][1])) {
			prevv[i + 1][0] = i + 5000;
			prevs[i + 1][0] = -1;
		}
		if (chmax(dp[i + 1][0], dp[i][0])) {
			prevv[i + 1][0] = i;
			prevs[i + 1][0] = i;
		}
	}
	
	
	vector<int> ans;
	
	int pos = n;
	if (dp[n][0] < dp[n][1]) pos = n + 5000;
	
	while (true) {
		int f = 0;
		if (pos >= 5000) {
			pos -= 5000;
			f = 1;
		}
		
		if (prevv[pos][f] == -1) break;
		
		if (f) ans.push_back(prevs[pos][f]);
		
		pos = prevv[pos][f];
	}
	
	sort(ALL(ans));
	
	cout << max(dp[n][0], dp[n][1]) << endl;
	REP(i, ans.size()) printf("%d%c", ans[i] + 1, i == ans.size() - 1 ? '\n' : ' ');
	
	return 0;
}
0