2018.09.16 codeforces1041C. Coffee Break(双端队列)

传送门
真心sb题啊。
考场上最开始看成了一道写过的原题。。。
仔细想了一会发现看错了。
其实就是一个sb队列。
每次插入到队首去就行了。
代码:

#include
#define N 200005
using namespace std;
inline int read(){
	int ans=0;
	char ch=getchar();
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();
	return ans;
}
int n,m,d,q[N],ans[N],tot=0,hd,tl;
struct Node{int v,id;}a[N];
inline bool cmp(Node a,Node b){return a.v<b.v;}
int main(){
	n=read(),m=read(),d=read();
	for(int i=1;i<=n;++i)a[i].v=read(),a[i].id=i;
	sort(a+1,a+n+1,cmp);
	hd=1,tl=0;
	q[++tl]=1,ans[a[1].id]=++tot;
	for(int i=2;i<=n;++i){
		if(a[i].v-d-1>=a[q[hd]].v)ans[a[i].id]=ans[a[q[hd]].id],++hd;
		else ans[a[i].id]=++tot;
		q[++tl]=i;
	}
	cout<<tot<<'\n';
	for(int i=1;i<=n;++i)cout<<ans[i]<<' ';
	return 0;
}

你可能感兴趣的:(#,队列,-----数据结构-----,数据结构与分治算法)