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;
}