P10550 [THUPC2024] 貿易 題解

正式場上拿了這題的首 \(A\)，讓隊伍不至於無獎而返。

思路

容易發現題目的買入賣出過程形似一個括號匹配。

那麼我們可以對每一種型別的物品做括號匹配。

若是一個匹配的括號在詢問區間內則可以記入答案。

就變成了一個二維數點問題。

離線樹狀樹組即可。

Code

#include <bits/stdc++.h>
using namespace std;

#define fro(i, x, y) for (int i = (x); i <= (y); i++)
#define pre(i, x, y) for (int i = (x); i >= (y); i--)

const int N = 5e5 + 10;

int n, q, a[N], c[N], p[N], t[N], l[N], r[N], ans[N];
vector<int> d[N];
vector<int> o[N];

inline void upd(int x) { while (x) t[x]++, x -= (x & (-x)); }
inline auto ask(int x) { int r = 0; while (x <= n) r += t[x], x += (x & (-x)); return r; }

signed main() {
  ios::sync_with_stdio(0), cin.tie(0);
  cin >> n >> q;
  fro(i, 1, n) cin >> a[i];
  fro(i, 1, n) cin >> c[i];
  fro(i, 1, n) {
    if (a[i] == 0) d[c[i]].push_back(i);
    if (a[i] == 1) if (!d[c[i]].empty()) p[i] = d[c[i]].back(), d[c[i]].pop_back();
  }
  fro(i, 1, q) cin >> l[i] >> r[i], o[r[i]].push_back(i);
  fro(i, 1, n) {
    if (p[i]) upd(p[i]);
    for (auto j : o[i]) {
      ans[j] = ask(l[j]);
    }
  }
  fro(i, 1, q) cout << ans[i] << "\n";
  return 0;
}