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