码迷,mamicode.com
首页 > 其他好文 > 详细

【CodeForces】868F. Yet Another Minimization Problem

时间:2019-07-05 20:51:22      阅读:83      评论:0      收藏:0      [点我收藏+]

标签:int   main   cstring   答案   stdin   work   最优   lin   ace   

原题链接

题目大意是有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;
}

【CodeForces】868F. Yet Another Minimization Problem

标签:int   main   cstring   答案   stdin   work   最优   lin   ace   

原文地址:https://www.cnblogs.com/ivorysi/p/11140649.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!