社区讨论

如果你中间一直有 WA 十个测试点

LLezy233
AT_abc348_g[ABC348G] Max (Sum - Max)参与者 4已保存回复 3

讨论操作

快速查看讨论及其快照的属性，并进行相关操作。

当前回复
3 条
当前快照
1 份
快照标识符
@m0ul1fpy
此快照首次捕获于
2024/09/09 13:48
去年
此快照最后确认于
2025/11/04 21:29
4 个月前
查看原帖
具体WA测试点是 15-19，30-34
注意看看你的主席树有没有写炸，有没有考虑当前节点个数超过k的场景，此时应该返回 rtL*k
CPP
ll _rangeQuery(int _l, int _r, ll rtL, ll rtR, int k) { // sum of max k
  if(info[_r].cnt-info[_l].cnt<=k) return info[_r].sum-info[_l].sum;
  int rs_cnt = info[rs(_r)].cnt-info[rs(_l)].cnt;
  ll mid = rtL+rtR;
  if(mid>=0) mid /= 2;
  else mid = (mid-1)/2;
  if(rs_cnt>=k) return _rangeQuery(rs(_l), rs(_r), mid+1, rtR, k);
  return _rangeQuery(ls(_l), ls(_r), rtL, mid, k-rs_cnt)
    +_rangeQuery(rs(_l), rs(_r), mid+1, rtR, rs_cnt);
}
改成
CPP
ll _rangeQuery(int _l, int _r, ll rtL, ll rtR, int k) { // sum of max k
  if(info[_r].cnt-info[_l].cnt<=k) return info[_r].sum-info[_l].sum;
  if(rtL==rtR) return rtL*k; // 多了这句
  int rs_cnt = info[rs(_r)].cnt-info[rs(_l)].cnt;
  ll mid = rtL+rtR;
  if(mid>=0) mid /= 2;
  else mid = (mid-1)/2;
  if(rs_cnt>=k) return _rangeQuery(rs(_l), rs(_r), mid+1, rtR, k);
  return _rangeQuery(ls(_l), ls(_r), rtL, mid, k-rs_cnt)
    +_rangeQuery(rs(_l), rs(_r), mid+1, rtR, rs_cnt);
}

回复

共 3 条回复，欢迎继续交流。

正在加载回复...