結果
問題 | No.404 部分門松列 |
ユーザー |
![]() |
提出日時 | 2022-07-13 09:13:57 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 1,712 ms / 2,000 ms |
コード長 | 16,933 bytes |
コンパイル時間 | 3,687 ms |
コンパイル使用メモリ | 226,772 KB |
実行使用メモリ | 45,840 KB |
最終ジャッジ日時 | 2024-06-24 08:13:29 |
合計ジャッジ時間 | 24,871 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 31 |
ソースコード
#include <bits/stdc++.h>using namespace std;// #pragma GCC target("arch=skylake-avx512")// #include <atcoder/all>// using namespace atcoder;// #define NDEBUG// #define _GLIBCXX_DEBUG#pragma region template// Defineusing ll = long long;using ull = unsigned long long;using ld = long double;template <class T> using pvector = vector<pair<T, T>>;template <class T> using rpriority_queue = priority_queue<T, vector<T>, greater<T>>;constexpr const ll dx[4] = {1, 0, -1, 0};constexpr const ll dy[4] = {0, 1, 0, -1};constexpr const ll MOD = 1e9 + 7;constexpr const ll mod = 998244353;constexpr const ll INF = 1LL << 60;constexpr const ll inf = 1 << 30;constexpr const char rt = '\n';constexpr const char sp = ' ';#define rt(i, n) (i == (ll) (n) -1 ? rt : sp)#define len(x) ((ll) (x).size())#define all(x) (x).begin(), (x).end()#define rall(x) (x).rbegin(), (x).rend()#define mp make_pair#define mt make_tuple#define pb push_back#define eb emplace_back#define ifn(x) if (not(x))#define elif else if#define elifn else ifn#define fi first#define se second#define uniq(x) (sort(all(x)), (x).erase(unique(all(x)), (x).end()))#define bis(x, y) ((ll) (lower_bound(all(x), y) - (x).begin()))using graph = vector<vector<ll>>;template <class T> using wgraph = vector<vector<pair<ll, T>>>;bool __DIRECTED__ = true;bool __ZERO_INDEXED__ = false;istream &operator>>(istream &is, graph &g) {ll a, b;is >> a >> b;if (__ZERO_INDEXED__ == false) a--, b--;g[a].pb(b);if (__DIRECTED__ == false) g[b].pb(a);return is;}template <class T> istream &operator>>(istream &is, wgraph<T> &g) {ll a, b;T c;is >> a >> b >> c;if (__ZERO_INDEXED__ == false) a--, b--;g[a].pb({b, c});if (__DIRECTED__ == false) g[b].pb({a, c});return is;}template <class T> bool chmax(T &a, const T &b) {if (a < b) {a = b;return 1;}return 0;}template <class T> bool chmin(T &a, const T &b) {if (a > b) {a = b;return 1;}return 0;}// Debug#ifdef NDEBUG#define debug(...)#define dumpi(a, h, w)#define vdumpi(a, n)#define dump(a, h, w)#define vdump(a, n)#else#define debug(...) \{ \cerr << __LINE__ << ": " << #__VA_ARGS__ << " = "; \for (auto &&__i : {__VA_ARGS__}) cerr << "[" << __i << "] "; \cerr << rt; \}#define dumpi(a, h, w) \{ \cerr << __LINE__ << ": " << #a << " = [" << rt; \rep(__i, h) { \if (__i) cerr << ",\n"; \cerr << "["; \rep(__j, w) { \if (__j) cerr << ", "; \if (abs(a[__i][__j]) >= INF / 2 and a[__i][__j] <= -INF / 2) cerr << '-'; \if (abs(a[__i][__j]) >= INF / 2) cerr << "∞"; \else \cerr << a[__i][__j]; \} \cerr << "]"; \} \cerr << "\n]" << rt; \}#define vdumpi(a, n) \{ \cerr << __LINE__ << ": " << #a << " = ["; \rep(__i, n) { \if (__i) cerr << ", "; \if (abs(a[__i]) >= INF / 2 and a[__i] <= -INF / 2) cerr << '-'; \if (abs(a[__i]) >= INF / 2) cerr << "∞"; \else \cerr << a[__i]; \} \cerr << "]" << rt; \}#define dump(a, h, w) \{ \cerr << __LINE__ << ": " << #a << " = [" << rt; \rep(__i, h) { \if (__i) cerr << ",\n"; \cerr << "["; \rep(__j, w) { \if (__j) cerr << ", "; \cerr << a[__i][__j]; \} \cerr << "]"; \} \cerr << "\n]" << rt; \}#define vdump(a, n) \{ \cerr << __LINE__ << ": " << #a << " = ["; \rep(__i, n) { \if (__i) cerr << ", "; \cerr << a[__i]; \} \cerr << "]" << rt; \}#endiftemplate <class S, class T> istream &operator>>(istream &is, pair<S, T> &p) {is >> p.first >> p.second;return is;}template <class S, class T> ostream &operator<<(ostream &os, const pair<S, T> &p) {os << p.first << ' ' << p.second;return os;}// Loop#define inc(i, a, n) for (ll i = (a), _##i = (n); i <= _##i; ++i)#define dec(i, a, n) for (ll i = (a), _##i = (n); i >= _##i; --i)#define rep(i, n) for (ll i = 0, _##i = (n); i < _##i; ++i)#define each(i, a) for (auto &&i : a)// Stream#define fout(n) cout << fixed << setprecision(n)struct io {io() { cin.tie(nullptr), ios::sync_with_stdio(false); }} io;// Speed#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,avx2,tune=native")#pragma GCC optimize("Ofast,unroll-loops")// Mathinline constexpr ll gcd(const ll a, const ll b) { return b ? gcd(b, a % b) : a; }inline constexpr ll lcm(const ll a, const ll b) { return a / gcd(a, b) * b; }inline constexpr ll modulo(const ll n, const ll m = MOD) {ll k = n % m;return k + m * (k < 0);}inline constexpr ll chmod(ll &n, const ll m = MOD) {n %= m;return n += m * (n < 0);}inline constexpr ll mpow(ll a, ll n, const ll m = MOD) {ll r = 1;rep(i, 64) {if (n & (1LL << i)) r *= a;chmod(r, m);a *= a;chmod(a, m);}return r;}inline ll inv(const ll n, const ll m = MOD) {ll a = n, b = m, x = 1, y = 0;while (b) {ll t = a / b;a -= t * b;swap(a, b);x -= t * y;swap(x, y);}return modulo(x, m);}unsigned long long binary_gcd(unsigned long long x, unsigned long long y) {if (!x | !y) return x | y;unsigned long long cx = __builtin_ctzll(x), cy = __builtin_ctzll(y);x >>= cx, y >>= cy;while (x ^ y) {if (x > y) {x = (x - y) >> __builtin_ctzll(x ^ y);} else {y = (y - x) >> __builtin_ctzll(x ^ y);}}return x << min(cx, cy);}inline long long binary_gcd(long long x, long long y) {return binary_gcd((unsigned long long) (abs(x)), (unsigned long long) (abs(y)));}#define codeforces \ll testcases; \cin >> testcases; \rep(testcase, testcases)#define gcj(s) cout << s << testcase + 1 << ": "#pragma endregion/*** @brief Succinct Indexable Dictionary(完備辞書)*/struct SuccinctIndexableDictionary {size_t length;size_t blocks;vector<unsigned> bit, sum;SuccinctIndexableDictionary() = default;SuccinctIndexableDictionary(size_t length) : length(length), blocks((length + 31) >> 5) {bit.assign(blocks, 0U);sum.assign(blocks, 0U);}void set(int k) { bit[k >> 5] |= 1U << (k & 31); }void build() {sum[0] = 0U;for (int i = 1; i < blocks; i++) {sum[i] = sum[i - 1] + __builtin_popcount(bit[i - 1]);}}bool operator[](int k) { return (bool((bit[k >> 5] >> (k & 31)) & 1)); }int rank(int k) {return (sum[k >> 5] + __builtin_popcount(bit[k >> 5] & ((1U << (k & 31)) - 1)));}int rank(bool val, int k) { return (val ? rank(k) : k - rank(k)); }};template <typename T, int MAXLOG> struct WaveletTree {struct Node {SuccinctIndexableDictionary sid;Node *ch[2];Node() = default;Node(size_t length) : sid(length + 1), ch{nullptr} {}};Node *root;Node *build(vector<T> &v, vector<T> &rbuff, int bit, int l, int r) {if (l >= r || bit == -1) return nullptr;Node *node = new Node(r - l);int left = 0, right = 0;for (int k = l; k < r; k++) {if (((v[k] >> bit) & 1)) {rbuff[right++] = v[k];node->sid.set(k - l);} else {v[l + left++] = v[k];}}for (int k = 0; k < right; k++) {v[l + left + k] = rbuff[k];}node->sid.build();node->ch[0] = build(v, rbuff, bit - 1, l, l + left);node->ch[1] = build(v, rbuff, bit - 1, l + left, r);return node;}WaveletTree() = default;WaveletTree(vector<T> v) {vector<T> rbuff(v.size());root = build(v, rbuff, MAXLOG - 1, 0, v.size());}int rank(Node *t, int l, int r, const T &x, int level) {if (l >= r || t == nullptr) return 0;if (level == -1) return r - l;bool f = (x >> level) & 1;l = t->sid.rank(f, l), r = t->sid.rank(f, r);return rank(t->ch[f], l, r, x, level - 1);}int rank(const T &x, int r) { return rank(root, 0, r, x, MAXLOG - 1); }T kth_smallest(Node *t, int l, int r, int k, int level) {if (l >= r || t == nullptr) return 0;int cnt = t->sid.rank(false, r) - t->sid.rank(false, l);bool f = cnt <= k;l = t->sid.rank(f, l), r = t->sid.rank(f, r);if (f) return kth_smallest(t->ch[f], l, r, k - cnt, level - 1) | ((T(1) << level));return kth_smallest(t->ch[f], l, r, k, level - 1);}// k-th(0-indexed) smallest number in v[l,r)T kth_smallest(int l, int r, int k) {assert(0 <= k && k < r - l);return kth_smallest(root, l, r, k, MAXLOG - 1);}// k-th(0-indexed) largest number in v[l,r)T kth_largest(int l, int r, int k) { return kth_smallest(l, r, r - l - k - 1); }int range_freq(Node *t, int l, int r, T upper, int level) {if (t == nullptr || l >= r) return 0;bool f = ((upper >> level) & 1);int ret = 0;if (f) ret += t->sid.rank(false, r) - t->sid.rank(false, l);l = t->sid.rank(f, l), r = t->sid.rank(f, r);return range_freq(t->ch[f], l, r, upper, level - 1) + ret;}// count i s.t. (l <= i < r) && (v[i] < upper)int range_freq(int l, int r, T upper) { return range_freq(root, l, r, upper, MAXLOG - 1); }// count i s.t. (l <= i < r) && (lower <= v[i] < upper)int range_freq(int l, int r, T lower, T upper) {return range_freq(l, r, upper) - range_freq(l, r, lower);}// max v[i] s.t. (l <= i < r) && (v[i] < upper)T prev_value(int l, int r, T upper) {int cnt = range_freq(l, r, upper);return cnt == 0 ? T(-1) : kth_smallest(l, r, cnt - 1);}// min v[i] s.t. (l <= i < r) && (lower <= v[i])T next_value(int l, int r, T lower) {int cnt = range_freq(l, r, lower);return cnt == r - l ? T(-1) : kth_smallest(l, r, cnt);}};template <typename T, int MAXLOG> struct CompressedWaveletTree {WaveletTree<int, MAXLOG> mat;vector<T> ys;CompressedWaveletTree(const vector<T> &v) : ys(v) {sort(begin(ys), end(ys));ys.erase(unique(begin(ys), end(ys)), end(ys));vector<int> t(v.size());for (int i = 0; i < v.size(); i++) t[i] = get(v[i]);mat = WaveletTree<int, MAXLOG>(t);}inline int get(const T &x) { return lower_bound(begin(ys), end(ys), x) - begin(ys); }int rank(const T &x, int r) {auto pos = get(x);if (pos == ys.size() || ys[pos] != x) return 0;return mat.rank(pos, r);}T kth_smallest(int l, int r, int k) { return ys[mat.kth_smallest(l, r, k)]; }T kth_largest(int l, int r, int k) { return ys[mat.kth_largest(l, r, k)]; }int range_freq(int l, int r, T upper) { return mat.range_freq(l, r, get(upper)); }int range_freq(int l, int r, T lower, T upper) {return mat.range_freq(l, r, get(lower), get(upper));}T prev_value(int l, int r, T upper) {auto ret = mat.prev_value(l, r, get(upper));return ret == -1 ? T(-1) : ys[ret];}T next_value(int l, int r, T lower) {auto ret = mat.next_value(l, r, get(lower));return ret == -1 ? T(-1) : ys[ret];}};signed main() {ll n;cin >> n;vector<ll> a(n);rep(i, n) cin >> a[i];CompressedWaveletTree<ll, 20> cwt(a);map<ll, ll> res;rep(i, n) {ll c1 = cwt.range_freq(0, i, 0, a[i]);ll c2 = cwt.range_freq(i + 1, n, 0, a[i]);ll c3 = cwt.range_freq(0, i, a[i] + 1, INF);ll c4 = cwt.range_freq(i + 1, n, a[i] + 1, INF);ll c5 = cwt.range_freq(0, i, a[i], a[i] + 1);ll c6 = cwt.range_freq(i + 1, n, a[i], a[i] + 1);res[a[i]] += c1 * c2 + c3 * c4 + c5 * c6;}ll sum = 0;rep(i, n) {sum -= cwt.range_freq(0, i - 1, a[i], a[i] + 1);if (i != 0) sum += cwt.range_freq(i + 1, n, a[i - 1], a[i - 1] + 1);res[a[i]] -= sum;}vector<pair<ll, ll>> v = {{0, 0}};each(it, res) v.push_back(it);sort(begin(v), end(v));rep(i, v.size() - 1) v[i + 1].second += v[i].second;ll q;cin >> q;rep(i, q) {ll l, h;cin >> l >> h;ll hi = upper_bound(begin(v), end(v), make_pair(h, INF)) - begin(v) - 1;if (hi == -1) hi = 0;ll lo = upper_bound(begin(v), end(v), make_pair(l - 1, INF)) - begin(v) - 1;if (lo == -1) lo = 0;cout << v[hi].second - v[lo].second << rt;}}