H - Hello Ms. Ze(树状数组套主席树,线段树上二分)

这篇博客探讨了一种复杂的数据结构问题,涉及树状数组和主席树的结合使用。作者首先介绍了问题背景:有多种材料,每种材料有一定数量,需要进行两种操作——修改材料数量和计算在特定区间内能制作的最大衣物数。在不考虑修改操作的情况下,通过二分法可以求解区间内的最大衣物数。接着,引入了修改操作,并采用树状数组套主席树的结构,动态维护区间信息,实现了在线求解。最后,提供了完整的C++代码实现。
摘要由CSDN通过智能技术生成

H - Hello Ms. Ze

给定 n n n种不同的材料,第 i i i种材料有 a i a_i ai个,有 m m m个操作,操作分为两类:

  • 把第 x x x种材料修改为 y y y个,
  • 只用 [ l , r ] [l, r] [l,r]区间的材料制作衣服,每件衣服要用 k k k个不同的材料,最多能做多少件,

先不考虑修改操作,对于区间 [ l , r ] [l, r] [l,r]中的原料,我们可以做最多多少衣服,如何求解,

考虑二分答案,假设我们可以最多做 x x x件衣服,显然对于个数大于等于 x x x的材料,在每件衣服中我们只能使用一次,

假设我们当前二分的区间为 [ l , r ] [l, r] [l,r],设材料数量大于 m i d mid mid的有 a a a种,材料数量小于等于 m i d mid mid的数量和为 s u m sum sum

a + s u m m i d + 1 ≥ k a + \frac{sum}{mid + 1} \geq k a+mid+1sumk,说明答案在右区间,否则答案在左区间,其实这整个过程我们可以在线段树上直接二分执行。

考虑带修,树状数组套主席树,然后主席树上二分即可,由于答案值域在 [ 1 , 1 0 5 × 1 0 6 ] [1, 10 ^ 5 \times 10 ^ 6] [1,105×106],所以我直接动态开点建了一颗值域在 [ 1 , 1 < < 37 ] [1, 1 << 37] [1,1<<37]的主席树。

树套树

#include <bits/stdc++.h>

using namespace std;

const int N = 1e5 + 10;

const long long maxn = 1ll << 37;

int a[N], n, m;

int root[N], ls[N << 9], rs[N << 9], tot[N << 9], num;

long long sum[N << 9];

void update(int &rt, long long l, long long r, int x, int v) {
  if (!rt) {
    rt = ++num;
  }
  sum[rt] += x * v, tot[rt] += v;
  if (l == r) {
    return ;
  }
  long long mid = l + r >> 1;
  if (x <= mid) {
    update(ls[rt], l, mid, x, v);
  }
  else {
    update(rs[rt], mid + 1, r, x, v);
  }
}

inline int lowbit(int x) {
  return x & (-x);
}

void update(int pos, int x, int v) {
  while (pos <= n) {
    update(root[pos], 1, maxn, x, v);
    pos += lowbit(pos);
  }
}

int A[50], B[50], cnt1, cnt2;

long long query(long long l, long long r, long long S, int cnt, int k) {
  if (l == r) {
    return l;
  }
  long long mid = l + r >> 1;
  long long ans = 0, res = 0;
  for (int i = 1; i <= cnt1; i++) {
    ans -= sum[ls[A[i]]];
    res -= tot[ls[A[i]]];
  }
  for (int i = 1; i <= cnt2; i++) {
    ans += sum[ls[B[i]]];
    res += tot[ls[B[i]]];
  }
  if ((S + ans) / (mid + 1) + cnt - res >= k) {
    S += ans, cnt -= res;
    for (int i = 1; i <= cnt1; i++) {
      A[i] = rs[A[i]];
    }
    for (int i = 1; i <= cnt2; i++) {
      B[i] = rs[B[i]];
    }
    return query(mid + 1, r, S, cnt, k);
  }
  else {
    for (int i = 1; i <= cnt1; i++) {
      A[i] = ls[A[i]];
    }
    for (int i = 1; i <= cnt2; i++) {
      B[i] = ls[B[i]];
    }
    return query(l, mid, S, cnt, k);
  }
}

int main() {
  // freopen("in.txt", "r", stdin);
  // freopen("out.txt", "w", stdout);
  scanf("%d %d", &n, &m);
  for (int i = 1; i <= n; i++) {
    scanf("%d", &a[i]);
    update(i, a[i], 1);
  }
  for (int i = 1, op, l, r, k; i <= m; i++) {
    scanf("%d %d %d", &op, &l, &r);
    if (op == 2) {
      update(l, a[l], -1);
      a[l] = r;
      update(l, a[l], 1);
    }
    else {
      scanf("%d", &k);
      cnt1 = cnt2 = 0;
      for (int j = l - 1; j; j -= lowbit(j)) {
        A[++cnt1] = root[j];
      }
      for (int j = r; j; j -= lowbit(j)) {
        B[++cnt2] = root[j];
      }
      printf("%lld\n", query(1, maxn, 0, r - l + 1, k));
    }
  }
  return 0;
}
  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值