結果
問題 | No.1505 Zero-Product Ranges |
ユーザー |
|
提出日時 | 2023-02-28 11:15:13 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 26 ms / 2,000 ms |
コード長 | 3,530 bytes |
コンパイル時間 | 2,685 ms |
コンパイル使用メモリ | 250,496 KB |
最終ジャッジ日時 | 2025-02-11 00:07:38 |
ジャッジサーバーID (参考情報) |
judge2 / judge6 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 49 |
ソースコード
#pragma region Macros// #pragma GCC target("avx,avx2,fma")// #pragma GCC optimize("O3")// #pragma GCC optimize("unroll-loops")// #pragma GCC target("avx,avx2,fma,sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,tune=native")#include <bits/extc++.h>// #include <bits/stdc++.h>using namespace std;using namespace __gnu_pbds;// using namespace __gnu_cxx;// #include <atcoder/fenwicktree>// #include <atcoder/segtree>// #include <atcoder/maxflow>// using namespace atcoder;// #include <boost/multiprecision/cpp_int.hpp>// namespace mp = boost::multiprecision;// using Bint = mp::cpp_int;#define TO_STRING(var) # var#define pb emplace_back#define int ll#define endl '\n'using ll = long long;using ld = long double;const ld PI = acos(-1);const ld EPS = 1e-10;const ll INFL = 1LL << 61;const int MOD = 998244353;// const int MOD = 1000000007;__attribute__((constructor))void constructor() {ios::sync_with_stdio(false);cin.tie(nullptr);cout << fixed << setprecision(15);}template<int mod> class modint{public:int val = 0;modint(int x = 0) { while (x < 0) x += mod; val = x % mod; }modint(const modint &r) { val = r.val; } // コピーコンストラクタmodint operator -(){ return modint(-val); } // 単項modint operator +(const modint &r) { return modint(*this) += r; }modint operator -(const modint &r) { return modint(*this) -= r; }modint operator *(const modint &r) { return modint(*this) *= r; }modint operator /(const modint &r) { return modint(*this) /= r; }modint &operator +=(const modint &r) {val += r.val;if (val >= mod) val -= mod;return *this;}modint &operator -=(const modint &r) {if (val < r.val) val += mod;val -= r.val;return *this;}modint &operator *=(const modint &r) {val = val * r.val % mod;return *this;}modint &operator /=(const modint &r) {int a = r.val, b = mod, u = 1, v = 0;while (b) {int t = a / b;a -= t * b; swap(a, b);u -= t * v; swap(u, v);}val = val * u % mod;if (val < 0) val += mod;return *this;}bool operator ==(const modint& r) { return this -> val == r.val; }bool operator <(const modint& r) { return this -> val < r.val; }bool operator !=(const modint& r) { return this -> val != r.val; }};using mint = modint<MOD>;istream &operator >>(istream &is, mint& x) {int t; is >> t;x = t;return (is);}ostream &operator <<(ostream &os, const mint& x) {return os << x.val;}mint modpow(const mint &a, int n) {if (n == 0) return 1;mint t = modpow(a, n / 2);t = t * t;if (n & 1) t = t * a;return t;}int modpow(int x, int N, int mod) {int ret = 1;while (N > 0) {if (N % 2 == 1) ret = ret * x % mod;x = x * x % mod;N /= 2;}return ret;}int ceil(int x, int y) { return (x > 0 ? (x + y - 1) / y : x / y); }#pragma endregionsigned main() {int N;cin >> N;vector<int> A(N);vector<int> zero;for (int i = 0; i < N; i++) {cin >> A[i];if (!A[i]) zero.pb(i);}int ans = 0;for (int i = 0; i < N; i++) {if (A[i] == 0) {ans += N - i;} else {auto k = lower_bound(zero.begin(),zero.end(), i);if (k == zero.end()) continue;ans += N - *k;}}cout << ans << endl;}