[Jsoi2011]柠檬

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 想知道，它最多能用这一串贝壳

变出多少柠檬。请你帮忙解决这个问题。

Solution

一个奇怪的利用单调栈来优化的dp

Code

#include <vector>

#include <stdio.h>

#include <string.h>

#include <iostream>

#include <algorithm>

const int N = 100005;

int cnt[N], S[N], A[N];

long long f[N];

int n;

long long Calc(int Begin, int Len) {

	return f[Begin - 1] + (long long) A[Begin] * Len * Len;

}

int P(int j1, int j2) {

	int l = 1, r = n, mid, res = n + 1;

	while (l <= r) {

		mid = l + r >> 1;

		if (Calc(j1, mid - S[j1] + 1) >= Calc(j2, mid - S[j2] + 1))

			r = mid - 1, res = mid;

		else l = mid + 1;

	}

	return res;

}

std:: vector<int> Q[N];

#define ET(i) (i->size() > 1)

#define F(i, j) (*i)[i->size() - j]

int main () {

	scanf("%d", &n);

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

		scanf("%d", &A[i]), S[i] = ++cnt[A[i]];

		std:: vector<int>* Que = &(Q[A[i]]);

		while (ET(Que) and P(F(Que, 2), i) <= P(F(Que, 1), i))

			Que->pop_back();

		Que->push_back(i);

		while (ET(Que) and P(F(Que, 2), F(Que, 1)) <= S[i])

			Que->pop_back();

		f[i] = Calc(F(Que, 1), S[i] - S[F(Que, 1)] + 1);

	}

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

	return 0;

}