【CodeForces】868F. Yet Another Minimization Problem

原题链接

题目大意是有N个数，分成K段，每一段的花费是这个数里相同的数的数对个数，要求花费最小

如果只是区间里相同数对个数的话，莫队就够了

而这里是！边单调性优化边莫队（只是类似莫队）！而移动的次数和分治的复杂度是一样的！

这个时候就不能用单调栈+二分了，得用分治

分治的方法就是\(Solve(l,r,ql,qr)\)表示我想计算区间\([l,r]\)的答案，然后转移过来的区间在\([ql,qr]\)

暴力计算出\(f[mid]\)的值，找到最优转移点是\(k\)，然后分成\(Solve(l,mid,ql,k)\)和\(Solve(mid +1,r,k,qr)\)做下去

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <vector>
#include <bitset>
#define fi first
#define se second
#define pii pair<int,int>
#define mp make_pair
#define pb push_back
#define space putchar(' ')
#define enter putchar('\n')
#define MAXN 100005
//#define ivorysi
using namespace std;
typedef long long int64;
template<class T>
void read(T &res) {
	res = 0;T f = 1;char c = getchar();
	while(c < '0' || c > '9') {
		if(c == '-') f = -1;
		c = getchar();
	}
	while(c >= '0' && c <= '9') {
		res = res * 10 + c - '0';
		c = getchar();
	}
	res *= f;
}
template<class T>
void out(T x) {
	if(x < 0) {x = -x;putchar('-');}
	if(x >= 10) out(x / 10);
	putchar('0' + x % 10);
}
int N,K,a[MAXN],cnt[MAXN];
int64 f[22][MAXN],w = 0;
int nw,p,q;
int64 work(int l,int r) {
	while(q < r) w += cnt[a[++q]]++;
	while(p > l) w += cnt[a[--p]]++;
	while(p < l) w -= --cnt[a[p++]];
	while(q > r) w -= --cnt[a[q--]];
	return w;
}
int64 Calc(int a,int b) {
	return f[nw - 1][a] + work(a + 1,b);
}
void Solve(int l,int r,int ql,int qr) {
	if(l > r) return;
	int mid = (l + r) >> 1,p = min(qr,mid - 1);
	int k;
	f[nw][mid] = Calc(ql,mid);k = ql;
	for(int i = ql + 1 ; i <= p ; ++i) {
		int64 x = Calc(i,mid);
		if(x <= f[nw][mid]) {
			f[nw][mid] = x;k = i;
		}
	}
	Solve(l,mid - 1,ql,k);
	Solve(mid + 1,r,k,qr);
}
int main() {
#ifdef ivorysi
	freopen("f1.in","r",stdin);
#endif
	read(N);read(K);
	for(int i = 1 ; i <= N ; ++i) {read(a[i]);f[1][i] = f[1][i - 1] + cnt[a[i]]++;}
	
	for(int i = 2 ; i <= K ; ++i) {
		memset(cnt,0,sizeof(cnt));
		p = 1;q = 0;w = 0;
		nw = i;
		Solve(0,N,0,N);
	}
	out(f[K][N]);enter;
}