Codeforces Round #567 (Div. 2)-D. Irrigation

地址：https://codeforces.com/contest/1181/problem/D

思路：思维+树状数组+二分查找

先将城市举办次数以及Q个查询的次数由小到大排序。对于城市举办次数每次从低层次到高层次依次增高，依次只要遍历城市，每次将同一高度的上升的下一高度即可，记录此时的总举办次数Max，同时判断查询是否在Max之内，在则判断是在第几个城市ans，而城市举办次数相同时要按照城市下标由小到大举办，因此可用树状数组+二分查找来找出ans

Code:

#include<iostream>
#include<algorithm>
#include<cstdio>
using namespace std;
typedef long long LL;
typedef pair<LL,int> pr;

const int MAX_N=5e5+5;
int n,m,Q;
int C[MAX_N];
LL d[MAX_N];
pr qi[MAX_N],di[MAX_N];
int res[MAX_N];

int Lowbit(int x){
	return x&(-x);
}

void Update(int id,int x){
	while(id<=m){
		C[id]+=x;
		id+=Lowbit(id);
	}
}

int Query(int id){
	int ans=0;
	while(id>0){
		ans+=C[id];
		id-=Lowbit(id);
	}
	return ans;
}
int main()
{
	scanf("%d%d%d",&n,&m,&Q);
	for(int i=0,x;i<n;++i)
	{
		scanf("%d",&x);
		++d[x];
	}
	for(int i=1;i<=m;++i)
		di[i-1]={d[i],i};
	sort(di,di+m);
	for(int i=0;i<Q;++i)
	{
		scanf("%I64d",&qi[i].first);
		qi[i].second=i;
	}
	sort(qi,qi+Q);
	int ql=0,l=0,r=0;
	LL x1=0,x2,ti=0,Min=n,Max=0,ans;
	while(l<m){
		x1=di[l].first;
		while(r<m&&di[r].first==x1){
			Update(di[r].second,1);
			++r;
		}
		x2=di[r].first;
		Max=Min+(x2-x1)*r;
		while(ql<Q&&qi[ql].first<=Max){
			ans=(qi[ql].first-Min)%r;
			if(!ans)	ans=r;
			int l1=1,r1=m,h1;
			while(l1<=r1){
				h1=(l1+r1)/2;
				if(Query(h1)>=ans)	r1=h1-1;
				else	l1=h1+1;
			}
			res[qi[ql].second]=l1;
			++ql;
		}
		Min=Max;
		l=r;
	}
	while(ql<Q){
		ans=(qi[ql].first-Min)%m;
		if(!ans)	ans=r;
		int l1=1,r1=m,h1;
		while(l1<=r1){
			h1=(l1+r1)/2;
			if(Query(h1)>=ans)	r1=h1-1;
			else	l1=h1+1;
		}
		res[qi[ql].second]=l1;
		++ql;
	}
	for(int i=0;i<Q;++i)
		printf("%d\n",res[i]);
	
	return 0;
}