[BZOJ4907]柠檬

时间：2022-03-10 15:46:38

4709: [Jsoi2011]柠檬

Time Limit: 10 Sec  Memory Limit: 128 MB

Description

Flute 很喜欢柠檬。它准备了一串用树枝串起来的贝壳，打算用一种魔法把贝壳变成柠檬。贝壳一共有 N (1 ≤ N
 ≤ 100,000) 只，按顺序串在树枝上。为了方便，我们从左到右给贝壳编号 1..N。每只贝壳的大小不一定相同，
贝壳 i 的大小为 si(1 ≤ si ≤10,000)。变柠檬的魔法要求，Flute 每次从树枝一端取下一小段连续的贝壳，并
选择一种贝壳的大小 s0。如果 这一小段贝壳中 大小为 s0 的贝壳有 t 只，那么魔法可以把这一小段贝壳变成 s
0t^2 只柠檬。Flute 可以取任意多次贝壳，直到树枝上的贝壳被全部取完。各个小段中，Flute 选择的贝壳大小 s
0 可以不同。而最终 Flute 得到的柠檬数，就是所有小段柠檬数的总和。Flute 想知道，它最多能用这一串贝壳
变出多少柠檬。请你帮忙解决这个问题。

Input

第 1 行：一个整数，表示 N。
第 2 .. N + 1 行：每行一个整数，第 i + 1 行表示 si。

Output

仅一个整数，表示 Flute 最多能得到的柠檬数。

Sample Input

5
2
2
5
2
3

Sample Output

21
HINT:Flute 先从左端取下 4 只贝壳，它们的大小为 2, 2, 5, 2。选择 s0 = 2，那么这一段
里有 3 只大小为 s0 的贝壳，通过魔法可以得到 2×3^2 = 18 只柠檬。再从右端取下最后一
只贝壳，通过魔法可以得到 1×3^1 = 3 只柠檬。总共可以得到 18 + 3 = 21 只柠檬。没有
比这更优的方案了。
题解
膜一波ZYFdalao
首先我们需要分析出一个性质:同一段被取下的贝壳,其第一个和最后一个的大小一定是一样的.
证明很容易.用反证法来看,如果有一段贝壳,开头和结尾不一样,显然把这两段分开施法更优
这样,我们又可以想出一个暴力:每一个节点的最大价值f[i],肯定是由前面某一个和他权值相同的点转移而来
设出现次数为s[i],权值为a[i]
显然我们可以写出暴力式子:f[i]=max{f[j-1]+a[i]*(s[i]-s[j]+1)2,a[i]==a[j]}
但是这样肯定会T的,所以我们考虑优化
对于两个决策点j1和j2,设j1<j2
如果一开始f[j1-1]+a[i]*(s[i]-s[j1]+1)2<f[j2-1]+a[i]*(s[i]-s[j2]+1)2
随着i不断向后,由于平方的增速很快,越靠前的j位置,越有可能反超,使得f[j1-1]+a[i']*(s[i']-s[j1]+1)2>f[j2-1]+a[i']*(s[i']-s[j2]+1)2
所以我们就不能再用单调队列了,因为如果我们删去了前面的某个点,它在后面可能会变得更优
于是我们考虑用单调栈维护这个东西(单调栈的实现方法随意,我个人使用了vector)
每次决策的时候,我们就用栈顶来更新
由于一个贝壳可以自己画为一段,所以应该在插入之后再计算f值
当上述"反超"的情况出现时,肯定是最先发生在栈顶和栈顶第二个元素
假设我们已经插入因此我们就看,第二个元素有"多快"超过栈顶
这个所谓的"速度"用经过a[i]的个数来体现,也就是说,看第二个元素经过几个a[i]超过栈顶,记为cnt1,如果cnt1<=s[i],我们就可以弹栈了.
但是,可能会出现一种情况:第三个元素超过栈顶,而第二个元素还没有超过栈顶
设j2<j1<i2<i1,如果j2,j1,i2三者满足这种情况,那么j2,j1,i1肯定更加满足这种情况,所以直接删掉j1就行
所以,在我们插入之前,可以先进行一波上述弹栈:如果发现第二个元素超过栈顶比栈顶超过i快,就弹栈;
所以最后的顺序是:排查->插入->计算
代码见下:
 #include<cstdio>

 #include<cstring>

 #include<vector>

 #include<deque>

 #include<algorithm>

 using namespace std;

 typedef long long LL;

 const int N=;

 const int V=;

 LL a[N],f[N];

 int id[N],cnt[V],n;

 vector<int>q[V];

 inline int max(int a,int b){return a>b?a:b;}

 inline LL calc(int x,int y){return f[x-]+a[x]*y*y;}

 inline int k(int x,int y)

 {

     int le=,ri=n,ret=n+;

     while(le<=ri)

     {

         int mi=(le+ri)>>;

         if(calc(x,mi-id[x]+)>=calc(y,mi-id[y]+))

             ret=mi,ri=mi-;

         else le=mi+;

     }

     return ret;

 }

 int main()

 {

     scanf("%d",&n);int o;

     for(int i=;i<=n;i++)

     {

         scanf("%lld",&a[i]);o=a[i];

         cnt[o]++,id[i]=cnt[o];

         while(q[o].size()>=&&k(q[o][q[o].size()-],q[o][q[o].size()-])<=k(q[o][q[o].size()-],i))q[o].pop_back();

         q[o].push_back(i);

         while(q[o].size()>=&&k(q[o][q[o].size()-],q[o][q[o].size()-])<=id[i])q[o].pop_back();

         f[i]=calc(q[o][q[o].size()-],id[i]-id[q[o][q[o].size()-]]+);

     }

     printf("%lld",f[n]);

 }

BZOJ4907

相关文章