如果你中间一直有 WA 十个测试点
查看原帖
如果你中间一直有 WA 十个测试点
851600
Lezy233楼主2024/9/9 13:47

具体WA测试点是 15-19,30-34

注意看看你的主席树有没有写炸,有没有考虑当前节点个数超过k的场景,此时应该返回 rtL*k

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

改成

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);
}
2024/9/9 13:47
加载中...