BZOJ4709 [Jsoi2011]柠檬

84 篇文章 1 订阅
11 篇文章 0 订阅

考虑DP,f[i]表示前i个的最大价值

我们发现对于任意一个i,他所选的最后一段所指定的颜色一定是i的颜色,因为否则的话i这个点就会没有贡献,一定不如最后一段只选一个i

那么在最后一段所选的颜色一定是i的情况下,最后一段的开头的颜色也一定是i的颜色,否则开头一段也没有贡献,不然单分出去一段

那么对于f[i],假设之前存在一个点j,这个点的颜色a[j]与i的颜色a[i]相等,那么我们可以让f[i]由j转移,价值为f[j-1]+a[j]*(s[i]-s[j]+1)^2,s[i]代表前i个数里a[i]出现的次数

我们发现对于任意一个颜色x,随着这个颜色的数不断出现,由每一个之前的a[j]=x的j转移所产生的价值都会不断增加,且j越小,增加的越块,如果有j1<j2,并且j1当前的价值比j2大了,那么j2就没有用了,于是我们可以维护一个单调栈,每次如果栈顶第二个的元素超过了栈顶,就把栈顶弹出,决策的时候直接用栈顶决策

而我们发现可能会出现栈顶第三个元素超过了栈顶,而栈顶第二个元素还没超过栈顶的情况,我们需要避免这种情况。我们发现对于任意的j1<j2<i1<i2,如果j1超过i1的时间比j2超过i1的时间要早,那么j1超过i2的时间也一定比j2超过i2的时间早,证明显然,并且我们对于任意的(j1,j2),j1<j2,我们都能直接算出在第几个颜色与j1,j2相等的数出现的时候,j1会超过j2,那么在我们即将把i压入栈顶的时候,只要当前栈顶第二个元素超过i的时间比当前栈顶超过i的时间要早,那么栈顶就也没用了,我们就可以弹栈,这样的话栈里的每一个元素超过上一个元素的时间也是单调的,于是就没有问题了

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<ctime>
#include<cmath>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<map>
#include<set>
#include<bitset>
#include<queue>
#include<stack>
using namespace std;
#define MAXN 100010
#define MAXM 10010
#define INF 1000000000
#define MOD 1000000007
#define eps 1e-8
#define ll long long
ll f[MAXN];
int s[MAXN];
int c[MAXM];
int a[MAXN];
vector<int>st[MAXM];
int n;
ll cal(int x,int y){
	return f[x-1]+(ll)a[x]*y*y;
}
int beyond(int x,int y){
	int l=1,r=n;
	int re=n+1;
	while(l<=r){
		int mid=l+r>>1;
		if(cal(x,mid-s[x]+1)>=cal(y,mid-s[y]+1)){
			re=mid;
			r=mid-1;
		}else{
			l=mid+1;
		}
	}
	return re;
}
int main(){
	int i,x;
	scanf("%d",&n);
	for(i=1;i<=n;i++){
		scanf("%d",&x);
		a[i]=x;
		s[i]=++c[x];
		while(st[x].size()>=2&&
				beyond(st[x][st[x].size()-2],st[x][st[x].size()-1])<=
				beyond(st[x][st[x].size()-1],i)
		){
			st[x].pop_back();
		}
		st[x].push_back(i);
		while(st[x].size()>=2&&beyond(st[x][st[x].size()-2],st[x][st[x].size()-1])<=s[i]){
			st[x].pop_back();
		}
		f[i]=cal(st[x][st[x].size()-1],s[i]-s[st[x][st[x].size()-1]]+1);
	}
	printf("%lld\n",f[n]);
	return 0;
}

/*
5
1 3 2 3 1 
*/


评论 9
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值