Codeforces 873E Awards For Contestants ST表

原文链接https://www.cnblogs.com/zhouzhendong/p/9255885.html

题目传送门 - CF873E

题意

　　现在要给 $n(n\leq 3000)$ 个学生颁奖。

　　记 $a_i$ 为第 $i$ 个学生在本次比赛中做出的题目数量。

　　记 $b_i$ 为第 $i$ 个学生所获的奖项，其中 $1,2,3$ 分别表示他获得一、二、三等奖， $-1$ 表示不获奖，当然，一等奖最好，没奖最差。

　　记 $cnt_i$ 为满足 $b_k=i$ 的 $k$ 的个数。

　　一个合法的颁奖方案必须满足以下条件：

　　　　对于任何的 $i,x,y\in {1,2,3}$，

　　　　　　1.　　$cnt_i>0$

　　　　　　2.　　$cnt_x\leq 2cnt_y$

　　　　如果第 $i$ 个人比第 $j$ 个人做题数多，则第 $i$ 个人获得的奖不比第 $j$ 个人差。

　　现在，记 $b_i=j$ 的 $i$ 中，使得 $a_i$ 最大的记为 $c_j$ ， 使得 $a_i$ 最小的记为 $d_j$。

　　请你在所有合法方案中选择 $d_1-c_2$ 最大的；如果有多个，再在其中选择 $d_2-c_3$ 最大的；如果有多个，再在其中选择 $d_3-c_{-1}$ 最大的；如果还有多个，那么选择任意方案。

　　按照输入的人的顺序输出你选择的方案中各个人的奖项。

题解

　　VP 的时候想了个很真的假算法，好久以后才发现的。

　　正确的做法：

　　首先将所有人按照 $a$ 降序排列。

　　枚举 $d_1$ 的位置 $i$

　　　　枚举 $d_2$ 的位置 $i+j$

　　　　　　在满足 $i\leq 2j,j\leq 2i$ 的情况下，继续下面的操作：

　　　　　　令 $d3=i+j+k$ ，则 $k$ 需要满足：对于任何的 $i,x,y\in {1,2,3}$，1.$cnt_i>0$　2.$cnt_x\leq 2cnt_y$。

　　　　　　那么，$k$ 的取值范围是 $\left[\max(\left\lfloor\cfrac{i+1}2\right\rfloor,\left\lfloor\cfrac{j+1}2\right\rfloor,\min(2i,2j,n)\right]$ ，于是我们需要找到一个 $k$ 满足 $a_k-a_{k+1}$ 最大。

　　　　　　由于没有修改，我们可以用 ST表 预处理，$O(1)$ 实现询问。于是我们就可以得到要选择的 $k$ 了。每次通过新的 $i,j,k$ 生成新的 $d_1,d_2,d_3$ 与目前最优解择优录用。

　　最终根据最优解定奖，别忘了最后要排序一遍把所有的人排回输入顺序。

代码

#include <bits/stdc++.h>
using namespace std;
const int N=3005;
int n,ST[N][14];
struct pb{
	int a,id,res;
}a[N];
bool cmpa(pb a,pb b){
	return a.a>b.a;
}
bool cmpid(pb a,pb b){
	return a.id<b.id;
}
int calc(int i){
	return a[i].a-a[i+1].a;
}
int Query(int L,int R){
	if (L>R)
		return -1;
	int d=log(R-L+1)/log(2.0);
	int a=ST[L+(1<<d)-1][d],b=ST[R][d];
	return calc(a)>calc(b)?a:b;
}
int main(){
	scanf("%d",&n);
	for (int i=1;i<=n;i++)
		scanf("%d",&a[i]),a[i].id=i;
	sort(a+1,a+n+1,cmpa);
	a[n+1].a=0;
	for (int i=1;i<=n;i++){
		ST[i][0]=i;
		for (int j=1;j<=12;j++){
			ST[i][j]=ST[i][j-1];
			if (i-(1<<(j-1))>0&&calc(ST[i-(1<<(j-1))][j-1])>calc(ST[i][j]))
				ST[i][j]=ST[i-(1<<(j-1))][j-1];
		}
	}
	int d1=-1,d2=-1,d3=-1;
	for (int i=1;i<=n;i++)
		for (int j=1;i+j<=n;j++){
			if (i>2*j||j>2*i)
				continue;
			int dd1=i,dd2=i+j,dd3=Query(dd2+max((i+1)/2,(j+1)/2),min(dd2+min(i*2,j*2),n));
			int k=dd3-dd2;
			if (dd3==-1||k<1)
				continue;
			if (d1==-1||calc(dd1)>calc(d1)||(calc(dd1)==calc(d1)&&(calc(dd2)>calc(d2)||(calc(dd2)==calc(d2)&&calc(dd3)>calc(d3)))))
				d1=dd1,d2=dd2,d3=dd3;
		}
	for (int i=1;i<=d1;i++)
		a[i].res=1;
	for (int i=d1+1;i<=d2;i++)
		a[i].res=2;
	for (int i=d2+1;i<=d3;i++)
		a[i].res=3;
	for (int i=d3+1;i<=n;i++)
		a[i].res=-1;
	sort(a+1,a+n+1,cmpid);
	for (int i=1;i<=n;i++)
		printf("%d ",a[i].res);
	return 0;
}