結果

問題 No.1505 Zero-Product Ranges
ユーザー manjuuu_onsen
提出日時 2021-05-14 22:07:20
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 31 ms / 2,000 ms
コード長 631 bytes
コンパイル時間 1,617 ms
コンパイル使用メモリ 172,712 KB
実行使用メモリ 6,124 KB
最終ジャッジ日時 2024-10-02 01:20:37
合計ジャッジ時間 3,444 ms
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 49
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function 'int main()':
main.cpp:31:18: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17' [-Wc++17-extensions]
   31 |         for(auto [u, v] : v) {
      |                  ^

ソースコード

diff #

#include<bits/stdc++.h>

using namespace std;
using ll = long long;
using pint = pair<int,int>;

template<class T> 
vector<pair<T, int>> RLE(vector<T> vec) {
	vector<pair<T, int>> ret = {{vec.front(), 1}};

	for(int i = 1; i < (int)vec.size(); i++) {
		if(vec[i - 1] == vec[i]) ret.back().second++;
		else {
			ret.push_back({vec[i], 1});
		}
	}
	return ret;
}

int main()
{
	int N; cin >> N;
	vector<int> A(N);
	for(auto &i : A) {
		cin >> i;
		i = (i == 0 ? 0 : 1);
	}

	auto v = RLE(A);
	ll sum  =0;
	for(auto [u, v] : v) {
		if(u == 0)continue;
		sum += (ll)v * (v + 1) / 2;
	}

	cout << (ll)(N) * (N + 1) / 2 - sum << endl;

}
0