結果
問題 | No.1505 Zero-Product Ranges |
ユーザー | 👑 emthrm |
提出日時 | 2021-05-14 22:07:05 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 21 ms / 2,000 ms |
コード長 | 1,541 bytes |
コンパイル時間 | 1,882 ms |
コンパイル使用メモリ | 199,128 KB |
最終ジャッジ日時 | 2025-01-21 11:10:56 |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 49 |
ソースコード
#define _USE_MATH_DEFINES #include <bits/stdc++.h> using namespace std; #define FOR(i,m,n) for(int i=(m);i<(n);++i) #define REP(i,n) FOR(i,0,n) #define ALL(v) (v).begin(),(v).end() using ll = long long; constexpr int INF = 0x3f3f3f3f; constexpr long long LINF = 0x3f3f3f3f3f3f3f3fLL; constexpr double EPS = 1e-8; constexpr int MOD = 1000000007; // constexpr int MOD = 998244353; constexpr int dy[] = {1, 0, -1, 0}, dx[] = {0, -1, 0, 1}; constexpr int dy8[] = {1, 1, 0, -1, -1, -1, 0, 1}, dx8[] = {0, -1, -1, -1, 0, 1, 1, 1}; template <typename T, typename U> inline bool chmax(T &a, U b) { return a < b ? (a = b, true) : false; } template <typename T, typename U> inline bool chmin(T &a, U b) { return a > b ? (a = b, true) : false; } struct IOSetup { IOSetup() { std::cin.tie(nullptr); std::ios_base::sync_with_stdio(false); std::cout << fixed << setprecision(20); } } iosetup; template <typename T = char, typename U = std::string> std::vector<std::pair<T, int>> run_length_encoding(const U &s) { int n = s.size(); std::vector<std::pair<T, int>> res; T now = s[0]; int cnt = 1; for (int i = 1; i < n; ++i) { if (s[i] != now) { res.emplace_back(now, cnt); cnt = 0; } now = s[i]; ++cnt; } res.emplace_back(now, cnt); return res; } int main() { int n; cin >> n; vector<int> a(n); REP(i, n) cin >> a[i]; ll ans = n * (n + 1LL) / 2; for (auto [ai, len] : run_length_encoding<int>(a)) { if (ai == 1) ans -= len * (len + 1LL) / 2; } cout << ans << '\n'; return 0; }