結果

問題 No.1765 While Shining
ユーザー V_Melville
提出日時 2021-12-02 13:50:52
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 35 ms / 2,000 ms
コード長 494 bytes
コンパイル時間 3,040 ms
コンパイル使用メモリ 245,156 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-07-05 01:52:52
合計ジャッジ時間 4,520 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 0; i < (n); ++i)

using std::cin;
using std::cout;
using std::vector;
using ll = long long;

int main() {
	int n;
	cin >> n;
	
	vector<int> a(n);
	rep(i, n) cin >> a[i];
	
	vector<int> dp(n);
	for (int i = n-2; i >= 0; --i) {
		if (a[i] == 0) dp[i] = 0;
		else if (a[i+1] == 1) dp[i] = 1;
		else {
			if (i == n-2) dp[i] = 1;
			else dp[i] = dp[i+2] + 2;
		}
	}
	
	ll ans = 0;
	rep(i, n) ans += dp[i];
	cout << ans << '\n';
	
	return 0;
}
0