Codeforces 1181D Irrigation 离线+权值线段树求第k大

传送门

思路: 首先根据举办次数从小到大排序,然后对询问进行离线(按照ki的权值从小到大排序), 对于每个k询问,我们去看是否满足cnt[i+1].num*i-sum[i]<=k, (sum[i]代表1到i的举办总次数,cnt[i+1].num*i代表从1到i这i个地点举办次数都达到cnt[i+1].num而举办的总场数), 若<=k则将k%i,然后在权值线段树中查询第k大的数字(注意k为0则将k设为i), 若>k则将cnt[i+1].id更新到权值线段树中,并更新sum.

 

#include <algorithm>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <string>
#include <queue>
#include <cmath>
#include <stack>
#include <time.>
#include <map>
#include <set>
#define mem(a,x) memset(a,x,sizeof(a))
#define gi(x) scanf("%d",&x)
#define gi2(x,y) scanf("%d%d",&x,&y)
#define gi3(x,y,z) scanf("%d%d%d",&x,&y,&z)
#define gll(x) scanf("%lld",&x)
#defin